from asynctest import TestCase
from tortoise import Tortoise

from alice.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()