add ddl

2020-05-12 19:33:21 +08:00
parent 75e7a46e85
commit 4e7e1626aa
11 changed files with 318 additions and 121 deletions
--- a/tests/init.py
+++ b/tests/init.py
@@ -1,68 +0,0 @@
-import datetime
-from enum import IntEnum
-from tortoise import fields, Model
-
-
-class ProductType(IntEnum):
-    article = 1
-    page = 2
-
-
-class PermissionAction(IntEnum):
-    create = 1
-    delete = 2
-    update = 3
-    read = 4
-
-
-class Status(IntEnum):
-    on = 1
-    off = 0
-
-
-class User(Model):
-    username = fields.CharField(max_length=20, unique=True)
-    password = fields.CharField(max_length=200)
-    last_login = fields.DatetimeField(description='Last Login', default=datetime.datetime.now)
-    is_active = fields.BooleanField(default=True, description='Is Active')
-    is_superuser = fields.BooleanField(default=False, description='Is SuperUser')
-    avatar = fields.CharField(max_length=200, default='')
-    intro = fields.TextField(default='')
-    created_at = fields.DatetimeField(auto_now_add=True)
-
-    def __str__(self):
-        return f'{self.pk}#{self.username}'
-
-
-class Category(Model):
-    slug = fields.CharField(max_length=200)
-    name = fields.CharField(max_length=200)
-    created_at = fields.DatetimeField(auto_now_add=True)
-
-    def __str__(self):
-        return f'{self.pk}#{self.name}'
-
-
-class Product(Model):
-    categories = fields.ManyToManyField('models.Category')
-    name = fields.CharField(max_length=50)
-    view_num = fields.IntField(description='View Num')
-    sort = fields.IntField()
-    is_reviewed = fields.BooleanField(description='Is Reviewed')
-    type = fields.IntEnumField(ProductType, description='Product Type')
-    image = fields.CharField(max_length=200)
-    body = fields.TextField()
-    created_at = fields.DatetimeField(auto_now_add=True)
-
-    def __str__(self):
-        return f'{self.pk}#{self.name}'
-
-
-class Config(Model):
-    label = fields.CharField(max_length=200)
-    key = fields.CharField(max_length=20)
-    value = fields.JSONField()
-    status: Status = fields.IntEnumField(Status, default=Status.on)
-
-    def __str__(self):
-        return f'{self.pk}#{self.label}'
--- a/tests/backends/init.py
+++ b/tests/backends/init.py
--- a/tests/backends/mysql/init.py
+++ b/tests/backends/mysql/init.py
@@ -2,28 +2,26 @@ from asynctest import TestCase
 from tortoise import Tortoise

 from alice.backends.mysql import MysqlDDL
-from tests import User
+from tests.models import Category

 TORTOISE_ORM = {
    'connections': {
-        'default': 'mysql://root:123456@127.0.0.1:3306/fastapi-admin'
+        'default': 'mysql://root:123456@127.0.0.1:3306/test'
    },
    'apps': {
        'models': {
-            'models': ['tests'],
+            'models': ['tests.models'],
            'default_connection': 'default',
        }
    }
 }


-class TestMysql(TestCase):
+class DBTestCase(TestCase):
    async def setUp(self) -> None:
        await Tortoise.init(config=TORTOISE_ORM)
-
-    async def test_create_table(self):
-        ddl = MysqlDDL(Tortoise.get_connection('default'), User)
-        print(ddl.create_table())
+        self.client = Tortoise.get_connection('default')
+        self.ddl = MysqlDDL(self.client)

    async def tearDown(self) -> None:
        await Tortoise.close_connections()
