def empty_db_connection():
"""
Connect to db, create tables and begin connection-level (outer) transaction.
Will also create the database first if it does not exist.
Maybe overkill, but a neat trick, cf:
http://alextechrants.blogspot.co.uk/2013/08/unit-testing-sqlalchemy-apps.html
We use a connection-level (non-ORM) transaction that encloses everything
**including the table creation**.
This ensures that the models and queries are always in sync for testing.
Note that the transaction must be explicitly rolled back. An alternative
approach would be to delete all tables, but presumably that would
take slightly longer than a rollback (be interesting to check).
"""
# Could be parameterized, but then it wouldn't make sense as a session-level fixture
# Unless PyTest is smart enough to maintain fixtures in parallel?
# test_db_url = getattr(request.module, "test_db_url", testdb_empty_url)
test_db_url = testdb_empty_url
if not db_utils.check_database_exists(test_db_url):
db_utils.create_empty_database(default_admin_db_url,
test_db_url.database)
engine = create_engine(test_db_url)
connection = engine.connect()
transaction = connection.begin()
# Create tables (will be rolled back to clean)
db_utils.create_tables_and_indexes(connection)
# Return to test function
yield connection
# TearDown
transaction.rollback()
connection.close()
engine.dispose()
def empty_db_connection():
"""
Connect to db, create tables and begin connection-level (outer) transaction.
Will also create the database first if it does not exist.
Maybe overkill, but a neat trick, cf:
http://alextechrants.blogspot.co.uk/2013/08/unit-testing-sqlalchemy-apps.html
We use a connection-level (non-ORM) transaction that encloses everything
**including the table creation**.
This ensures that the models and queries are always in sync for testing.
Note that the transaction must be explicitly rolled back. An alternative
approach would be to delete all tables, but presumably that would
take slightly longer than a rollback (be interesting to check).
"""
# Could be parameterized, but then it wouldn't make sense as a session-level fixture
# Unless PyTest is smart enough to maintain fixtures in parallel?
# test_db_url = getattr(request.module, "test_db_url", testdb_empty_url)
test_db_url = testdb_empty_url
if not db_utils.check_database_exists(test_db_url):
db_utils.create_empty_database(default_admin_db_url, test_db_url.database)
engine = create_engine(test_db_url)
connection = engine.connect()
transaction = connection.begin()
# Create tables (will be rolled back to clean)
db_utils.create_tables_and_indexes(connection)
# Return to test function
yield connection
# TearDown
transaction.rollback()
connection.close()
engine.dispose()
def main():
args = handle_args()
dburl = dbconfig.make_db_url(dbconfig.default_admin_db_params, args.dbname)
if not db_utils.check_database_exists(dburl):
db_utils.create_empty_database(dbconfig.default_admin_db_url,
args.dbname)
logger.info('Database "{}" created.'.format(dburl.database))
engine = create_engine(dburl)
db_utils.create_tables_and_indexes(engine.connect())
return 0
def main():
args = handle_args()
dburl = dbconfig.make_db_url(dbconfig.default_admin_db_params, args.dbname)
if not db_utils.check_database_exists(dburl):
db_utils.create_empty_database(dbconfig.default_admin_db_url,
args.dbname)
logger.info('Database "{}" created.'.format(dburl.database))
engine = create_engine(dburl)
db_utils.create_tables_and_indexes(engine.connect())
return 0