93 lines
3.2 KiB
Java
93 lines
3.2 KiB
Java
package org.example.service;
|
||
|
||
import org.junit.jupiter.api.DisplayName;
|
||
import org.junit.jupiter.api.Test;
|
||
import org.springframework.ai.chat.model.ChatModel;
|
||
import org.springframework.ai.tool.ToolCallback;
|
||
import org.springframework.ai.tool.ToolCallbackProvider;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.boot.test.context.SpringBootTest;
|
||
import org.springframework.boot.test.context.TestConfiguration;
|
||
import org.springframework.context.ApplicationContext;
|
||
import org.springframework.context.annotation.Bean;
|
||
|
||
import static org.junit.jupiter.api.Assertions.*;
|
||
|
||
/**
|
||
* Chat + Embedding 解耦验证测试
|
||
* <p>
|
||
* 验证 ChatModel Bean 注入、ModelRoutingConfig 路由、ChatService 接口兼容。
|
||
* Chat: DeepSeek via OpenAI-compatible API
|
||
* Embedding: Ollama BGE-M3 (需要本地 ollama 运行)
|
||
*/
|
||
@SpringBootTest
|
||
@DisplayName("Chat + Embedding 解耦验证")
|
||
class ChatAndEmbeddingSmokeTest {
|
||
|
||
@Autowired
|
||
private ApplicationContext context;
|
||
|
||
@Autowired
|
||
private ChatModel chatModel;
|
||
|
||
@Autowired
|
||
private ChatService chatService;
|
||
|
||
/**
|
||
* 提供 mock ToolCallbackProvider（MCP 已禁用时需要）
|
||
*/
|
||
@TestConfiguration
|
||
static class MockToolConfig {
|
||
@Bean
|
||
public ToolCallbackProvider toolCallbackProvider() {
|
||
return () -> new ToolCallback[0];
|
||
}
|
||
}
|
||
|
||
@Test
|
||
@DisplayName("Spring 容器启动成功")
|
||
void contextLoads() {
|
||
assertNotNull(context, "Spring 容器应为非空");
|
||
assertNotNull(chatModel, "ChatModel Bean 应注入成功");
|
||
assertNotNull(chatService, "ChatService Bean 应注入成功");
|
||
}
|
||
|
||
@Test
|
||
@DisplayName("ModelRoutingConfig @Primary ChatModel 生效")
|
||
void chatModelPrimaryBeanWorks() {
|
||
assertNotNull(chatModel, "@Primary ChatModel 应被自动注入");
|
||
System.out.println("✓ ChatModel 类型: " + chatModel.getClass().getName());
|
||
|
||
// 验证路由到 openAiChatModel (DeepSeek)
|
||
assertTrue(context.containsBean("openAiChatModel"), "openAiChatModel 应存在");
|
||
System.out.println(" Chat → openAiChatModel (DeepSeek) ✓");
|
||
}
|
||
|
||
@Test
|
||
@DisplayName("ChatService.createReactAgent 接受 ChatModel 接口")
|
||
void chatServiceAcceptsChatModelInterface() {
|
||
var agent = chatService.createReactAgent(chatModel, "测试系统提示词");
|
||
assertNotNull(agent, "ReactAgent 应创建成功");
|
||
assertEquals("intelligent_assistant", agent.name());
|
||
System.out.println("✓ ReactAgent 创建成功: " + agent.name());
|
||
}
|
||
|
||
@Test
|
||
@DisplayName("ChatModel 实现 ChatModel 接口（类型安全验证）")
|
||
void chatModelIsProperType() {
|
||
assertNotNull(chatModel, "注入的 Bean 应为 ChatModel 实例");
|
||
System.out.println("✓ ChatModel 接口实现: " + chatModel.getClass().getSimpleName());
|
||
}
|
||
|
||
@Test
|
||
@DisplayName("EmbeddingModel 状态")
|
||
void embeddingModelStatus() {
|
||
boolean hasEmbedding = context.containsBean("embeddingModel");
|
||
if (hasEmbedding) {
|
||
System.out.println("✓ EmbeddingModel 已配置");
|
||
} else {
|
||
System.out.println("⚠ EmbeddingModel 未找到 — 检查 Ollama 是否运行");
|
||
}
|
||
}
|
||
}
|