diff options
Diffstat (limited to 'MySQLdb/tests/configdb.py')
-rw-r--r-- | MySQLdb/tests/configdb.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/MySQLdb/tests/configdb.py b/MySQLdb/tests/configdb.py new file mode 100644 index 0000000..cd6d43d --- /dev/null +++ b/MySQLdb/tests/configdb.py @@ -0,0 +1,25 @@ +"""Configure database connection for tests.""" + +from os import environ, path + +tests_path = path.dirname(__file__) +conf_file = environ.get('TESTDB', 'default.cnf') +conf_path = path.join(tests_path, conf_file) +connect_kwargs = dict( + read_default_file = conf_path, + read_default_group = "MySQLdb-tests", +) + +def connection_kwargs(kwargs): + db_kwargs = connect_kwargs.copy() + db_kwargs.update(kwargs) + return db_kwargs + +def connection_factory(**kwargs): + import MySQLdb + db_kwargs = connection_kwargs(kwargs) + db = MySQLdb.connect(**db_kwargs) + return db + + + |