Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

tests/hermes_cli/test_sibling_config_migration.py

@@ -0,0 +1,124 @@
+"""Fleet-wide config migration (#91277 Phase 2 — #20438/#54926/#79048 class).
+
+`hermes update` migrated only the active profile's config.yaml; sibling
+profiles silently drifted config versions until their gateway hit a config
+the new code couldn't read. `_migrate_sibling_profile_configs()` runs the
+same non-interactive safe migration for every sibling home, scoped via the
+context-local HERMES_HOME override.
+
+These tests use REAL config files on disk and the REAL migration pipeline —
+only the profile-root location is pointed at tmp_path.
+"""
+
+import yaml
+from pathlib import Path
+
+import hermes_cli.update_cmd as update_cmd
+
+
+def _write_profile(root: Path, name: str, version: int) -> Path:
+    home = root / name
+    home.mkdir(parents=True)
+    (home / "config.yaml").write_text(
+        yaml.safe_dump({"_config_version": version, "model": {"provider": "nous"}}),
+        encoding="utf-8",
+    )
+    return home
+
+
+def _latest_version() -> int:
+    from hermes_cli.config import DEFAULT_CONFIG
+
+    return int(DEFAULT_CONFIG["_config_version"])
+
+
+def _setup(monkeypatch, tmp_path, active_home: Path):
+    import hermes_cli.profiles as profiles_mod
+
+    monkeypatch.setattr(profiles_mod, "_get_profiles_root", lambda: tmp_path / "profiles")
+    import hermes_constants
+
+    monkeypatch.setattr(
+        hermes_constants, "get_process_hermes_home", lambda: active_home
+    )
+    monkeypatch.setattr(
+        update_cmd, "_reload_config_modules", lambda: None
+    )  # module reload is orthogonal here; the real one re-imports from disk
+
+
+def test_sibling_behind_is_migrated_on_disk(monkeypatch, tmp_path):
+    active = _write_profile(tmp_path / "profiles", "active", _latest_version())
+    sibling = _write_profile(tmp_path / "profiles", "research", 12)
+    _setup(monkeypatch, tmp_path, active)
+
+    migrated = update_cmd._migrate_sibling_profile_configs()
+
+    names = [m[0] for m in migrated]
+    assert "research" in names
+    entry = next(m for m in migrated if m[0] == "research")
+    assert entry[1] == 12 and entry[2] == _latest_version()
+    # REAL file on disk carries the new version
+    on_disk = yaml.safe_load((sibling / "config.yaml").read_text())
+    assert on_disk["_config_version"] == _latest_version()
+    # and user settings survived
+    assert on_disk["model"]["provider"] == "nous"
+
+
+def test_active_profile_is_skipped(monkeypatch, tmp_path):
+    active = _write_profile(tmp_path / "profiles", "active", 12)
+    _setup(monkeypatch, tmp_path, active)
+
+    migrated = update_cmd._migrate_sibling_profile_configs()
+
+    assert "active" not in [m[0] for m in migrated]
+    # active home untouched (the caller's own migration handles it)
+    on_disk = yaml.safe_load((active / "config.yaml").read_text())
+    assert on_disk["_config_version"] == 12
+
+
+def test_current_sibling_untouched(monkeypatch, tmp_path):
+    active = _write_profile(tmp_path / "profiles", "active", _latest_version())
+    sibling = _write_profile(tmp_path / "profiles", "work", _latest_version())
+    _setup(monkeypatch, tmp_path, active)
+    before = (sibling / "config.yaml").read_bytes()
+
+    migrated = update_cmd._migrate_sibling_profile_configs()
+
+    assert migrated == []
+    assert (sibling / "config.yaml").read_bytes() == before
+
+
+def test_unconfigured_profile_skipped(monkeypatch, tmp_path):
+    active = _write_profile(tmp_path / "profiles", "active", _latest_version())
+    bare = tmp_path / "profiles" / "empty"
+    bare.mkdir()
+    _setup(monkeypatch, tmp_path, active)
+
+    assert update_cmd._migrate_sibling_profile_configs() == []
+    assert not (bare / "config.yaml").exists()
+
+
+def test_one_broken_profile_does_not_block_others(monkeypatch, tmp_path):
+    active = _write_profile(tmp_path / "profiles", "active", _latest_version())
+    broken_home = tmp_path / "profiles" / "broken"
+    broken_home.mkdir()
+    (broken_home / "config.yaml").write_text(":\nnot yaml: [", encoding="utf-8")
+    _write_profile(tmp_path / "profiles", "healthy", 12)
+    _setup(monkeypatch, tmp_path, active)
+
+    migrated = update_cmd._migrate_sibling_profile_configs()
+
+    assert [m[0] for m in migrated] == ["healthy"]
+
+
+def test_override_is_reset_after_run(monkeypatch, tmp_path):
+    """The ContextVar override must not leak past the sweep."""
+    from hermes_constants import get_hermes_home_override
+
+    active = _write_profile(tmp_path / "profiles", "active", _latest_version())
+    _write_profile(tmp_path / "profiles", "research", 12)
+    _setup(monkeypatch, tmp_path, active)
+
+    before = get_hermes_home_override()
+    update_cmd._migrate_sibling_profile_configs()
+    assert get_hermes_home_override() == before