chore: restore original directory structure (project under code-review-graph-main/)

code-review-graph-main/tests/test_prompts.py

@@ -0,0 +1,299 @@
+"""Tests for MCP prompt templates."""
+
+from fastmcp.prompts.prompt import Message
+
+from code_review_graph.prompts import (
+    architecture_map_prompt,
+    debug_issue_prompt,
+    onboard_developer_prompt,
+    pre_merge_check_prompt,
+    project_review_prompt,
+    review_changes_prompt,
+    unified_review_prompt,
+)
+
+
+def _text(msg: Message) -> str:
+    """Extract the text content from a fastmcp Message."""
+    return msg.content.text
+
+
+class TestReviewChangesPrompt:
+    def test_returns_list_with_messages(self):
+        result = review_changes_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = review_changes_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_default_base(self):
+        result = review_changes_prompt()
+        assert "HEAD~1" in _text(result[0])
+
+    def test_custom_base(self):
+        result = review_changes_prompt(base="main")
+        assert "main" in _text(result[0])
+
+    def test_mentions_detect_changes(self):
+        result = review_changes_prompt()
+        assert "detect_changes" in _text(result[0])
+
+    def test_mentions_affected_flows(self):
+        result = review_changes_prompt()
+        assert "affected_flows" in _text(result[0])
+
+    def test_mentions_test_gaps(self):
+        result = review_changes_prompt()
+        assert "test" in _text(result[0]).lower()
+
+
+class TestArchitectureMapPrompt:
+    def test_returns_list_with_messages(self):
+        result = architecture_map_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = architecture_map_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_mentions_communities(self):
+        result = architecture_map_prompt()
+        assert "communities" in _text(result[0]).lower()
+
+    def test_mentions_mermaid(self):
+        result = architecture_map_prompt()
+        assert "Mermaid" in _text(result[0])
+
+
+class TestDebugIssuePrompt:
+    def test_returns_list_with_messages(self):
+        result = debug_issue_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = debug_issue_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_includes_description(self):
+        result = debug_issue_prompt(description="login fails with 500 error")
+        assert "login fails with 500 error" in _text(result[0])
+
+    def test_empty_description(self):
+        result = debug_issue_prompt()
+        content = _text(result[0])
+        assert "debug" in content.lower()
+
+    def test_mentions_search(self):
+        result = debug_issue_prompt(description="test issue")
+        assert "semantic_search_nodes" in _text(result[0])
+
+    def test_mentions_get_minimal_context(self):
+        result = debug_issue_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+
+class TestOnboardDeveloperPrompt:
+    def test_returns_list_with_messages(self):
+        result = onboard_developer_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = onboard_developer_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_mentions_stats(self):
+        result = onboard_developer_prompt()
+        assert "list_graph_stats" in _text(result[0])
+
+    def test_mentions_architecture(self):
+        result = onboard_developer_prompt()
+        assert "architecture" in _text(result[0]).lower()
+
+    def test_mentions_critical_flows(self):
+        result = onboard_developer_prompt()
+        assert "critical" in _text(result[0]).lower()
+
+
+class TestPreMergeCheckPrompt:
+    def test_returns_list_with_messages(self):
+        result = pre_merge_check_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = pre_merge_check_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_default_base(self):
+        result = pre_merge_check_prompt()
+        # The pre-merge prompt is now generic (doesn't embed the base ref)
+        assert "pre-merge" in _text(result[0]).lower()
+
+    def test_custom_base(self):
+        # pre_merge_check_prompt still accepts base but the workflow
+        # is now generic — just verify it returns valid prompt
+        result = pre_merge_check_prompt(base="develop")
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_mentions_risk_scoring(self):
+        result = pre_merge_check_prompt()
+        assert "risk" in _text(result[0]).lower()
+
+    def test_mentions_test_gaps(self):
+        result = pre_merge_check_prompt()
+        assert "tests_for" in _text(result[0])
+
+    def test_mentions_dead_code(self):
+        result = pre_merge_check_prompt()
+        assert "dead_code" in _text(result[0])
+
+
+class TestUnifiedReviewPrompt:
+    def test_returns_list_with_messages(self):
+        result = unified_review_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = unified_review_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_default_base_and_tier(self):
+        result = unified_review_prompt()
+        text = _text(result[0])
+        assert "HEAD~1" in text
+        assert "Standard tier" in text
+
+    def test_custom_base(self):
+        result = unified_review_prompt(base="develop")
+        text = _text(result[0])
+        assert "base=develop" in text
+
+    def test_mentions_score_review(self):
+        result = unified_review_prompt()
+        assert "score_review" in _text(result[0])
+
+    def test_mentions_dedupe_findings(self):
+        result = unified_review_prompt()
+        assert "dedupe_findings" in _text(result[0])
+
+    def test_mentions_generate_report(self):
+        result = unified_review_prompt()
+        assert "generate_report" in _text(result[0])
+
+    def test_mentions_read_only(self):
+        result = unified_review_prompt()
+        assert "READ-ONLY" in _text(result[0])
+
+    def test_mentions_blocker_rule(self):
+        result = unified_review_prompt()
+        assert "FAIL" in _text(result[0])
+
+
+class TestProjectReviewPrompt:
+    def test_returns_list_with_messages(self):
+        result = project_review_prompt()
+        assert isinstance(result, list)
+        assert len(result) >= 1
+
+    def test_message_has_role_and_content(self):
+        result = project_review_prompt()
+        for msg in result:
+            assert isinstance(msg, Message)
+            assert msg.role == "user"
+            assert _text(msg)
+
+    def test_default_whole_project_scope(self):
+        result = project_review_prompt()
+        text = _text(result[0])
+        assert "whole-project" in text
+        assert "all_files=True" in text
+
+    def test_feature_scope_with_target(self):
+        result = project_review_prompt(scope="feature", target="payment")
+        text = _text(result[0])
+        assert "scope=feature" in text
+        assert "target=payment" in text
+
+    def test_mentions_architecture_scan(self):
+        result = project_review_prompt()
+        text = _text(result[0])
+        assert "get_architecture_overview" in text
+        assert "list_communities" in text
+
+    def test_mentions_high_risk_scan(self):
+        result = project_review_prompt()
+        text = _text(result[0])
+        assert "get_knowledge_gaps" in text
+        assert "get_hub_nodes" in text
+
+    def test_feature_mentions_semantic_search(self):
+        result = project_review_prompt(scope="feature", target="auth")
+        text = _text(result[0])
+        assert "semantic_search_nodes" in text
+        assert "get_impact_radius" in text
+
+    def test_mentions_read_only(self):
+        result = project_review_prompt()
+        assert "READ-ONLY" in _text(result[0])
+
+    def test_mentions_generate_report(self):
+        result = project_review_prompt()
+        assert "generate_report" in _text(result[0])
+
+
+class TestTokenEfficiencyPreamble:
+    """All prompts should include the token efficiency preamble."""
+
+    def test_review_has_preamble(self):
+        result = review_changes_prompt()
+        assert "get_minimal_context" in _text(result[0])
+        assert "detail_level" in _text(result[0])
+
+    def test_architecture_has_preamble(self):
+        result = architecture_map_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+    def test_debug_has_preamble(self):
+        result = debug_issue_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+    def test_onboard_has_preamble(self):
+        result = onboard_developer_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+    def test_pre_merge_has_preamble(self):
+        result = pre_merge_check_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+    def test_unified_review_has_preamble(self):
+        result = unified_review_prompt()
+        assert "get_minimal_context" in _text(result[0])
+
+    def test_project_review_has_preamble(self):
+        result = project_review_prompt()
+        assert "get_minimal_context" in _text(result[0])