File size: 1,270 Bytes
fd1472e cd7c282 fd1472e cd7c282 fd1472e cd7c282 fd1472e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
"""Tests for Orchestrator Factory domain support."""
from unittest.mock import MagicMock, patch
from src.config.domain import ResearchDomain
from src.orchestrators.factory import create_orchestrator
class TestFactoryDomain:
@patch("src.orchestrators.factory._get_advanced_orchestrator_class")
def test_create_simple_maps_to_advanced_with_domain(self, mock_get_cls):
mock_adv_cls = MagicMock()
mock_get_cls.return_value = mock_adv_cls
mock_search = MagicMock()
mock_judge = MagicMock()
create_orchestrator(
search_handler=mock_search,
judge_handler=mock_judge,
mode="simple",
domain=ResearchDomain.SEXUAL_HEALTH,
)
call_kwargs = mock_adv_cls.call_args.kwargs
assert call_kwargs["domain"] == ResearchDomain.SEXUAL_HEALTH
@patch("src.orchestrators.factory._get_advanced_orchestrator_class")
def test_create_advanced_uses_domain(self, mock_get_cls):
mock_adv_cls = MagicMock()
mock_get_cls.return_value = mock_adv_cls
create_orchestrator(mode="advanced", domain=ResearchDomain.SEXUAL_HEALTH)
call_kwargs = mock_adv_cls.call_args.kwargs
assert call_kwargs["domain"] == ResearchDomain.SEXUAL_HEALTH
|