203781210c
优化所有alembic迁移脚本的离线模式支持，添加已存在对象检查逻辑，避免重复执行出错。 为各个迁移脚本添加对应的单元测试，覆盖核心场景保障逻辑正确性。 修改前端配置文件，将默认API地址切换为线上服务器8.138.234.141:4000。 新增本地数据备份文件wonderq-local-data.dump和项目发布包wonderq-admin-release-20260823-01.tar。
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from importlib.util import module_from_spec, spec_from_file_location
|
|
from pathlib import Path
|
|
|
|
|
|
MIGRATION_PATH = Path(__file__).parents[1] / "alembic" / "versions" / "0024_vehicle_demand.py"
|
|
SPEC = spec_from_file_location("vehicle_demand_migration", MIGRATION_PATH)
|
|
assert SPEC and SPEC.loader
|
|
MIGRATION = module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(MIGRATION)
|
|
|
|
|
|
def test_vehicle_demand_columns_are_added_only_when_missing():
|
|
assert MIGRATION.missing_lead_columns(set()) == {
|
|
"leadType",
|
|
"contactName",
|
|
"customerId",
|
|
"vehicleDemand",
|
|
}
|
|
assert MIGRATION.missing_lead_columns({"leadType", "contactName"}) == {
|
|
"customerId",
|
|
"vehicleDemand",
|
|
}
|
|
|
|
|
|
def test_vehicle_demand_indexes_and_foreign_key_are_only_added_when_missing():
|
|
assert MIGRATION.missing_named_indexes(set(), MIGRATION.LEAD_INDEXES) == {
|
|
"ix_Lead_leadType",
|
|
"ix_Lead_customerId",
|
|
}
|
|
assert not MIGRATION.should_create_customer_fk(
|
|
[
|
|
{
|
|
"referred_table": "Customer",
|
|
"referred_columns": ["id"],
|
|
"constrained_columns": ["customerId"],
|
|
}
|
|
]
|
|
)
|