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>
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from prometheus.git.git_repository import GitRepository
|
|
from prometheus.graph.knowledge_graph import KnowledgeGraph
|
|
from prometheus.lang_graph.subgraphs.issue_classification_subgraph import (
|
|
IssueClassificationSubgraph,
|
|
)
|
|
from tests.test_utils.util import FakeListChatWithToolsModel
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_kg():
|
|
kg = Mock(spec=KnowledgeGraph)
|
|
# Configure the mock to return a list of AST node types
|
|
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
|
|
|
|
|
|
def test_issue_classification_subgraph_basic_initialization(mock_kg, mock_git_repo):
|
|
"""Test that IssueClassificationSubgraph initializes correctly with basic components."""
|
|
# Initialize fake model with empty responses
|
|
fake_model = FakeListChatWithToolsModel(responses=[])
|
|
fake_advanced_model = FakeListChatWithToolsModel(responses=[])
|
|
|
|
# Initialize the subgraph with required parameters
|
|
subgraph = IssueClassificationSubgraph(
|
|
advanced_model=fake_advanced_model,
|
|
model=fake_model,
|
|
kg=mock_kg,
|
|
local_path=mock_git_repo.playground_path,
|
|
repository_id=1,
|
|
)
|
|
|
|
# Verify the subgraph was created
|
|
assert subgraph.subgraph is not None
|