configdb.py 581 B

12345678910111213141516171819202122232425
  1. """Configure database connection for tests."""
  2. from os import environ, path
  3. tests_path = path.dirname(__file__)
  4. conf_file = environ.get('TESTDB', 'default.cnf')
  5. conf_path = path.join(tests_path, conf_file)
  6. connect_kwargs = dict(
  7. read_default_file = conf_path,
  8. read_default_group = "MySQLdb-tests",
  9. )
  10. def connection_kwargs(kwargs):
  11. db_kwargs = connect_kwargs.copy()
  12. db_kwargs.update(kwargs)
  13. return db_kwargs
  14. def connection_factory(**kwargs):
  15. import MySQLdb
  16. db_kwargs = connection_kwargs(kwargs)
  17. db = MySQLdb.connect(**db_kwargs)
  18. return db