101 lines
3 KiB
Python
101 lines
3 KiB
Python
|
"""Tests for the create-agent scaffold (WP-0007 T04)."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from click.testing import CliRunner
|
||
|
|
|
||
|
|
from kaizen_agentic.cli import cli
|
||
|
|
from kaizen_agentic.registry import AgentRegistry
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture
|
||
|
|
def runner() -> CliRunner:
|
||
|
|
return CliRunner()
|
||
|
|
|
||
|
|
|
||
|
|
class TestCreateAgent:
|
||
|
|
def test_scaffold_is_schema_valid_and_loads(
|
||
|
|
self, runner: CliRunner, tmp_path: Path
|
||
|
|
):
|
||
|
|
result = runner.invoke(
|
||
|
|
cli,
|
||
|
|
[
|
||
|
|
"create-agent",
|
||
|
|
"demo-helper",
|
||
|
|
"-c",
|
||
|
|
"testing",
|
||
|
|
"-d",
|
||
|
|
"Demo helper for tests",
|
||
|
|
"--target",
|
||
|
|
str(tmp_path),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
assert result.exit_code == 0, result.output
|
||
|
|
agent_path = tmp_path / "agents" / "agent-demo-helper.md"
|
||
|
|
assert agent_path.exists()
|
||
|
|
|
||
|
|
registry = AgentRegistry(tmp_path / "agents")
|
||
|
|
# Passes the schema and is loadable by the registry.
|
||
|
|
assert registry.validate_frontmatter_schema() == {}
|
||
|
|
agent = registry.get_agent("demo-helper")
|
||
|
|
assert agent is not None
|
||
|
|
assert agent.category.value == "testing"
|
||
|
|
assert agent.memory == "enabled"
|
||
|
|
|
||
|
|
def test_interactive_prompts_when_flags_missing(
|
||
|
|
self, runner: CliRunner, tmp_path: Path
|
||
|
|
):
|
||
|
|
result = runner.invoke(
|
||
|
|
cli,
|
||
|
|
["create-agent", "prompted", "--target", str(tmp_path)],
|
||
|
|
input="testing\nA prompted agent\n",
|
||
|
|
)
|
||
|
|
assert result.exit_code == 0, result.output
|
||
|
|
content = (tmp_path / "agents" / "agent-prompted.md").read_text()
|
||
|
|
assert "category: testing" in content
|
||
|
|
assert "description: A prompted agent" in content
|
||
|
|
|
||
|
|
def test_refuses_overwrite_without_force(self, runner: CliRunner, tmp_path: Path):
|
||
|
|
args = [
|
||
|
|
"create-agent",
|
||
|
|
"dup",
|
||
|
|
"-c",
|
||
|
|
"meta",
|
||
|
|
"-d",
|
||
|
|
"first",
|
||
|
|
"--target",
|
||
|
|
str(tmp_path),
|
||
|
|
]
|
||
|
|
assert runner.invoke(cli, args).exit_code == 0
|
||
|
|
second = runner.invoke(cli, args)
|
||
|
|
assert second.exit_code == 1
|
||
|
|
assert "already exists" in second.output
|
||
|
|
|
||
|
|
def test_force_overwrites(self, runner: CliRunner, tmp_path: Path):
|
||
|
|
base = ["create-agent", "dup", "--target", str(tmp_path)]
|
||
|
|
runner.invoke(cli, base + ["-c", "meta", "-d", "first"])
|
||
|
|
result = runner.invoke(cli, base + ["-c", "meta", "-d", "second", "--force"])
|
||
|
|
assert result.exit_code == 0
|
||
|
|
assert (
|
||
|
|
"description: second" in (tmp_path / "agents" / "agent-dup.md").read_text()
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_rejects_invalid_category(self, runner: CliRunner, tmp_path: Path):
|
||
|
|
result = runner.invoke(
|
||
|
|
cli,
|
||
|
|
[
|
||
|
|
"create-agent",
|
||
|
|
"x",
|
||
|
|
"-c",
|
||
|
|
"nonsense",
|
||
|
|
"-d",
|
||
|
|
"d",
|
||
|
|
"--target",
|
||
|
|
str(tmp_path),
|
||
|
|
],
|
||
|
|
)
|
||
|
|
assert result.exit_code != 0
|