298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 19:03:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test

Browse Source
This commit is contained in:
Chenlei Hu 2025-03-21 17:05:52 -04:00
parent 92de68aabd
commit 4a4c546276
1 changed files with 24 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests-unit/app_test/frontend_manager_test.py
View File

@ -69,8 +69,10 @@ def test_get_release_invalid_version(mock_provider):
def test_init_frontend_default():
version_string = DEFAULT_VERSION_STRING
frontend_path = FrontendManager.init_frontend(version_string)
assert frontend_path == FrontendManager.default_frontend_path()
frontend_init = FrontendManager.init_frontend(version_string)
assert isinstance(frontend_init, dict)
assert "web_root" in frontend_init
assert "version" in frontend_init
def test_init_frontend_invalid_version():
@ -138,37 +140,47 @@ def test_parse_version_string_invalid():
def test_init_frontend_default_with_mocks():
# Arrange
version_string = DEFAULT_VERSION_STRING
mock_path = "/mocked/path"
mock_version = (1, 0, 0)
# Act
with (
patch("app.frontend_management.check_frontend_version") as mock_check,
patch.object(
FrontendManager, "default_frontend_path", return_value="/mocked/path"
FrontendManager,
"init_default_frontend",
return_value={"web_root": mock_path, "version": mock_version},
),
):
frontend_path = FrontendManager.init_frontend(version_string)
frontend_init = FrontendManager.init_frontend(version_string)
# Assert
assert frontend_path == "/mocked/path"
mock_check.assert_called_once()
assert frontend_init["web_root"] == mock_path
assert frontend_init["version"] == mock_version
mock_check.assert_not_called() # check_frontend_version is now called inside init_default_frontend
def test_init_frontend_fallback_on_error():
# Arrange
version_string = "test-owner/test-repo@1.0.0"
mock_path = "/default/path"
mock_version = (1, 0, 0)
# Act
with (
patch.object(
FrontendManager, "init_frontend_unsafe", side_effect=Exception("Test error")
FrontendManager,
"init_frontend_unsafe",
side_effect=Exception("Test error")
),
patch("app.frontend_management.check_frontend_version") as mock_check,
patch.object(
FrontendManager, "default_frontend_path", return_value="/default/path"
FrontendManager,
"init_default_frontend",
return_value={"web_root": mock_path, "version": mock_version},
),
):
frontend_path = FrontendManager.init_frontend(version_string)
frontend_init = FrontendManager.init_frontend(version_string)
# Assert
assert frontend_path == "/default/path"
mock_check.assert_called_once()
assert frontend_init["web_root"] == mock_path
assert frontend_init["version"] == mock_version