--- a/tests/backends/mysql/test_ddl.py
+++ b/tests/backends/mysql/test_ddl.py
@@ -0,0 +1,53 @@
+from tests.backends.mysql import DBTestCase
+from tests.models import Category, User
+
+
+class TestDDL(DBTestCase):
+    def test_create_table(self):
+        ret = self.ddl.create_table(Category)
+        self.assertEqual(
+            ret, """CREATE TABLE IF NOT EXISTS `category` (
+    `id` INT NOT NULL PRIMARY KEY AUTO_INCREMENT,
+    `slug` VARCHAR(200) NOT NULL,
+    `name` VARCHAR(200) NOT NULL,
+    `created_at` DATETIME(6) NOT NULL  DEFAULT CURRENT_TIMESTAMP(6),
+    `user_id` INT NOT NULL COMMENT 'User',
+    CONSTRAINT `fk_category_user_e2e3874c` FOREIGN KEY (`user_id`) REFERENCES `user` (`id`) ON DELETE CASCADE
+) CHARACTER SET utf8mb4;""")
+
+    def test_drop_table(self):
+        ret = self.ddl.drop_table(Category)
+        self.assertEqual(ret, "DROP TABLE category IF EXISTS")
+
+    def test_add_column(self):
+        ret = self.ddl.add_column(Category, Category._meta.fields_map.get('name'))
+        self.assertEqual(ret, "ALTER TABLE category ADD `name` VARCHAR(200) NOT NULL")
+
+    def test_drop_column(self):
+        ret = self.ddl.drop_column(Category, 'name')
+        self.assertEqual(ret, "ALTER TABLE category DROP COLUMN name")
+
+    def test_add_index(self):
+        ret = self.ddl.add_index(Category, ['name'])
+        self.assertEqual(ret, "ALTER TABLE category ADD  INDEX idx_category_name_8b0cb9 (`name`)")
+        ret = self.ddl.add_index(Category, ['name'], True)
+        self.assertEqual(ret, "ALTER TABLE category ADD UNIQUE INDEX uid_category_name_8b0cb9 (`name`)")
+
+    def test_drop_index(self):
+        ret = self.ddl.drop_index(Category, ['name'])
+        self.assertEqual(ret, "ALTER TABLE category DROP INDEX idx_category_name_8b0cb9")
+        ret = self.ddl.drop_index(Category, ['name'], True)
+        self.assertEqual(ret, "ALTER TABLE category DROP INDEX uid_category_name_8b0cb9")
+
+    def test_add_fk(self):
+        ret = self.ddl.add_fk(Category, Category._meta.fields_map.get('user'))
+        self.assertEqual(ret,
+                         "ALTER TABLE category ADD CONSTRAINT `fk_category_user_366ffa6f` FOREIGN KEY (`user`) REFERENCES `user` (`id`) ON DELETE CASCADE")
+
+    def test_drop_fk(self):
+        ret = self.ddl.drop_fk(Category, Category._meta.fields_map.get('user'))
+        self.assertEqual(ret, "ALTER TABLE category DROP FOREIGN KEY fk_category_user_366ffa6f")
+
+    async def test_aa(self):
+        user = await User.get(username='test')
+        await user.save()
--- a/tests/models.py
+++ b/tests/models.py
@@ -0,0 +1,70 @@
+import datetime
+from enum import IntEnum
+from tortoise import fields, Model
+
+
+class ProductType(IntEnum):
+    article = 1
+    page = 2
+
+
+class PermissionAction(IntEnum):
+    create = 1
+    delete = 2
+    update = 3
+    read = 4
+
+
+class Status(IntEnum):
+    on = 1
+    off = 0
+
+
+class User(Model):
+    username = fields.CharField(max_length=20, unique=True)
+    password = fields.CharField(max_length=200)
+    last_login = fields.DatetimeField(description='Last Login', default=datetime.datetime.now)
+    is_active = fields.BooleanField(default=True, description='Is Active')
+    is_superuser = fields.BooleanField(default=False, description='Is SuperUser')
+    avatar = fields.CharField(max_length=200, default='')
+    intro = fields.TextField(default='')
+    created_at = fields.DatetimeField(auto_now_add=True)
+    updated_at = fields.DatetimeField(auto_now=True)
+
+    def __str__(self):
+        return f'{self.pk}#{self.username}'
+
+
+class Category(Model):
+    slug = fields.CharField(max_length=200)
+    name = fields.CharField(max_length=200)
+    user = fields.ForeignKeyField('models.User', description='User')
+    created_at = fields.DatetimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.pk}#{self.name}'
+
+
+class Product(Model):
+    categories = fields.ManyToManyField('models.Category')
+    name = fields.CharField(max_length=50)
+    view_num = fields.IntField(description='View Num')
+    sort = fields.IntField()
+    is_reviewed = fields.BooleanField(description='Is Reviewed')
+    type = fields.IntEnumField(ProductType, description='Product Type')
+    image = fields.CharField(max_length=200)
+    body = fields.TextField()
+    created_at = fields.DatetimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f'{self.pk}#{self.name}'
+
+
+class Config(Model):
+    label = fields.CharField(max_length=200)
+    key = fields.CharField(max_length=20)
+    value = fields.JSONField()
+    status: Status = fields.IntEnumField(Status, default=Status.on)
+
+    def __str__(self):
+        return f'{self.pk}#{self.label}'