54 lines
1.1 KiB
Python
54 lines
1.1 KiB
Python
from types import ModuleType
|
|
|
|
from kanta.migrate import MigrationRegistry
|
|
|
|
|
|
def test_register_and_apply():
|
|
reg = MigrationRegistry()
|
|
|
|
@reg.register
|
|
def migrate_v1(d, ctx):
|
|
d["version"] = 1
|
|
|
|
@reg.register
|
|
def migrate_v2(d, ctx):
|
|
d["version"] = 2
|
|
|
|
state = {}
|
|
new_ver = reg.apply(state, current_version=0, silent=True)
|
|
assert new_ver == 2
|
|
assert state["version"] == 2
|
|
|
|
|
|
def test_no_migrations_needed():
|
|
reg = MigrationRegistry()
|
|
|
|
@reg.register
|
|
def migrate_v1(d, ctx):
|
|
d["x"] = 1
|
|
|
|
state = {"x": 1}
|
|
new_ver = reg.apply(state, current_version=1, silent=True)
|
|
assert new_ver == 1
|
|
|
|
|
|
def test_from_module():
|
|
mod = ModuleType("fake_migrations")
|
|
|
|
def migrate_v1(d, ctx):
|
|
d["v"] = 1
|
|
|
|
def migrate_v2(d, ctx):
|
|
d["v"] = 2
|
|
|
|
mod.__dict__["migrate_v1"] = migrate_v1
|
|
mod.__dict__["migrate_v2"] = migrate_v2
|
|
|
|
reg = MigrationRegistry.from_module(mod)
|
|
assert reg.dbver == 2
|
|
|
|
state = {}
|
|
new_ver = reg.apply(state, current_version=0, silent=True)
|
|
assert new_ver == 2
|
|
assert state["v"] == 2
|