"""Tests for the heavy plugin (Organization) runtime."""

import json
from pathlib import Path

from kilostar.plugin_runtime.manifest import OrgManifest
from kilostar.plugin_runtime.agents_config import AgentsConfig
from kilostar.plugin_runtime.loader import discover_plugins, load_plugin
from kilostar.plugin_runtime.tool_bridge import make_dispatch_tool

_PLUGIN_ROOT = Path(__file__).parent.parent.parent / "data" / "plugin"
_EXAMPLE_DEPT = _PLUGIN_ROOT / "example_dept"


def test_example_dept_structure():
    """example_dept stub has the required files."""
    assert (_EXAMPLE_DEPT / "manifest.json").exists()
    assert (_EXAMPLE_DEPT / "agents.json").exists()
    assert (_EXAMPLE_DEPT / "README.md").exists()
    assert (_EXAMPLE_DEPT / "core" / "organization.py").exists()


def test_example_dept_manifest_valid():
    with open(_EXAMPLE_DEPT / "manifest.json", "r", encoding="utf-8") as f:
        data = json.load(f)
    manifest = OrgManifest.model_validate(data)
    assert manifest.name == "example_dept"
    assert manifest.actor_name == "org_example_dept"
    assert manifest.entry == "core.organization:ExampleOrganization"


def test_example_dept_agents_valid():
    with open(_EXAMPLE_DEPT / "agents.json", "r", encoding="utf-8") as f:
        data = json.load(f)
    config = AgentsConfig.model_validate(data)
    assert len(config.agents) == 2
    names = {a.name for a in config.agents}
    assert names == {"analyst", "executor"}
    assert config.orchestration.entry == "analyst"
    analyst = config.get("analyst")
    assert analyst is not None
    assert "executor" in analyst.peers


def test_discover_plugins_finds_example_dept():
    plugins = discover_plugins(_PLUGIN_ROOT)
    names = {p.name for p in plugins}
    assert "example_dept" in names


def test_load_plugin_returns_class():
    cls, manifest_dict, agents_dict, dir_str = load_plugin(_EXAMPLE_DEPT)
    assert cls.__name__ == "ExampleOrganization"
    assert manifest_dict["name"] == "example_dept"
    assert "agents" in agents_dict
    assert dir_str == str(_EXAMPLE_DEPT)


def test_make_dispatch_tool_signature():
    tool = make_dispatch_tool("example_dept", "示例部门", "演示用")
    assert tool.__name__ == "dispatch_to_example_dept"
    assert callable(tool)
    assert "演示用" in tool.__doc__