| """Unit tests for hierarchical orchestration middleware.""" |
|
|
| from unittest.mock import AsyncMock |
|
|
| import pytest |
|
|
| from src.middleware.sub_iteration import SubIterationMiddleware |
| from src.utils.models import AssessmentDetails, JudgeAssessment |
|
|
| pytestmark = pytest.mark.unit |
|
|
|
|
| @pytest.mark.asyncio |
| async def test_sub_iteration_middleware(): |
| team = AsyncMock() |
| team.execute.return_value = "Result" |
|
|
| judge = AsyncMock() |
| judge.assess.return_value = JudgeAssessment( |
| details=AssessmentDetails( |
| mechanism_score=10, |
| mechanism_reasoning="Good reasoning text here", |
| clinical_evidence_score=10, |
| clinical_reasoning="Good reasoning text here", |
| drug_candidates=[], |
| key_findings=[], |
| ), |
| sufficient=True, |
| confidence=1.0, |
| recommendation="synthesize", |
| next_search_queries=[], |
| reasoning="Good reasoning text here for the overall assessment which must be long enough.", |
| ) |
|
|
| middleware = SubIterationMiddleware(team, judge) |
| result, assessment = await middleware.run("task") |
|
|
| assert result == "Result" |
| assert assessment.sufficient |
| assert team.execute.call_count == 1 |
|
|
