package com.vincent.rsf.server.ai.service.impl.chat;

import com.vincent.rsf.server.ai.dto.AiChatTraceEventDto;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.ArgumentCaptor;
import org.mockito.Captor;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

import java.util.List;
import java.util.concurrent.atomic.AtomicLong;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

@ExtendWith(MockitoExtension.class)
class AiChatTraceEmitterTest {

    @Mock
    private AiSseEventPublisher aiSseEventPublisher;

    @Captor
    private ArgumentCaptor<AiChatTraceEventDto> payloadCaptor;

    @Test
    void shouldReuseTraceIdentityForLogicalTraceCards() {
        AiChatTraceEmitter traceEmitter = new AiChatTraceEmitter(
                aiSseEventPublisher,
                new SseEmitter(1000L),
                "req-1",
                11L,
                new AtomicLong(0)
        );

        traceEmitter.startAnalyze();
        traceEmitter.onToolStart("inventory.lookup", "builtin-stock", "tool-1", "{\"code\":\"A01\"}", 100L);
        traceEmitter.onToolResult("inventory.lookup", "builtin-stock", "tool-1", "{\"code\":\"A01\"}",
                "{\"stock\":12}", null, 64L, 164L, false);

        verify(aiSseEventPublisher, times(4)).emitSafely(any(), eq("trace"), payloadCaptor.capture());

        List<AiChatTraceEventDto> payloads = payloadCaptor.getAllValues();
        assertThat(payloads).extracting(AiChatTraceEventDto::getSequence)
                .containsExactly(1L, 1L, 2L, 2L);
        assertThat(payloads).extracting(AiChatTraceEventDto::getTraceId)
                .containsExactly("req-1-thinking-ANALYZE", "req-1-thinking-ANALYZE", "tool-1", "tool-1");
        assertThat(payloads).extracting(AiChatTraceEventDto::getTraceType)
                .containsExactly("thinking", "thinking", "tool", "tool");
        assertThat(payloads.get(1).getStatus()).isEqualTo("COMPLETED");
        assertThat(payloads.get(3).getStatus()).isEqualTo("COMPLETED");
        assertThat(payloads.get(3).getToolCallId()).isEqualTo("tool-1");
        assertThat(payloads.get(3).getInputSummary()).isEqualTo("{\"code\":\"A01\"}");
    }
}