tests/test_model_overrides_cross_mode.py

import types

import pytest

from src.interface.manual_input_interface import ManualInputController
from src.interface.file_upload_interface import FileUploadInterfaceController


def test_manual_input_controller_applies_overrides_to_ai_client():
    controller = ManualInputController()

    controller.set_model_overrides({"SpiritualDistressAnalyzer": "gemini-test-model"})

    assert controller.model_overrides["SpiritualDistressAnalyzer"] == "gemini-test-model"
    assert controller.ai_client.model_overrides["SpiritualDistressAnalyzer"] == "gemini-test-model"


def test_file_upload_controller_applies_overrides_to_ai_client():
    controller = FileUploadInterfaceController()

    controller.set_model_overrides({"SpiritualDistressAnalyzer": "gemini-test-model"})

    assert controller.model_overrides["SpiritualDistressAnalyzer"] == "gemini-test-model"
    assert controller.ai_client.model_overrides["SpiritualDistressAnalyzer"] == "gemini-test-model"


def test_file_upload_batch_passes_override_into_call(monkeypatch):
    controller = FileUploadInterfaceController()
    controller.set_model_overrides({"EntryClassifier": "gemini-batch-override"})

    # Minimal session + dataset state so the batch method can run without UI.
    controller.current_session = types.SimpleNamespace(
        verifications=[],
        verified_count=0,
        correct_count=0,
        incorrect_count=0,
        verified_message_ids=[],
        current_queue_index=0,
        is_complete=False,
        completed_at=None,
    )

    controller.current_file_result = types.SimpleNamespace(
        parsed_test_cases=[
            types.SimpleNamespace(
                message_id="m1",
                text="msg1",
                pre_classified_label="green",
            )
        ]
    )

    # Avoid touching file processing / exporters.
    controller._initialize_progress_tracker = lambda total_messages: None
    controller._create_session = lambda file_result, file_name: None
    controller._update_progress_display_text = lambda *args, **kwargs: ""
    controller.setup_progress_tracking = lambda total_messages: None
    controller.record_verification_with_timing = lambda *args, **kwargs: None
    controller.store.save_session = lambda session: None

    captured = {}

    def fake_call_entry_classifier_api(*, system_prompt=None, user_prompt=None, model_override=None, **kwargs):
        captured["system_prompt"] = system_prompt
        captured["user_prompt"] = user_prompt
        captured["model_override"] = model_override
        return "green"

    monkeypatch.setattr(controller.ai_client, "call_entry_classifier_api", fake_call_entry_classifier_api)

    # Run the simplest possible classification path.
    result = controller.run_batch_classification()

    assert captured["model_override"] == "gemini-batch-override"
    assert result is not None