wehub-resource-sync
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from opensquilla.agents.registry import AgentRegistry
|
|
from opensquilla.agents.scope import resolve_agent_workspace_dir
|
|
from opensquilla.gateway.config import AgentEntryConfig, GatewayConfig
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_registry_lists_builtin_and_configured_agents() -> None:
|
|
cfg = GatewayConfig(agents=[AgentEntryConfig(id="ops", model="openai/test")])
|
|
registry = AgentRegistry(cfg, persist_changes=False)
|
|
|
|
agents = await registry.list_agents()
|
|
|
|
assert [agent["id"] for agent in agents] == ["main", "ops"]
|
|
assert agents[0]["isBuiltin"] is True
|
|
assert agents[1]["model"] == "openai/test"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_registry_create_and_delete_mutates_config_without_persisting() -> None:
|
|
cfg = GatewayConfig()
|
|
registry = AgentRegistry(cfg, persist_changes=False)
|
|
|
|
created = await registry.create_agent(agent_id="Ops Team", model="openai/test")
|
|
await registry.delete_agent("ops-team")
|
|
|
|
assert created["id"] == "ops-team"
|
|
assert cfg.agents == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_registry_rejects_builtin_main_mutation() -> None:
|
|
registry = AgentRegistry(GatewayConfig(), persist_changes=False)
|
|
|
|
with pytest.raises(ValueError, match="builtin agent"):
|
|
await registry.create_agent(agent_id="main")
|
|
|
|
|
|
def test_resolve_agent_workspace_dir_uses_configured_agent_workspace(tmp_path) -> None:
|
|
root = tmp_path / "root"
|
|
agent_workspace = tmp_path / "ops-workspace"
|
|
cfg = GatewayConfig(
|
|
workspace_dir=str(root),
|
|
agents=[AgentEntryConfig(id="ops", workspace=str(agent_workspace))],
|
|
)
|
|
|
|
assert resolve_agent_workspace_dir("ops", cfg) == agent_workspace
|
|
assert resolve_agent_workspace_dir("main", cfg) == root
|