29 lines
723 B
Python
29 lines
723 B
Python
from asynctest import TestCase
|
|
from tortoise import Tortoise
|
|
|
|
from alice.backends.mysql import MysqlDDL
|
|
from alice.migrate import Migrate
|
|
|
|
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)
|
|
self.migrate = Migrate(ddl=self.ddl)
|
|
|
|
async def tearDown(self) -> None:
|
|
await Tortoise.close_connections()
|