Files
LEDMatrix/test/test_store_install_from_url_rollback.py
T
ChuckandClaude Opus 5.5 c00bf5e8e6 fix(plugin-system): unload/update race, failed-load cleanup, limits validation, schema lookup, install rollback (#653)
* fix(plugin-system): unload/update race, failed-load module cleanup, limits validation, schema lookup, install rollback, op-queue dedupe

- unload_plugin takes the per-plugin lock (5s bounded) before cleanup(),
  and an update() that finishes after its plugin was unloaded no longer
  sets the state back to ENABLED.
- A load that fails after import drops plugin_<id> and its submodules
  and forgets its manager fonts, so a fixed plugin reloads new code.
- Resource limits are validated as non-negative numbers: 400 at
  POST /plugins/limits, bad cached records ignored with one warning.
  Route docstrings note health/metrics reset and limits only change the
  web process's view.
- SchemaManager.get_schema_path resolves each search dir via
  resolve_plugin_dir (manifest id, ledmatrix-<id>) before the literal
  paths; plugins/ still before plugin-repos/. Misses cached 30s and
  logged once at DEBUG.
- install_from_url sets an existing copy aside and restores it if the
  move fails, under the per-plugin reinstall lock.
- Operation queue refuses a second pending op for a plugin and trims
  _operations with history.
- get_vegas_render_width reads display_manager.width first.
- get_logger in store/schema/health/resource/saved_repositories;
  UTF-8 reads in store_manager and state_manager.
- Docs: update_interval precedence (manifest over config) stated where
  users are told to set it in config.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

* fix(web): build the limits 400 message from the field name, not an exception

CodeQL flagged str(e) flowing into the response. invalid_limit_field()
returns the offending field without raising, and limits_from_dict uses it.

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

---------

Co-authored-by: Claude Opus 5.5 <noreply@anthropic.com>
2026-09-28 10:41:40 -04:00

87 lines
2.9 KiB
Python

"""install_from_url keeps the installed copy until the new one is in place.
It deleted the existing plugin directory and then moved the download over it,
with no way back: a move that failed part-way left the user with no plugin at
all. And it did so outside the per-plugin reinstall lock install_plugin()
takes, so two overlapping installs of one id could interleave.
"""
import json
import threading
import pytest
from src.plugin_system import store_manager as store_module
from src.plugin_system.plugin_dirs import BACKUP_MARKER
from src.plugin_system.store_manager import PluginStoreManager
MANIFEST = {
"id": "demo", "name": "Demo", "class_name": "P",
"display_modes": ["demo"], "version": "2.0.0",
}
@pytest.fixture
def store(tmp_path, monkeypatch):
mgr = PluginStoreManager(
plugins_dir=str(tmp_path / "plugins"),
uninstalled_registry_path=str(tmp_path / "uninstalled.json"))
mgr.plugins_dir.mkdir(parents=True, exist_ok=True)
monkeypatch.setattr(mgr, "_install_dependencies", lambda *a, **k: True)
def fake_clone(repo_url, target, branches):
target = type(mgr.plugins_dir)(target)
target.mkdir(parents=True, exist_ok=True)
(target / "manifest.json").write_text(json.dumps(MANIFEST))
(target / "manager.py").write_text("NEW = True\n")
return "main"
monkeypatch.setattr(mgr, "_install_via_git", fake_clone)
old = mgr.plugins_dir / "demo"
old.mkdir()
(old / "manager.py").write_text("OLD = True\n")
return mgr
def _leftover_backups(store):
return [p for p in store.plugins_dir.iterdir() if BACKUP_MARKER in p.name]
def test_failed_move_restores_the_existing_install(store, monkeypatch):
def broken_move(src, dst):
raise OSError("disk full")
monkeypatch.setattr(store_module.shutil, "move", broken_move)
result = store.install_from_url("https://github.com/x/y")
assert result["success"] is False
assert (store.plugins_dir / "demo" / "manager.py").read_text() == "OLD = True\n"
assert _leftover_backups(store) == []
def test_successful_replace_leaves_no_backup(store):
result = store.install_from_url("https://github.com/x/y")
assert result["success"] is True
assert (store.plugins_dir / "demo" / "manager.py").read_text() == "NEW = True\n"
assert _leftover_backups(store) == []
def test_replace_happens_under_the_reinstall_lock(store, monkeypatch):
real_move = store_module.shutil.move
seen = {}
def spying_move(src, dst):
lock = store._get_reinstall_lock("demo")
probe = threading.Thread(
target=lambda: seen.setdefault("free", lock.acquire(blocking=False)))
probe.start()
probe.join()
return real_move(src, dst)
monkeypatch.setattr(store_module.shutil, "move", spying_move)
assert store.install_from_url("https://github.com/x/y")["success"] is True
assert seen == {"free": False}