import pytest


@pytest.mark.asyncio
@pytest.mark.parametrize("enable_diarization", [False, True])
async def test_basic_process(
    dummy_transcript,
    dummy_llm,
    dummy_processors,
    enable_diarization,
    dummy_diarization,
):
    # goal is to start the server, and send rtc audio to it
    # validate the events received
    from pathlib import Path

    from reflector.settings import settings
    from reflector.tools.process import process_audio_file

    # LLM_BACKEND no longer exists in settings
    # settings.LLM_BACKEND = "test"
    settings.TRANSCRIPT_BACKEND = "whisper"

    # event callback
    marks = {}

    async def event_callback(event):
        if event.processor not in marks:
            marks[event.processor] = 0
        marks[event.processor] += 1

    # invoke the process and capture events
    path = Path(__file__).parent / "records" / "test_mathieu_hello.wav"

    if enable_diarization:
        # Test with diarization - may fail if pyannote.audio is not installed
        try:
            await process_audio_file(
                path.as_posix(), event_callback, enable_diarization=True
            )
        except SystemExit:
            pytest.skip("pyannote.audio not installed - skipping diarization test")
    else:
        # Test without diarization - should always work
        await process_audio_file(
            path.as_posix(), event_callback, enable_diarization=False
        )

    print(f"Diarization: {enable_diarization}, Marks: {marks}")

    # validate the events
    # Each processor should be called for each audio segment processed
    # The final processors (Topic, Title, Summary) should be called once at the end
    assert marks["TranscriptLinerProcessor"] > 0
    assert marks["TranscriptTranslatorPassthroughProcessor"] > 0
    assert marks["TranscriptTopicDetectorProcessor"] == 1
    assert marks["TranscriptFinalSummaryProcessor"] == 1
    assert marks["TranscriptFinalTitleProcessor"] == 1

    if enable_diarization:
        assert marks["TestAudioDiarizationProcessor"] == 1