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>
61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from langchain_core.language_models.chat_models import BaseChatModel
|
|
|
|
from prometheus.docker.base_container import BaseContainer
|
|
from prometheus.git.git_repository import GitRepository
|
|
from prometheus.graph.knowledge_graph import KnowledgeGraph
|
|
from prometheus.lang_graph.graphs.issue_graph import IssueGraph
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_advanced_model():
|
|
return Mock(spec=BaseChatModel)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_base_model():
|
|
return Mock(spec=BaseChatModel)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_kg():
|
|
kg = Mock(spec=KnowledgeGraph)
|
|
kg.get_all_ast_node_types.return_value = ["FunctionDef", "ClassDef", "Module", "Import", "Call"]
|
|
kg.root_node_id = 0
|
|
return kg
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_git_repo():
|
|
git_repo = Mock(spec=GitRepository)
|
|
git_repo.playground_path = "mock/playground/path"
|
|
return git_repo
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_container():
|
|
return Mock(spec=BaseContainer)
|
|
|
|
|
|
def test_issue_graph_basic_initialization(
|
|
mock_advanced_model,
|
|
mock_base_model,
|
|
mock_kg,
|
|
mock_git_repo,
|
|
mock_container,
|
|
):
|
|
"""Test that IssueGraph initializes correctly with basic components."""
|
|
graph = IssueGraph(
|
|
advanced_model=mock_advanced_model,
|
|
base_model=mock_base_model,
|
|
kg=mock_kg,
|
|
git_repo=mock_git_repo,
|
|
container=mock_container,
|
|
repository_id=1,
|
|
)
|
|
|
|
assert graph.graph is not None
|
|
assert graph.git_repo == mock_git_repo
|