generated from coulomb/repo-seed
114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from statehub_register import (
|
|
RegisterInference,
|
|
_invoke_llm,
|
|
_normalise_inference,
|
|
_parse_json_object,
|
|
write_registration_files,
|
|
)
|
|
|
|
|
|
def test_parse_json_object_accepts_fenced_json():
|
|
parsed = _parse_json_object(
|
|
"""Here is the answer:
|
|
|
|
```json
|
|
{"repo_slug": "demo-repo", "in_scope": ["tests"]}
|
|
```
|
|
"""
|
|
)
|
|
|
|
assert parsed == {"repo_slug": "demo-repo", "in_scope": ["tests"]}
|
|
|
|
|
|
def test_normalise_inference_slugifies_and_normalises_prefix():
|
|
inference = _normalise_inference(
|
|
{
|
|
"project_description": "Provides demo automation.",
|
|
"domain_slug": "Custodian",
|
|
"repo_slug": "Demo Repo!",
|
|
"workplan_prefix": "demo",
|
|
"in_scope": ["one", "two"],
|
|
"out_of_scope": ["three"],
|
|
}
|
|
)
|
|
|
|
assert inference.domain_slug == "custodian"
|
|
assert inference.repo_slug == "demo-repo"
|
|
assert inference.workplan_prefix == "DEMO-WP"
|
|
assert inference.in_scope == ["one", "two"]
|
|
assert inference.out_of_scope == ["three"]
|
|
|
|
|
|
def test_invoke_llm_uses_mock_adapter(monkeypatch):
|
|
payload = {"project_description": "Mocked repo.", "repo_slug": "mocked-repo"}
|
|
monkeypatch.setenv("STATEHUB_REGISTER_MOCK_LLM_RESPONSE", json.dumps(payload))
|
|
args = argparse.Namespace(
|
|
llm_provider="mock",
|
|
llm_model=None,
|
|
llm_api_key=None,
|
|
llm_timeout=5,
|
|
)
|
|
|
|
assert json.loads(_invoke_llm("infer this repo", args)) == payload
|
|
|
|
|
|
def test_write_registration_files_primes_codex_repo(tmp_path: Path):
|
|
inference = RegisterInference(
|
|
in_scope=["Run the demo service."],
|
|
out_of_scope=["Operate unrelated services."],
|
|
current_state="Status: active; implementation is small.",
|
|
)
|
|
|
|
written = write_registration_files(
|
|
project_path=tmp_path,
|
|
project_name="demo-service",
|
|
project_description="Provides a demo service.",
|
|
domain="custodian",
|
|
topic_id="cee7bedf-2b48-46ef-8601-006474f2ad7a",
|
|
topic_slug="custodian",
|
|
repo_slug="demo-service",
|
|
wp_prefix="DEMO-WP",
|
|
intent_markdown="# INTENT\n\nDemo service intent.\n",
|
|
inference=inference,
|
|
)
|
|
|
|
assert {path.name for path in written} == {
|
|
"INTENT.md",
|
|
"SCOPE.md",
|
|
"AGENTS.md",
|
|
".custodian-brief.md",
|
|
"DEMO-WP-0001-statehub-bootstrap.md",
|
|
}
|
|
assert (tmp_path / "INTENT.md").read_text() == "# INTENT\n\nDemo service intent.\n"
|
|
assert "**Repo slug:** demo-service" in (tmp_path / "AGENTS.md").read_text()
|
|
assert "Run the demo service." in (tmp_path / "SCOPE.md").read_text()
|
|
workplan = (tmp_path / "workplans" / "DEMO-WP-0001-statehub-bootstrap.md").read_text()
|
|
assert "id: DEMO-WP-0001" in workplan
|
|
assert "id: DEMO-WP-0001-T01" in workplan
|
|
assert "make fix-consistency REPO=demo-service" in workplan
|
|
|
|
|
|
def test_write_registration_files_is_idempotent_without_force(tmp_path: Path):
|
|
inference = RegisterInference()
|
|
kwargs = {
|
|
"project_path": tmp_path,
|
|
"project_name": "demo",
|
|
"project_description": "Provides a demo.",
|
|
"domain": "custodian",
|
|
"topic_id": "topic",
|
|
"topic_slug": "custodian",
|
|
"repo_slug": "demo",
|
|
"wp_prefix": "DEMO-WP",
|
|
"intent_markdown": "# INTENT\n\nDemo.\n",
|
|
"inference": inference,
|
|
}
|
|
|
|
assert write_registration_files(**kwargs)
|
|
assert write_registration_files(**kwargs) == []
|