From 5d86d07618c5e0b2bb72ebf7ca14e3191e87898b Mon Sep 17 00:00:00 2001
From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Tue, 9 Apr 2013 22:05:17 +0100
Subject: [PATCH] Testing external script independent from psycopg
 implementation

Patch needed to run the test suite on psycopg2cffi or psycopg2ct.
---
 tests/test_notify.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/tests/test_notify.py b/tests/test_notify.py
index d048241d..f8383899 100755
--- a/tests/test_notify.py
+++ b/tests/test_notify.py
@@ -57,17 +57,18 @@ class NotifiesTests(ConnectingTestCase):
         script = ("""\
 import time
 time.sleep(%(sec)s)
-import psycopg2
-import psycopg2.extensions
+import %(module)s as psycopg2
+import %(module)s.extensions as ext
 conn = psycopg2.connect(%(dsn)r)
-conn.set_isolation_level(psycopg2.extensions.ISOLATION_LEVEL_AUTOCOMMIT)
+conn.set_isolation_level(ext.ISOLATION_LEVEL_AUTOCOMMIT)
 print conn.get_backend_pid()
 curs = conn.cursor()
 curs.execute("NOTIFY " %(name)r %(payload)r)
 curs.close()
 conn.close()
-"""
-            % { 'dsn': dsn, 'sec': sec, 'name': name, 'payload': payload})
+""" % {
+        'module': psycopg2.__name__,
+        'dsn': dsn, 'sec': sec, 'name': name, 'payload': payload})
 
         return Popen([sys.executable, '-c', script_to_py3(script)], stdout=PIPE)