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

add regression tests for <NOKEY> issue

Browse Source
This commit is contained in:
Lincoln Stein 2024-02-29 21:37:40 -05:00 committed by Kent Keirsey
parent 0832e1818e
commit 8fb297e5f6
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/app/services/model_install/test_model_install.py
View File

@ -31,6 +31,7 @@ def test_registration(mm2_installer: ModelInstallServiceBase, embedding_file: Pa
assert len(matches) == 0 assert len(matches) == 0
key = mm2_installer.register_path(embedding_file) key = mm2_installer.register_path(embedding_file)
assert key is not None assert key is not None
assert key != "<NOKEY>"
assert len(key) == 32 assert len(key) == 32
@ -58,12 +59,13 @@ def test_registration_meta_override_fail(mm2_installer: ModelInstallServiceBase,
def test_registration_meta_override_succeed(mm2_installer: ModelInstallServiceBase, embedding_file: Path) -> None: def test_registration_meta_override_succeed(mm2_installer: ModelInstallServiceBase, embedding_file: Path) -> None:
store = mm2_installer.record_store store = mm2_installer.record_store
key = mm2_installer.register_path( key = mm2_installer.register_path(
embedding_file, {"name": "banana_sushi", "source": "fake/repo_id", "current_hash": "New Hash"} embedding_file, {"name": "banana_sushi", "source": "fake/repo_id", "current_hash": "New Hash", "key": "xyzzy"}
) )
model_record = store.get_model(key) model_record = store.get_model(key)
assert model_record.name == "banana_sushi" assert model_record.name == "banana_sushi"
assert model_record.source == "fake/repo_id" assert model_record.source == "fake/repo_id"
assert model_record.current_hash == "New Hash" assert model_record.current_hash == "New Hash"
assert model_record.key == "xyzzy"
def test_install( def test_install(
@ -129,6 +131,7 @@ def test_background_install(
model_record = mm2_installer.record_store.get_model(key) model_record = mm2_installer.record_store.get_model(key)
assert model_record is not None assert model_record is not None
assert model_record.path == destination assert model_record.path == destination
assert model_record.key != "<NOKEY>"
assert Path(mm2_app_config.models_dir / model_record.path).exists() assert Path(mm2_app_config.models_dir / model_record.path).exists()
# see if metadata was properly passed through # see if metadata was properly passed through