80 lines
3.3 KiB
Python
80 lines
3.3 KiB
Python
import pytest
|
|
from pytest_mock import MockerFixture
|
|
from tortoise import Tortoise
|
|
|
|
from aerich.ddl.mysql import MysqlDDL
|
|
from aerich.ddl.postgres import PostgresDDL
|
|
from aerich.ddl.sqlite import SqliteDDL
|
|
from aerich.exceptions import NotSupportError
|
|
from aerich.migrate import Migrate
|
|
|
|
|
|
def test_migrate(mocker: MockerFixture):
|
|
mocker.patch("click.prompt", return_value=True)
|
|
apps = Tortoise.apps
|
|
models = apps.get("models")
|
|
diff_models = apps.get("diff_models")
|
|
Migrate.diff_models(diff_models, models)
|
|
if isinstance(Migrate.ddl, SqliteDDL):
|
|
with pytest.raises(NotSupportError):
|
|
Migrate.diff_models(models, diff_models, False)
|
|
else:
|
|
Migrate.diff_models(models, diff_models, False)
|
|
Migrate._merge_operators()
|
|
if isinstance(Migrate.ddl, MysqlDDL):
|
|
assert Migrate.upgrade_operators == [
|
|
"ALTER TABLE `email` DROP FOREIGN KEY `fk_email_user_5b58673d`",
|
|
"ALTER TABLE `category` ADD `name` VARCHAR(200) NOT NULL",
|
|
"ALTER TABLE `user` ADD UNIQUE INDEX `uid_user_usernam_9987ab` (`username`)",
|
|
"ALTER TABLE `user` RENAME COLUMN `last_login_at` TO `last_login`",
|
|
]
|
|
assert Migrate.downgrade_operators == [
|
|
"ALTER TABLE `category` DROP COLUMN `name`",
|
|
"ALTER TABLE `user` DROP INDEX `uid_user_usernam_9987ab`",
|
|
"ALTER TABLE `user` RENAME COLUMN `last_login` TO `last_login_at`",
|
|
"ALTER TABLE `email` ADD CONSTRAINT `fk_email_user_5b58673d` FOREIGN KEY "
|
|
"(`user_id`) REFERENCES `user` (`id`) ON DELETE CASCADE",
|
|
]
|
|
elif isinstance(Migrate.ddl, PostgresDDL):
|
|
assert Migrate.upgrade_operators == [
|
|
'ALTER TABLE "email" DROP CONSTRAINT "fk_email_user_5b58673d"',
|
|
'ALTER TABLE "category" ADD "name" VARCHAR(200) NOT NULL',
|
|
'ALTER TABLE "user" ADD CONSTRAINT "uid_user_usernam_9987ab" UNIQUE ("username")',
|
|
'ALTER TABLE "user" RENAME COLUMN "last_login_at" TO "last_login"',
|
|
]
|
|
assert Migrate.downgrade_operators == [
|
|
'ALTER TABLE "category" DROP COLUMN "name"',
|
|
'ALTER TABLE "user" DROP CONSTRAINT "uid_user_usernam_9987ab"',
|
|
'ALTER TABLE "user" RENAME COLUMN "last_login" TO "last_login_at"',
|
|
'ALTER TABLE "email" ADD CONSTRAINT "fk_email_user_5b58673d" FOREIGN KEY ("user_id") REFERENCES "user" ("id") ON DELETE CASCADE',
|
|
]
|
|
elif isinstance(Migrate.ddl, SqliteDDL):
|
|
assert Migrate.upgrade_operators == [
|
|
'ALTER TABLE "email" DROP FOREIGN KEY "fk_email_user_5b58673d"',
|
|
'ALTER TABLE "category" ADD "name" VARCHAR(200) NOT NULL',
|
|
'ALTER TABLE "user" ADD UNIQUE INDEX "uid_user_usernam_9987ab" ("username")',
|
|
'ALTER TABLE "user" RENAME COLUMN "last_login_at" TO "last_login"',
|
|
]
|
|
assert Migrate.downgrade_operators == []
|
|
|
|
|
|
def test_sort_all_version_files(mocker):
|
|
mocker.patch(
|
|
"os.listdir",
|
|
return_value=[
|
|
"1_datetime_update.json",
|
|
"11_datetime_update.json",
|
|
"10_datetime_update.json",
|
|
"2_datetime_update.json",
|
|
],
|
|
)
|
|
|
|
Migrate.migrate_location = "."
|
|
|
|
assert Migrate.get_all_version_files() == [
|
|
"1_datetime_update.json",
|
|
"2_datetime_update.json",
|
|
"10_datetime_update.json",
|
|
"11_datetime_update.json",
|
|
]
|