132 lines
3.5 KiB
Python
132 lines
3.5 KiB
Python
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from rdb_discovery.tasks import generate_agent_prompt, get_next_task
|
|
|
|
|
|
def test_generate_agent_prompt_includes_task_info(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: First goal.\n\n"
|
|
"Acceptance Criteria:\n"
|
|
"- a\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: Second goal.\n\n"
|
|
"Acceptance Criteria:\n"
|
|
"- b\n"
|
|
"- c\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "TASK-002" in prompt
|
|
assert "Second" in prompt
|
|
assert "Todo" in prompt
|
|
assert "Second goal." in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_includes_acceptance_criteria(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: x.\n\n"
|
|
"Acceptance Criteria:\n"
|
|
"- alpha\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: y.\n\n"
|
|
"Acceptance Criteria:\n"
|
|
"- bravo\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "- alpha" not in prompt
|
|
assert "- bravo" in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_includes_read_instructions(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: x.\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: y.\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "README.md" in prompt
|
|
assert "TASKS.md" in prompt
|
|
assert "PROJECT_STATE.md" in prompt
|
|
assert "AGENT_HANDOFF.md" in prompt
|
|
assert "context/agent-guidelines.md" in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_includes_one_task_constraint(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: x.\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: y.\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "ONE task only" in prompt or "one task only" in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_includes_validation(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: x.\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: y.\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "Validation" in prompt
|
|
assert "pytest" in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_includes_reporting(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n"
|
|
"Goal: x.\n\n"
|
|
"## TASK-002 — Second\nStatus: Todo\n\n"
|
|
"Goal: y.\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert "Reporting" in prompt
|
|
|
|
|
|
def test_generate_agent_prompt_no_todo_task(tmp_path: Path) -> None:
|
|
(tmp_path / "TASKS.md").write_text(
|
|
"# TASKS\n\n"
|
|
"## TASK-001 — First\nStatus: Done\n\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert prompt == "No Todo task found."
|
|
|
|
|
|
def test_generate_agent_prompt_no_tasks_file(tmp_path: Path) -> None:
|
|
prompt = generate_agent_prompt(tmp_path)
|
|
|
|
assert prompt == "No Todo task found."
|