def full_database_setup():
"""Sets up a clean database based on the model metadata. It also
calculates the FK relationships between tables so we can delete them in
order. It yields a tuple the _DB and ordered list of tables."""
rand_id = str(randint(1, 9999))
config = dataactcore.config.CONFIG_DB
config['db_name'] = 'unittest{}_data_broker'.format(rand_id)
dataactcore.config.CONFIG_DB = config
create_database(config['db_name'])
db = GlobalDB.db()
run_migrations()
creation_order = baseModel.Base.metadata.sorted_tables
yield (db, list(reversed(creation_order))) # drop order
GlobalDB.close()
drop_database(config['db_name'])
def full_database_setup():
"""Sets up a clean database based on the model metadata. It also
calculates the FK relationships between tables so we can delete them in
order. It yields a tuple the _DB and ordered list of tables."""
rand_id = str(randint(1, 9999))
config = dataactcore.config.CONFIG_DB
config['db_name'] = 'unittest{}_data_broker'.format(rand_id)
dataactcore.config.CONFIG_DB = config
create_database(config['db_name'])
db = GlobalDB.db()
run_migrations()
creation_order = baseModel.Base.metadata.sorted_tables
yield (db, list(reversed(creation_order))) # drop order
GlobalDB.close()
drop_database(config['db_name'])
def tearDownClass(cls):
"""Tear down class-level resources."""
GlobalDB.close()
drop_database(CONFIG_DB['db_name'])