psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-07-03 11:13:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

Bail out early from tests if connection to the test db fails.

Browse Source
This commit is contained in:
Daniele Varrazzo 2010-12-01 20:56:25 +00:00
parent 5f80dd38c4
commit 46a5f140c7
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/__init__.py
View File

@ -1,6 +1,7 @@
#!/usr/bin/env python #!/usr/bin/env python
import os import os
import sys
from testutils import unittest from testutils import unittest
dbname = os.environ.get('PSYCOPG2_TESTDB', 'psycopg2_test') dbname = os.environ.get('PSYCOPG2_TESTDB', 'psycopg2_test')
@ -31,6 +32,17 @@ if dbport is not None:
if dbuser is not None: if dbuser is not None:
dsn += ' user=%s' % dbuser dsn += ' user=%s' % dbuser
# If connection to test db fails, bail out early.
import psycopg2
try:
cnn = psycopg2.connect(dsn)
except Exception, e:
print "Failed connection to test db:", e.__class__.__name__, e
print "Please set env vars 'PSYCOPG2_TESTDB*' to valid values."
sys.exit(1)
else:
cnn.close()
import bugX000 import bugX000
import extras_dictcursor import extras_dictcursor
import test_dates import test_dates