mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-06 22:10:14 -03:00
66d1c96783
- Add POST /api/lm/switch-channel endpoint with git init / ZIP fallback - Add _backup_git/_restore_git helpers with safe rollback - Version-info endpoint now returns has_git flag for auto-detection - Check-updates always returns releases (changelog) regardless of channel - Nightly mode shows 'N commits behind main' with commit hash and date - View on GitHub link points to /commits/main in nightly mode - Channel toggle UI with pill-style buttons in update modal - Confirmation dialog with Esc / backdrop-dismiss support - Channel derived from has_git on every page load, no localStorage - i18n: 11 new keys translated across 9 non-English locales - CSS: unified card-style sections in _base.css - Tests: 8 new tests covering switch-channel, nightly response, init_git_repo
501 lines
15 KiB
Python
501 lines
15 KiB
Python
import logging
|
|
import os
|
|
import shutil
|
|
from aiohttp import ClientError
|
|
from aiohttp import web
|
|
import pytest
|
|
|
|
from py.routes import update_routes
|
|
|
|
|
|
def _fake_request(body=None, query_params=None):
|
|
from multidict import MultiDict
|
|
|
|
q = MultiDict(query_params or {})
|
|
|
|
req = type("Req", (), {
|
|
"has_body": body is not None,
|
|
"match_info": {},
|
|
"rel_url": type("U", (), {"query": q})(),
|
|
"query": q,
|
|
"app": {},
|
|
})()
|
|
|
|
async def _json():
|
|
return body or {}
|
|
|
|
req.json = _json
|
|
return req
|
|
|
|
|
|
class OfflineDownloader:
|
|
async def make_request(self, *_, **__):
|
|
return False, "Cannot connect to host"
|
|
|
|
|
|
class RaisingDownloader:
|
|
async def make_request(self, *_, **__):
|
|
raise ClientError("offline")
|
|
|
|
|
|
async def _stub_downloader(instance):
|
|
return instance
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_remote_version_offline_logs_without_traceback(monkeypatch, caplog):
|
|
caplog.set_level(logging.WARNING)
|
|
monkeypatch.setattr(update_routes, "get_downloader", lambda: _stub_downloader(OfflineDownloader()))
|
|
|
|
version, changelog, releases = await update_routes.UpdateRoutes._get_remote_version()
|
|
|
|
assert version == "v0.0.0"
|
|
assert changelog == []
|
|
assert releases == []
|
|
assert "Failed to fetch GitHub releases" in caplog.text
|
|
assert "Cannot connect to host" in caplog.text
|
|
assert "Traceback" not in caplog.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_remote_version_network_error_logs_warning(monkeypatch, caplog):
|
|
caplog.set_level(logging.WARNING)
|
|
monkeypatch.setattr(update_routes, "get_downloader", lambda: _stub_downloader(RaisingDownloader()))
|
|
|
|
version, changelog, releases = await update_routes.UpdateRoutes._get_remote_version()
|
|
|
|
assert version == "v0.0.0"
|
|
assert changelog == []
|
|
assert releases == []
|
|
assert "Unable to reach GitHub for release info" in caplog.text
|
|
assert "Traceback" not in caplog.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_nightly_version_network_error_logs_warning(monkeypatch, caplog):
|
|
caplog.set_level(logging.WARNING)
|
|
monkeypatch.setattr(update_routes, "get_downloader", lambda: _stub_downloader(RaisingDownloader()))
|
|
|
|
version, changelog, behind_by, commit_date = await update_routes.UpdateRoutes._get_nightly_version()
|
|
|
|
assert version == "main"
|
|
assert changelog == []
|
|
assert behind_by == 0
|
|
assert commit_date == ""
|
|
assert "Unable to reach GitHub for nightly version" in caplog.text
|
|
assert "Traceback" not in caplog.text
|
|
|
|
|
|
def test_clean_excludes_covers_user_data_dirs():
|
|
"""git clean must receive -e excludes for every user-managed dir."""
|
|
excludes = update_routes._clean_excludes()
|
|
assert "-e" in excludes # at least one exclude flag present
|
|
for name in update_routes._PRESERVE_DIRS:
|
|
assert name in excludes
|
|
assert f"{name}/**" in excludes
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_perform_git_update_preserves_user_dirs(monkeypatch, tmp_path):
|
|
"""``git clean`` must be called with -e excludes for user data dirs.
|
|
|
|
Regression test for portable-mode updates wiping wildcards/, stats/,
|
|
backups/, etc. because ``git clean -fd`` removed untracked, non-ignored
|
|
directories.
|
|
"""
|
|
calls = []
|
|
|
|
class FakeGit:
|
|
def reset(self, *args, **kwargs):
|
|
calls.append(("reset", args))
|
|
|
|
def clean(self, *args, **kwargs):
|
|
calls.append(("clean", args))
|
|
|
|
def checkout(self, *args, **kwargs):
|
|
calls.append(("checkout", args))
|
|
|
|
class FakeRemote:
|
|
def fetch(self):
|
|
calls.append(("fetch", ()))
|
|
|
|
def pull(self, *args, **kwargs):
|
|
calls.append(("pull", args))
|
|
|
|
class FakeRemotes:
|
|
origin = FakeRemote()
|
|
|
|
class FakeCommit:
|
|
hexsha = "abcdef123456"
|
|
|
|
class FakeHeads:
|
|
def __getitem__(self, name):
|
|
class Head:
|
|
def checkout(self_inner):
|
|
calls.append(("head-checkout", (name,)))
|
|
return Head()
|
|
|
|
class FakeBranches:
|
|
names = ["main"]
|
|
|
|
def __iter__(self):
|
|
class B:
|
|
name = "main"
|
|
return iter([B()])
|
|
|
|
class FakeRepo:
|
|
def __init__(self, path):
|
|
calls.append(("repo", (path,)))
|
|
|
|
git = FakeGit()
|
|
remotes = FakeRemotes()
|
|
head = type("H", (), {"commit": FakeCommit()})()
|
|
branches = FakeBranches()
|
|
heads = FakeHeads()
|
|
|
|
def create_head(self, name, ref):
|
|
calls.append(("create_head", (name, ref)))
|
|
|
|
class FakeGitModule:
|
|
class Repo:
|
|
def __new__(cls, path):
|
|
return FakeRepo(path)
|
|
|
|
class exc:
|
|
class GitError(Exception):
|
|
pass
|
|
|
|
import builtins
|
|
|
|
real_import = builtins.__import__
|
|
|
|
def fake_import(name, *args, **kwargs):
|
|
if name == "git":
|
|
return FakeGitModule
|
|
return real_import(name, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(builtins, "__import__", fake_import)
|
|
|
|
success, version = await update_routes.UpdateRoutes._perform_git_update(
|
|
str(tmp_path), nightly=True
|
|
)
|
|
|
|
assert success is True
|
|
clean_calls = [c for c in calls if c[0] == "clean"]
|
|
assert len(clean_calls) == 1
|
|
clean_args = clean_calls[0][1]
|
|
# Every preserved dir must be excluded via -e
|
|
for name in update_routes._PRESERVE_DIRS:
|
|
assert name in clean_args, f"{name} missing from git clean excludes"
|
|
assert f"{name}/**" in clean_args, f"{name}/** missing from git clean excludes"
|
|
# Ensure there's an -e before each name occurrence
|
|
idx = clean_args.index(name)
|
|
assert clean_args[idx - 1] == "-e"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_perform_git_update_stable_preserves_user_dirs(monkeypatch, tmp_path):
|
|
"""Stable (tag) update path must also pass -e excludes to git clean."""
|
|
calls = []
|
|
|
|
class FakeGit:
|
|
def reset(self, *args, **kwargs):
|
|
calls.append(("reset", args))
|
|
|
|
def clean(self, *args, **kwargs):
|
|
calls.append(("clean", args))
|
|
|
|
def checkout(self, *args, **kwargs):
|
|
calls.append(("checkout", args))
|
|
|
|
class FakeRemote:
|
|
def fetch(self):
|
|
calls.append(("fetch", ()))
|
|
|
|
class FakeRemotes:
|
|
origin = FakeRemote()
|
|
|
|
class FakeCommit:
|
|
committed_datetime = "2026-01-01"
|
|
|
|
class FakeTag:
|
|
name = "v9.9.9"
|
|
commit = FakeCommit()
|
|
|
|
class FakeRepo:
|
|
def __init__(self, path):
|
|
calls.append(("repo", (path,)))
|
|
|
|
git = FakeGit()
|
|
remotes = FakeRemotes()
|
|
tags = [FakeTag()]
|
|
|
|
class FakeGitModule:
|
|
class Repo:
|
|
def __new__(cls, path):
|
|
return FakeRepo(path)
|
|
|
|
class exc:
|
|
class GitError(Exception):
|
|
pass
|
|
|
|
import builtins
|
|
|
|
real_import = builtins.__import__
|
|
|
|
def fake_import(name, *args, **kwargs):
|
|
if name == "git":
|
|
return FakeGitModule
|
|
return real_import(name, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(builtins, "__import__", fake_import)
|
|
|
|
success, version = await update_routes.UpdateRoutes._perform_git_update(
|
|
str(tmp_path), nightly=False
|
|
)
|
|
|
|
assert success is True
|
|
assert version == "v9.9.9"
|
|
clean_calls = [c for c in calls if c[0] == "clean"]
|
|
assert len(clean_calls) == 1
|
|
clean_args = clean_calls[0][1]
|
|
for name in update_routes._PRESERVE_DIRS:
|
|
assert name in clean_args, f"{name} missing from git clean excludes (stable)"
|
|
|
|
def test_init_git_repo_creates_valid_repo(tmp_path, monkeypatch):
|
|
if not shutil.which("git"):
|
|
pytest.skip("git executable not found")
|
|
|
|
plugin_root = tmp_path / "plugin"
|
|
plugin_root.mkdir()
|
|
(plugin_root / ".tracking").write_text("pyproject.toml")
|
|
(plugin_root / "settings.json").write_text('{"some": "value"}')
|
|
|
|
try:
|
|
success, version = update_routes.UpdateRoutes._init_git_repo(str(plugin_root))
|
|
except Exception as e:
|
|
pytest.skip(f"Network unavailable for git fetch: {e}")
|
|
|
|
assert success is True
|
|
assert version.startswith("main-")
|
|
assert len(version) > len("main-")
|
|
assert (plugin_root / ".git").is_dir()
|
|
assert not (plugin_root / ".tracking").exists()
|
|
assert (plugin_root / "settings.json").exists()
|
|
assert (plugin_root / "pyproject.toml").exists()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_switch_channel_invalid_channel_returns_error():
|
|
req = _fake_request({"channel": "bad_channel"})
|
|
resp = await update_routes.UpdateRoutes.switch_channel(req)
|
|
|
|
data = _raw_body(resp)
|
|
assert not data["success"]
|
|
assert "Invalid channel" in data["error"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_switch_channel_to_nightly_without_git_inits_repo(monkeypatch, tmp_path):
|
|
routes_file = tmp_path / "py" / "routes" / "update_routes.py"
|
|
routes_file.parent.mkdir(parents=True)
|
|
routes_file.write_text("")
|
|
monkeypatch.setattr(update_routes, "__file__", str(routes_file))
|
|
monkeypatch.setattr(update_routes, "ensure_settings_file", lambda logger: str(tmp_path / "settings.json"))
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes,
|
|
"_init_git_repo",
|
|
staticmethod(lambda plugin_root: (True, "main-fedcba9")),
|
|
)
|
|
|
|
req = _fake_request({"channel": "nightly"})
|
|
resp = await update_routes.UpdateRoutes.switch_channel(req)
|
|
data = _raw_body(resp)
|
|
|
|
assert data["success"] is True
|
|
assert data["channel"] == "nightly"
|
|
assert data["new_version"] == "main-fedcba9"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_switch_channel_to_nightly_with_git_calls_git_update(monkeypatch, tmp_path):
|
|
routes_file = tmp_path / "py" / "routes" / "update_routes.py"
|
|
routes_file.parent.mkdir(parents=True)
|
|
routes_file.write_text("")
|
|
monkeypatch.setattr(update_routes, "__file__", str(routes_file))
|
|
monkeypatch.setattr(update_routes, "ensure_settings_file", lambda logger: str(tmp_path / "settings.json"))
|
|
|
|
(tmp_path / ".git").mkdir()
|
|
|
|
async def _fake_git_update(*args, **kwargs):
|
|
return True, "main-1111111"
|
|
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes, "_perform_git_update", _fake_git_update
|
|
)
|
|
|
|
req = _fake_request({"channel": "nightly"})
|
|
resp = await update_routes.UpdateRoutes.switch_channel(req)
|
|
data = _raw_body(resp)
|
|
|
|
assert data["success"] is True
|
|
assert data["channel"] == "nightly"
|
|
assert data["new_version"] == "main-1111111"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_switch_channel_to_release_with_git_downloads_zip(monkeypatch, tmp_path):
|
|
routes_file = tmp_path / "py" / "routes" / "update_routes.py"
|
|
routes_file.parent.mkdir(parents=True)
|
|
routes_file.write_text("")
|
|
monkeypatch.setattr(update_routes, "__file__", str(routes_file))
|
|
monkeypatch.setattr(update_routes, "ensure_settings_file", lambda logger: str(tmp_path / "settings.json"))
|
|
|
|
(tmp_path / ".git").mkdir()
|
|
|
|
async def _fake_zip(*args, **kwargs):
|
|
return True, "v9.9.9"
|
|
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes, "_download_and_replace_zip", _fake_zip
|
|
)
|
|
|
|
req = _fake_request({"channel": "release"})
|
|
resp = await update_routes.UpdateRoutes.switch_channel(req)
|
|
data = _raw_body(resp)
|
|
|
|
assert data["success"] is True
|
|
assert data["channel"] == "release"
|
|
assert data["new_version"] == "v9.9.9"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_switch_channel_to_release_without_git_still_downloads_zip(monkeypatch, tmp_path):
|
|
routes_file = tmp_path / "py" / "routes" / "update_routes.py"
|
|
routes_file.parent.mkdir(parents=True)
|
|
routes_file.write_text("")
|
|
monkeypatch.setattr(update_routes, "__file__", str(routes_file))
|
|
monkeypatch.setattr(update_routes, "ensure_settings_file", lambda logger: str(tmp_path / "settings.json"))
|
|
|
|
async def _fake_zip(*args, **kwargs):
|
|
return True, "v2.0.0"
|
|
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes, "_download_and_replace_zip", _fake_zip
|
|
)
|
|
|
|
req = _fake_request({"channel": "release"})
|
|
resp = await update_routes.UpdateRoutes.switch_channel(req)
|
|
data = _raw_body(resp)
|
|
|
|
assert data["success"] is True
|
|
assert data["channel"] == "release"
|
|
assert data["new_version"] == "v2.0.0"
|
|
|
|
|
|
class _NightlyDownloader:
|
|
"""Returns a fake main-branch commit AND a compare response."""
|
|
|
|
commit_sha = "7777777"
|
|
commit_msg = "test: add nightly feature"
|
|
commit_date = "2026-07-27T12:00:00Z"
|
|
behind_by = 5
|
|
|
|
async def make_request(self, method, url, **kwargs):
|
|
if "/compare/" in url:
|
|
return True, {"behind_by": self.behind_by}
|
|
return True, {
|
|
"sha": self.commit_sha,
|
|
"commit": {
|
|
"message": self.commit_msg,
|
|
"committer": {"date": self.commit_date},
|
|
},
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_nightly_version_parses_behind_by(monkeypatch):
|
|
monkeypatch.setattr(update_routes, "get_downloader", lambda: _stub_downloader(_NightlyDownloader()))
|
|
|
|
version, changelog, behind_by, commit_date = await update_routes.UpdateRoutes._get_nightly_version(
|
|
local_hash="abc1234"
|
|
)
|
|
|
|
assert version == "main-7777777"
|
|
assert behind_by == 5
|
|
assert commit_date == "2026-07-27"
|
|
assert len(changelog) == 1
|
|
assert changelog[0] == "test: add nightly feature"
|
|
|
|
|
|
class _CheckUpdatesDownloader:
|
|
"""Fake downloader returning both a release list and a nightly commit + compare."""
|
|
|
|
commit_sha = "8888888"
|
|
commit_date = "2026-07-28T00:00:00Z"
|
|
|
|
async def make_request(self, method, url, **kwargs):
|
|
if "/releases" in url:
|
|
return True, [
|
|
{
|
|
"tag_name": "v3.0.0",
|
|
"body": "- Feature A\n- Feature B",
|
|
"published_at": "2026-07-20T00:00:00Z",
|
|
}
|
|
]
|
|
if "/compare/" in url:
|
|
return True, {"behind_by": 3}
|
|
return True, {
|
|
"sha": self.commit_sha + "0" * 33,
|
|
"commit": {
|
|
"message": "latest commit",
|
|
"committer": {"date": self.commit_date},
|
|
},
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_updates_nightly_response_includes_behind_and_date(monkeypatch, tmp_path):
|
|
monkeypatch.setattr(update_routes, "get_downloader", lambda: _stub_downloader(_CheckUpdatesDownloader()))
|
|
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes,
|
|
"_get_local_version",
|
|
staticmethod(lambda: "v1.0.0"),
|
|
)
|
|
monkeypatch.setattr(
|
|
update_routes.UpdateRoutes,
|
|
"_get_git_info",
|
|
staticmethod(lambda: {
|
|
"commit_hash": "abc1234",
|
|
"short_hash": "abc1234",
|
|
"branch": "main",
|
|
"commit_date": "2026-01-01",
|
|
}),
|
|
)
|
|
|
|
routes_file = tmp_path / "py" / "routes" / "update_routes.py"
|
|
routes_file.parent.mkdir(parents=True)
|
|
routes_file.write_text("")
|
|
monkeypatch.setattr(update_routes, "__file__", str(routes_file))
|
|
(tmp_path / ".git").mkdir()
|
|
|
|
req = _fake_request(query_params={"nightly": "true"})
|
|
resp = await update_routes.UpdateRoutes.check_updates(req)
|
|
data = _raw_body(resp)
|
|
|
|
assert data["success"] is True
|
|
assert data["nightly"] is True
|
|
assert data["has_git"] is True
|
|
assert data["behind_by"] == 3
|
|
assert data["commit_date"] == "2026-07-28"
|
|
assert data["latest_version"] == "main-8888888"
|
|
assert isinstance(data["releases"], list)
|
|
assert len(data["releases"]) == 1
|
|
assert data["releases"][0]["version"] == "v3.0.0"
|
|
|
|
|
|
def _raw_body(response):
|
|
import json
|
|
return json.loads(response._body.decode())
|