20 lines
579 B
Python
20 lines
579 B
Python
from asynctest import TestCase
|
|
from tortoise import Tortoise
|
|
|
|
from aerich.ddl.mysql import MysqlDDL
|
|
|
|
TORTOISE_ORM = {
|
|
"connections": {"default": "mysql://root:123456@127.0.0.1:3306/test",},
|
|
"apps": {"models": {"models": ["tests.models"], "default_connection": "default",},},
|
|
}
|
|
|
|
|
|
class DBTestCase(TestCase):
|
|
async def setUp(self) -> None:
|
|
await Tortoise.init(config=TORTOISE_ORM)
|
|
self.client = Tortoise.get_connection("default")
|
|
self.ddl = MysqlDDL(self.client)
|
|
|
|
async def tearDown(self) -> None:
|
|
await Tortoise.close_connections()
|