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'])