xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-09 10:26:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

rewrite test for create_model_path

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin 2024-09-22 17:00:49 +09:00
parent 6fbceea4a6
commit 0df1e73220
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests-unit/prompt_server_test/download_models_test.py
View File

@ -202,19 +202,21 @@ async def test_download_model_invalid_folder_path():
assert result.status == 'error'
assert result.already_existed is False
# For create_model_path function
def test_create_model_path(tmp_path, monkeypatch):
mock_models_dir = tmp_path / "models"
monkeypatch.setattr('folder_paths.models_dir', str(mock_models_dir))
model_name = "model.safetensors"
folder_path = os.path.join(tmp_path, "mock_dir")
model_name = "test_model.sft"
model_directory = "test_dir"
file_path = create_model_path(model_name, folder_path)
file_path = create_model_path(model_name, model_directory, mock_models_dir)
assert file_path == str(mock_models_dir / model_directory / model_name)
assert file_path == os.path.join(folder_path, "model.safetensors")
assert os.path.exists(os.path.dirname(file_path))
with pytest.raises(Exception, match="Invalid model directory"):
create_model_path("../path_traversal.safetensors", folder_path)
with pytest.raises(Exception, match="Invalid model directory"):
create_model_path("/etc/some_root_path", folder_path)
@pytest.mark.asyncio
async def test_check_file_exists_when_file_exists(tmp_path):