async def setUp(self): if Tortoise._inited: await self._tearDownDB() first_db_config = test.getDBConfig( app_label='models', modules=['tortoise.tests.testmodels'], ) second_db_config = test.getDBConfig( app_label='events', modules=['tortoise.tests.testmodels'], ) merged_config = { 'connections': { **first_db_config['connections'], **second_db_config['connections'] }, 'apps': { **first_db_config['apps'], **second_db_config['apps'] }, } await Tortoise.init(merged_config, _create_db=True) await Tortoise.generate_schemas() self.db = Tortoise.get_connection('models') self.second_db = Tortoise.get_connection('events')
async def asyncSetUp(self): await super().asyncSetUp() if Tortoise._inited: await self._tearDownDB() first_db_config = test.getDBConfig(app_label="models", modules=["tests.testmodels"]) second_db_config = test.getDBConfig(app_label="events", modules=["tests.testmodels"]) merged_config = { "connections": {**first_db_config["connections"], **second_db_config["connections"]}, "apps": {**first_db_config["apps"], **second_db_config["apps"]}, } await Tortoise.init(merged_config, _create_db=True) await Tortoise.generate_schemas() self.db = Tortoise.get_connection("models") self.second_db = Tortoise.get_connection("events")
async def asyncSetUp(self): await super().asyncSetUp() if Tortoise._inited: await self._tearDownDB() self.db_config = test.getDBConfig(app_label="models", modules=["tests.testmodels"]) if self.db_config["connections"]["models"]["engine"] != "tortoise.backends.asyncpg": raise test.SkipTest("PostgreSQL only")
async def asyncSetUp(self): await super().asyncSetUp() if Tortoise._inited: await self._tearDownDB() self.db_config = test.getDBConfig(app_label="models", modules=["tests.testmodels"]) if not self.is_asyncpg and not self.is_psycopg: raise test.SkipTest("PostgreSQL only")
async def setUp(self): try: Tortoise.apps = {} Tortoise._connections = {} Tortoise._inited = False except ConfigurationError: pass Tortoise._inited = False self.sqls = "" self.engine = test.getDBConfig( app_label="models", modules=[])["connections"]["models"]["engine"]