b52318eeae
- Add intelligent-router.sh hook for automatic agent routing - Add AUTO-TRIGGER-SUMMARY.md documentation - Add FINAL-INTEGRATION-SUMMARY.md documentation - Complete Prometheus integration (6 commands + 4 tools) - Complete Dexto integration (12 commands + 5 tools) - Enhanced Ralph with access to all agents - Fix /clawd command (removed disable-model-invocation) - Update hooks.json to v5 with intelligent routing - 291 total skills now available - All 21 commands with automatic routing 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
from prometheus.lang_graph.nodes.edit_message_node import EditMessageNode
|
|
from prometheus.models.context import Context
|
|
|
|
|
|
@pytest.fixture
|
|
def edit_node():
|
|
return EditMessageNode(
|
|
context_key="bug_fix_context", analyzer_message_key="issue_bug_analyzer_messages"
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def base_state():
|
|
return {
|
|
"issue_title": "Test Bug",
|
|
"issue_body": "This is a test bug description",
|
|
"issue_comments": [
|
|
{"username": "user1", "comment": "Comment 1"},
|
|
{"username": "user2", "comment": "Comment 2"},
|
|
],
|
|
"bug_fix_context": [
|
|
Context(
|
|
relative_path="foobar.py",
|
|
content="# Context 1",
|
|
start_line_number=1,
|
|
end_line_number=1,
|
|
)
|
|
],
|
|
"issue_bug_analyzer_messages": ["Analysis message"],
|
|
}
|
|
|
|
|
|
def test_first_message_formatting(edit_node, base_state):
|
|
# Using context managers for patching
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.format_issue_info"
|
|
) as mock_format_issue:
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.get_last_message_content"
|
|
) as mock_last_message:
|
|
mock_format_issue.return_value = "Formatted Issue Info"
|
|
mock_last_message.return_value = "Last Analysis Message"
|
|
|
|
result = edit_node(base_state)
|
|
|
|
assert isinstance(result, dict)
|
|
assert "edit_messages" in result
|
|
assert len(result["edit_messages"]) == 1
|
|
assert isinstance(result["edit_messages"][0], HumanMessage)
|
|
|
|
message_content = result["edit_messages"][0].content
|
|
assert "Formatted Issue Info" in message_content
|
|
assert "# Context 1" in message_content
|
|
assert "Last Analysis Message" in message_content
|
|
|
|
|
|
def test_followup_message_with_build_fail(edit_node, base_state):
|
|
# Add build failure to state
|
|
base_state["build_fail_log"] = "Build failed: error in compilation"
|
|
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.get_last_message_content"
|
|
) as mock_last_message:
|
|
mock_last_message.return_value = "Last Analysis Message"
|
|
|
|
result = edit_node(base_state)
|
|
message_content = result["edit_messages"][0].content
|
|
|
|
assert "Build failed: error in compilation" in message_content
|
|
assert "Please implement these revised changes carefully" in message_content
|
|
|
|
|
|
def test_followup_message_with_test_fail(edit_node, base_state):
|
|
# Add test failure to state
|
|
base_state["reproducing_test_fail_log"] = "Test failed: assertion error"
|
|
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.get_last_message_content"
|
|
) as mock_last_message:
|
|
mock_last_message.return_value = "Last Analysis Message"
|
|
|
|
result = edit_node(base_state)
|
|
message_content = result["edit_messages"][0].content
|
|
|
|
assert "Test failed: assertion error" in message_content
|
|
assert "Please implement these revised changes carefully" in message_content
|
|
|
|
|
|
def test_followup_message_with_existing_test_fail(edit_node, base_state):
|
|
# Add existing test failure to state
|
|
base_state["existing_test_fail_log"] = "Existing test failed"
|
|
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.get_last_message_content"
|
|
) as mock_last_message:
|
|
mock_last_message.return_value = "Last Analysis Message"
|
|
|
|
result = edit_node(base_state)
|
|
message_content = result["edit_messages"][0].content
|
|
|
|
assert "Existing test failed" in message_content
|
|
assert "Please implement these revised changes carefully" in message_content
|
|
|
|
|
|
def test_error_priority(edit_node, base_state):
|
|
# Add multiple error types to test priority handling
|
|
base_state["reproducing_test_fail_log"] = "Test failed"
|
|
base_state["build_fail_log"] = "Build failed"
|
|
base_state["existing_test_fail_log"] = "Existing test failed"
|
|
|
|
with patch(
|
|
"prometheus.lang_graph.nodes.edit_message_node.get_last_message_content"
|
|
) as mock_last_message:
|
|
mock_last_message.return_value = "Last Analysis Message"
|
|
|
|
result = edit_node(base_state)
|
|
message_content = result["edit_messages"][0].content
|
|
|
|
# Should prioritize reproducing test failure
|
|
assert "Test failed" in message_content
|
|
assert "Build failed" not in message_content
|
|
assert "Existing test failed" not in message_content
|