From 1108172e4cc42ac334e1b9f1ae3c81c9cf928caa Mon Sep 17 00:00:00 2001
From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Sat, 3 Apr 2010 22:23:39 +0100
Subject: [PATCH] The test suite can be run against the green path.

---
 tests/__init__.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/tests/__init__.py b/tests/__init__.py
index 5863de61..4e52ad25 100755
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -8,6 +8,13 @@ dbhost = os.environ.get('PSYCOPG2_TESTDB_HOST', None)
 dbport = os.environ.get('PSYCOPG2_TESTDB_PORT', None)
 dbuser = os.environ.get('PSYCOPG2_TESTDB_USER', None)
 
+# Check if we want to test psycopg's green path.
+green = os.environ.get('PSYCOPG2_TEST_GREEN', None)
+if green:
+    import psycopg2.extensions
+    import psycopg2.extras
+    psycopg2.extensions.set_wait_callback(psycopg2.extras.wait_select)
+
 # Construct a DSN to connect to the test database:
 dsn = 'dbname=%s' % dbname
 if dbhost is not None: