From 8674142ba8a3e293f8aa998c7c70cd29834515a3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adam=20Ciarcin=CC=81ski?= Date: Thu, 6 Aug 2020 17:59:12 +0200 Subject: [PATCH] Fix test_migrate --- tests/test_migrate.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/tests/test_migrate.py b/tests/test_migrate.py index ed53778..3a51e1b 100644 --- a/tests/test_migrate.py +++ b/tests/test_migrate.py @@ -1,6 +1,7 @@ from tortoise import Tortoise from aerich.ddl.mysql import MysqlDDL +from aerich.ddl.postgres import PostgresDDL from aerich.migrate import Migrate @@ -19,6 +20,15 @@ def test_migrate(): "ALTER TABLE `category` DROP COLUMN `name`", "ALTER TABLE `user` DROP INDEX `uid_user_usernam_9987ab`", ] + elif isinstance(Migrate.ddl, PostgresDDL): + assert Migrate.upgrade_operators == [ + 'ALTER TABLE "category" ADD "name" VARCHAR(200) NOT NULL', + 'ALTER TABLE "user" ADD CONSTRAINT "uid_user_usernam_9987ab" UNIQUE ("username")', + ] + assert Migrate.downgrade_operators == [ + 'ALTER TABLE "category" DROP COLUMN "name"', + 'ALTER TABLE "user" DROP CONSTRAINT "uid_user_usernam_9987ab"', + ] else: assert Migrate.upgrade_operators == [ 'ALTER TABLE "category" ADD "name" VARCHAR(200) NOT NULL',