测试策略与 Mock 实践

@SpringBootTest
@AutoConfigureMockMvc
public class ChatClientUnitTest {

    @MockBean
    private ChatModel chatModel;  // Mock 底层模型，避免真实 API 调用

    @Autowired
    private ChatClient.Builder chatClientBuilder;

    private ChatClient chatClient;

    @BeforeEach
    void setUp() {
        chatClient = chatClientBuilder
            .defaultTools(new WeatherTools())  // 真实工具对象
            .build();
    }

    @Test
    void testToolCallingFlow() {
        // 1. 预设 AI 返回"工具调用"指令
        AssistantMessage toolCallMessage = new AssistantMessage(
            "获取天气信息",
            Map.of(),
            List.of(new ToolCall(
                "getWeather",
                "getWeather",
                new HashMap<>(Map.of("city", "广州"))
            ))
        );

        ChatResponse mockResponse = new ChatResponse(
            List.of(new Generation(toolCallMessage))
        );

        when(chatModel.call(any(Prompt.class)))
            .thenReturn(mockResponse);

        // 2. 执行
        String result = chatClient.prompt()
            .user("今天广州天气怎么样？")
            .call()
            .content();

        // 3. 验证
        assertThat(result).contains("广州");
        verify(chatModel, atLeastOnce()).call(any(Prompt.class));
    }

    @Test
    void testAdvisorModifiesPrompt() {
        // 预设 AI 简单回复
        AssistantMessage aiReply = new AssistantMessage(
            "你好！我是飞翔科技智能助手。有什么可以帮助你？");
        ChatResponse mockResponse = new ChatResponse(
            List.of(new Generation(aiReply))
        );

        when(chatModel.call(any(Prompt.class)))
            .thenAnswer(invocation -> {
                Prompt actualPrompt = invocation.getArgument(0);
                String userContent = actualPrompt.getInstructions().stream()
                    .filter(m -> m.getMessageType() == MessageType.USER)
                    .map(Message::getContent)
                    .findFirst()
                    .orElse("");
                // 验证 Prompt 中是否被 Advisor 注入了内容
                assertThat(userContent).doesNotContain("SYSTEM_INJECT");
                return mockResponse;
            });

        String result = chatClient.prompt()
            .user("测试消息")
            .call()
            .content();

        assertThat(result).contains("飞翔科技");
    }
}

@SpringBootTest
@Testcontainers
public class RagIntegrationTest {

    @Container
    static PostgreSQLContainer<?> postgres = new PostgreSQLContainer<>(
        "pgvector/pgvector:pg16"
    )
        .withDatabaseName("vectordb")
        .withUsername("test")
        .withPassword("test");

    @DynamicPropertySource
    static void configure(DynamicPropertyRegistry registry) {
        registry.add("spring.datasource.url", postgres::getJdbcUrl);
        registry.add("spring.datasource.username", postgres::getUsername);
        registry.add("spring.datasource.password", postgres::getPassword);
    }

    @Autowired
    private VectorStore vectorStore;

    @Test
    void testDocumentIngestionAndSearch() {
        // 1. 注入测试文档
        Document doc1 = new Document(
            "数据结构是计算机科学的基础课程，涵盖线性表、树、图等核心概念。",
            Map.of("course", "CS101", "teacher", "白歌")
        );
        Document doc2 = new Document(
            "线性代数主要研究向量空间和线性变换，是机器学习的基础数学工具。",
            Map.of("course", "MATH201", "teacher", "大翔")
        );

        vectorStore.add(List.of(doc1, doc2));

        // 2. 语义检索
        SearchRequest request = SearchRequest.query("计算机基础课程有哪些")
            .withTopK(2)
            .withSimilarityThreshold(0.6);

        List<Document> results = vectorStore
            .similaritySearch(request);

        // 3. 验证
        assertThat(results).isNotEmpty();
        assertThat(results.get(0).getContent())
            .contains("数据结构");
        assertThat(results.get(0).getMetadata())
            .containsEntry("course", "CS101");
    }

    @Test
    void testMetadataFiltering() {
        Document doc1 = new Document(
            "机器学习实战课程", Map.of("level", "advanced")
        );
        Document doc2 = new Document(
            "编程基础入门", Map.of("level", "beginner")
        );
        vectorStore.add(List.of(doc1, doc2));

        SearchRequest request = SearchRequest
            .query("适合新手的课程")
            .withFilterExpression("level == 'beginner'")
            .withTopK(1);

        List<Document> results = vectorStore
            .similaritySearch(request);

        assertThat(results).hasSize(1);
        assertThat(results.get(0).getContent())
            .contains("基础入门");
    }
}

// ❌ 错误：没有 Mock ChatModel
@SpringBootTest
public class ChatTest {
    @Autowired
    private ChatClient chatClient;

    @Test
    void testSomething() {
        // 真的调用了 OpenAI API！
        chatClient.prompt().user("hello").call().content();
    }
}

// ✅ 正确：Mock ChatModel
@SpringBootTest
public class ChatTest {
    @MockBean
    private ChatModel chatModel;

    @Autowired
    private ChatClient.Builder builder;

    @Test
    void testSomething() {
        when(chatModel.call(any(Prompt.class)))
            .thenReturn(new ChatResponse(
                List.of(new Generation(
                    new AssistantMessage("Mock reply")))));
        // 零费用、确定性测试
    }
}