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>
76 lines
2.4 KiB
Python
76 lines
2.4 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from langchain_core.messages import HumanMessage
|
|
|
|
from prometheus.docker.base_container import BaseContainer
|
|
from prometheus.graph.knowledge_graph import KnowledgeGraph
|
|
from prometheus.lang_graph.nodes.general_test_node import GeneralTestNode
|
|
from prometheus.lang_graph.subgraphs.build_and_test_state import BuildAndTestState
|
|
from tests.test_utils.util import FakeListChatWithToolsModel
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_container():
|
|
return Mock(spec=BaseContainer)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_kg():
|
|
kg = Mock(spec=KnowledgeGraph)
|
|
kg.get_file_tree.return_value = "./\n├── tests/\n│ └── test_main.py"
|
|
return kg
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_llm():
|
|
return FakeListChatWithToolsModel(responses=["Tests executed successfully"])
|
|
|
|
|
|
@pytest.fixture
|
|
def basic_state():
|
|
return BuildAndTestState(
|
|
{"exist_test": True, "test_messages": [], "test_command_summary": None}
|
|
)
|
|
|
|
|
|
def test_format_human_message(mock_container, mock_kg, fake_llm, basic_state):
|
|
"""Test basic message formatting."""
|
|
node = GeneralTestNode(fake_llm, mock_container, mock_kg)
|
|
message = node.format_human_message(basic_state)
|
|
|
|
assert isinstance(message, HumanMessage)
|
|
assert mock_kg.get_file_tree() in message.content
|
|
|
|
|
|
def test_call_with_no_tests(mock_container, mock_kg, fake_llm):
|
|
"""Test behavior when no tests exist."""
|
|
node = GeneralTestNode(fake_llm, mock_container, mock_kg)
|
|
state = BuildAndTestState({"exist_test": False})
|
|
|
|
result = node(state)
|
|
|
|
assert "build_messages" in result
|
|
assert "no test framework" in result["build_messages"][0].content
|
|
|
|
|
|
def test_call_normal_execution(mock_container, mock_kg, fake_llm, basic_state):
|
|
"""Test normal execution flow."""
|
|
node = GeneralTestNode(fake_llm, mock_container, mock_kg)
|
|
|
|
result = node(basic_state)
|
|
|
|
assert "test_messages" in result
|
|
assert len(result["test_messages"]) == 1
|
|
assert result["test_messages"][0].content == "Tests executed successfully"
|
|
|
|
|
|
def test_format_human_message_with_summary(mock_container, mock_kg, fake_llm):
|
|
"""Test message formatting with test summary."""
|
|
node = GeneralTestNode(fake_llm, mock_container, mock_kg)
|
|
state = BuildAndTestState({"test_command_summary": "Previous test used pytest"})
|
|
|
|
message = node.format_human_message(state)
|
|
|
|
assert "Previous test used pytest" in message.content
|