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

import com.vincent.rsf.server.ai.service.AiCallLogService;
import com.vincent.rsf.server.ai.service.MountedToolCallback;
import com.vincent.rsf.server.ai.store.AiCachedToolResult;
import com.vincent.rsf.server.ai.store.AiToolResultStore;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.ai.tool.ToolCallback;
import org.springframework.ai.tool.definition.ToolDefinition;

import java.util.concurrent.atomic.AtomicLong;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
class AiToolObservationServiceTest {

    @Mock
    private AiToolResultStore aiToolResultStore;
    @Mock
    private AiCallLogService aiCallLogService;
    @Mock
    private AiChatTraceEmitter traceEmitter;
    @Mock
    private MountedToolCallback mountedToolCallback;
    @Mock
    private ToolDefinition toolDefinition;

    @Test
    void shouldSkipStartTraceWhenToolResultComesFromCache() {
        AiToolObservationService aiToolObservationService = new AiToolObservationService(aiToolResultStore, aiCallLogService);
        when(mountedToolCallback.getToolDefinition()).thenReturn(toolDefinition);
        when(toolDefinition.name()).thenReturn("inventory.lookup");
        when(mountedToolCallback.getMountName()).thenReturn("builtin-stock");
        when(aiToolResultStore.getToolResult(1L, "req-1", "inventory.lookup", "{\"code\":\"A01\"}"))
                .thenReturn(AiCachedToolResult.builder()
                        .success(true)
                        .output("{\"stock\":12}")
                        .build());

        ToolCallback[] callbacks = aiToolObservationService.wrapToolCallbacks(
                new ToolCallback[]{mountedToolCallback},
                "req-1",
                11L,
                new AtomicLong(0),
                new AtomicLong(0),
                new AtomicLong(0),
                21L,
                31L,
                1L,
                traceEmitter
        );

        String output = callbacks[0].call("{\"code\":\"A01\"}");

        assertThat(output).isEqualTo("{\"stock\":12}");
        verify(traceEmitter, never()).onToolStart(eq("inventory.lookup"), eq("builtin-stock"), eq("req-1-tool-1"), eq("{\"code\":\"A01\"}"), anyLong());
        verify(traceEmitter).onToolResult(eq("inventory.lookup"), eq("builtin-stock"), eq("req-1-tool-1"),
                eq("{\"code\":\"A01\"}"), eq("{\"stock\":12}"), eq(null), eq(0L), anyLong(), eq(false));
        verify(mountedToolCallback, never()).call("{\"code\":\"A01\"}");
    }
}