Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(db): update test for migration chain for missing from 0

Browse Source
This commit is contained in:
psychedelicious 2023-12-11 12:48:28 +11:00
parent 41db92b9e8
commit 77065b1ce1
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_sqlite_migrator.py
View File

@ -126,16 +126,19 @@ def test_migration_set_gets_migration(migration_no_op: Migration):
assert migration_set.get(1) is None assert migration_set.get(1) is None
def test_migration_set_validates_migration_path(no_op_migrate_callback: MigrateCallback): def test_migration_set_validates_migration_chain(no_op_migrate_callback: MigrateCallback):
migration_set = MigrationSet() migration_set = MigrationSet()
migration_set.validate_migration_chain() migration_set.register(Migration(from_version=1, to_version=2, migrate=no_op_migrate_callback))
with pytest.raises(MigrationError, match="Migration chain is fragmented"):
# no migration from 0 to 1
migration_set.validate_migration_chain()
migration_set.register(Migration(from_version=0, to_version=1, migrate=no_op_migrate_callback)) migration_set.register(Migration(from_version=0, to_version=1, migrate=no_op_migrate_callback))
migration_set.validate_migration_chain() migration_set.validate_migration_chain()
migration_set.register(Migration(from_version=1, to_version=2, migrate=no_op_migrate_callback))
migration_set.register(Migration(from_version=2, to_version=3, migrate=no_op_migrate_callback)) migration_set.register(Migration(from_version=2, to_version=3, migrate=no_op_migrate_callback))
migration_set.validate_migration_chain() migration_set.validate_migration_chain()
migration_set.register(Migration(from_version=4, to_version=5, migrate=no_op_migrate_callback)) migration_set.register(Migration(from_version=4, to_version=5, migrate=no_op_migrate_callback))
with pytest.raises(MigrationError, match="Migration chain is fragmented"): with pytest.raises(MigrationError, match="Migration chain is fragmented"):
# no migration from 3 to 4
migration_set.validate_migration_chain() migration_set.validate_migration_chain()