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"]
