wehub-resource-sync
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""Tests for managed Pi agent dir seeding (extensions / packages)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from omnigent.inner.pi_settings import prepare_managed_pi_agent_dir
|
|
|
|
|
|
def test_prepare_managed_pi_agent_dir_copies_settings_and_symlinks_npm(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Gateway managed dir gets global settings metadata and npm install tree."""
|
|
global_agent = tmp_path / "global-agent"
|
|
global_agent.mkdir()
|
|
(global_agent / "settings.json").write_text(
|
|
json.dumps(
|
|
{
|
|
"extensions": ["/tmp/my-ext.ts"],
|
|
"packages": ["npm:@foo/bar"],
|
|
}
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
npm_dir = global_agent / "npm" / "foo"
|
|
npm_dir.mkdir(parents=True)
|
|
|
|
managed = tmp_path / "managed"
|
|
managed.mkdir()
|
|
(managed / "models.json").write_text("{}", encoding="utf-8")
|
|
|
|
prepare_managed_pi_agent_dir(
|
|
managed,
|
|
overlay={"retry": {"maxRetries": 5}},
|
|
global_agent_dir=global_agent,
|
|
)
|
|
|
|
written = json.loads((managed / "settings.json").read_text(encoding="utf-8"))
|
|
assert written["extensions"] == ["/tmp/my-ext.ts"]
|
|
assert written["packages"] == ["npm:@foo/bar"]
|
|
assert written["retry"] == {"maxRetries": 5}
|
|
assert (managed / "npm").is_symlink()
|
|
assert (managed / "npm").resolve() == (global_agent / "npm").resolve()
|
|
|
|
|
|
def test_prepare_managed_pi_agent_dir_empty_global_writes_overlay_only(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
"""Missing global settings still writes the Omnigent overlay."""
|
|
global_agent = tmp_path / "empty-global"
|
|
global_agent.mkdir()
|
|
managed = tmp_path / "managed"
|
|
managed.mkdir()
|
|
|
|
prepare_managed_pi_agent_dir(
|
|
managed,
|
|
overlay={"retry": {"enabled": True}},
|
|
global_agent_dir=global_agent,
|
|
)
|
|
|
|
written = json.loads((managed / "settings.json").read_text(encoding="utf-8"))
|
|
assert written == {"retry": {"enabled": True}}
|
|
|
|
|
|
def test_prepare_managed_pi_agent_dir_deep_merges_nested_overlay(tmp_path: Path) -> None:
|
|
"""Nested settings (e.g. compaction) merge like Pi project overrides."""
|
|
global_agent = tmp_path / "global-agent"
|
|
global_agent.mkdir()
|
|
(global_agent / "settings.json").write_text(
|
|
json.dumps({"compaction": {"enabled": True, "reserveTokens": 16384}}),
|
|
encoding="utf-8",
|
|
)
|
|
managed = tmp_path / "managed"
|
|
managed.mkdir()
|
|
|
|
prepare_managed_pi_agent_dir(
|
|
managed,
|
|
overlay={"compaction": {"reserveTokens": 8192}},
|
|
global_agent_dir=global_agent,
|
|
)
|
|
|
|
written = json.loads((managed / "settings.json").read_text(encoding="utf-8"))
|
|
assert written["compaction"] == {
|
|
"enabled": True,
|
|
"reserveTokens": 8192,
|
|
}
|