Add PSYCOPG2_TESTDB_PASSWORD variable for tests

Add a check for the PSYCOPG2_TESTDB_PASSWORD environment variable if the
database user (or database settings) requires a password.
This commit is contained in:
Jason Erickson 2011-01-28 14:06:31 -07:00
parent 9433a6879f
commit dcd2e19bde

View File

@ -30,6 +30,7 @@ dbname = os.environ.get('PSYCOPG2_TESTDB', 'psycopg2_test')
dbhost = os.environ.get('PSYCOPG2_TESTDB_HOST', None) dbhost = os.environ.get('PSYCOPG2_TESTDB_HOST', None)
dbport = os.environ.get('PSYCOPG2_TESTDB_PORT', None) dbport = os.environ.get('PSYCOPG2_TESTDB_PORT', None)
dbuser = os.environ.get('PSYCOPG2_TESTDB_USER', None) dbuser = os.environ.get('PSYCOPG2_TESTDB_USER', None)
dbpass = os.environ.get('PSYCOPG2_TESTDB_PASSWORD', None)
# Check if we want to test psycopg's green path. # Check if we want to test psycopg's green path.
green = os.environ.get('PSYCOPG2_TEST_GREEN', None) green = os.environ.get('PSYCOPG2_TEST_GREEN', None)
@ -53,6 +54,8 @@ if dbport is not None:
dsn += ' port=%s' % dbport dsn += ' port=%s' % dbport
if dbuser is not None: if dbuser is not None:
dsn += ' user=%s' % dbuser dsn += ' user=%s' % dbuser
if dbpass is not None:
dsn += ' password=%s' % dbpass
# If connection to test db fails, bail out early. # If connection to test db fails, bail out early.
import psycopg2 import psycopg2