refactor(tests): Add strict type hints to third batch of test files
This commit is contained in:
@@ -1,17 +1,18 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
import json
|
||||
from typing import Any
|
||||
import orchestrator_pm
|
||||
import conductor_tech_lead
|
||||
import multi_agent_conductor
|
||||
from models import Track, Ticket
|
||||
|
||||
@pytest.fixture
|
||||
def mock_ai_client() -> None:
|
||||
def mock_ai_client() -> Any:
|
||||
with patch("ai_client.send") as mock_send:
|
||||
yield mock_send
|
||||
|
||||
def test_generate_tracks(mock_ai_client):
|
||||
def test_generate_tracks(mock_ai_client: Any) -> None:
|
||||
# Tier 1 (PM) response mock
|
||||
mock_ai_client.return_value = json.dumps([
|
||||
{"id": "track_1", "title": "Infrastructure Setup", "description": "Setup basic project structure"},
|
||||
@@ -26,8 +27,7 @@ def test_generate_tracks(mock_ai_client):
|
||||
assert tracks[1]["id"] == "track_2"
|
||||
mock_ai_client.assert_called_once()
|
||||
|
||||
def test_generate_tickets(mock_ai_client):
|
||||
# Tier 2 (Tech Lead) response mock
|
||||
def test_generate_tickets(mock_ai_client: Any) -> None:
|
||||
mock_ai_client.return_value = json.dumps([
|
||||
{"id": "T-001", "description": "Define interfaces", "depends_on": []},
|
||||
{"id": "T-002", "description": "Implement interfaces", "depends_on": ["T-001"]}
|
||||
@@ -102,7 +102,7 @@ def test_conductor_engine_parse_json_tickets() -> None:
|
||||
assert track.tickets[1].id == "T2"
|
||||
assert track.tickets[1].depends_on == ["T1"]
|
||||
|
||||
def test_run_worker_lifecycle_blocked(mock_ai_client):
|
||||
def test_run_worker_lifecycle_blocked(mock_ai_client: Any) -> None:
|
||||
ticket = Ticket(id="T1", description="desc", status="todo", assigned_to="user")
|
||||
context = multi_agent_conductor.WorkerContext(ticket_id="T1", model_name="model", messages=[])
|
||||
mock_ai_client.return_value = "BLOCKED because of missing info"
|
||||
|
||||
Reference in New Issue
Block a user