Fix drop table

2021-02-04 14:23:46 +08:00
parent 048e428eac
commit abfa60133f
4 changed files with 16 additions and 13 deletions
--- a/tests/test_ddl.py
+++ b/tests/test_ddl.py
@@ -50,7 +50,7 @@ COMMENT ON COLUMN "category"."user_id" IS 'User';"""


 def test_drop_table():
-    ret = Migrate.ddl.drop_table(Category)
+    ret = Migrate.ddl.drop_table(Category._meta.db_table)
    if isinstance(Migrate.ddl, MysqlDDL):
        assert ret == "DROP TABLE IF EXISTS `category`"
    else: