package com.vincent.rsf.server.ai.service.impl.mcp;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import com.vincent.rsf.server.ai.config.AiDefaults;
|
import com.vincent.rsf.server.ai.dto.AiMcpToolPreviewDto;
|
import com.vincent.rsf.server.ai.entity.AiMcpMount;
|
import com.vincent.rsf.server.ai.mapper.AiMcpMountMapper;
|
import com.vincent.rsf.server.ai.service.BuiltinMcpToolRegistry;
|
import com.vincent.rsf.server.ai.service.McpMountRuntimeFactory;
|
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 org.springframework.ai.tool.metadata.ToolMetadata;
|
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.when;
|
|
@ExtendWith(MockitoExtension.class)
|
class AiMcpAdminServiceTest {
|
|
@Mock
|
private AiMcpMountMapper aiMcpMountMapper;
|
@Mock
|
private BuiltinMcpToolRegistry builtinMcpToolRegistry;
|
@Mock
|
private McpMountRuntimeFactory mcpMountRuntimeFactory;
|
|
@Test
|
void shouldMergeGovernedCatalogIntoPreviewResult() {
|
AiMcpAdminService aiMcpAdminService = new AiMcpAdminService(
|
aiMcpMountMapper,
|
builtinMcpToolRegistry,
|
mcpMountRuntimeFactory,
|
new ObjectMapper()
|
);
|
AiMcpMount mount = new AiMcpMount()
|
.setTransportType(AiDefaults.MCP_TRANSPORT_BUILTIN)
|
.setBuiltinCode(AiDefaults.MCP_BUILTIN_RSF_WMS)
|
.setName("builtin");
|
ToolCallback callback = mock(ToolCallback.class);
|
ToolDefinition toolDefinition = mock(ToolDefinition.class);
|
ToolMetadata toolMetadata = mock(ToolMetadata.class);
|
McpMountRuntimeFactory.McpMountRuntime runtime = mock(McpMountRuntimeFactory.McpMountRuntime.class);
|
when(toolDefinition.name()).thenReturn("rsf_query_task");
|
when(toolDefinition.description()).thenReturn("desc");
|
when(toolDefinition.inputSchema()).thenReturn("{schema}");
|
when(toolMetadata.returnDirect()).thenReturn(Boolean.FALSE);
|
when(callback.getToolDefinition()).thenReturn(toolDefinition);
|
when(callback.getToolMetadata()).thenReturn(toolMetadata);
|
when(runtime.getToolCallbacks()).thenReturn(new ToolCallback[]{callback});
|
when(runtime.getErrors()).thenReturn(List.of());
|
when(mcpMountRuntimeFactory.create(List.of(mount), 1L)).thenReturn(runtime);
|
when(builtinMcpToolRegistry.listBuiltinToolCatalog(AiDefaults.MCP_BUILTIN_RSF_WMS)).thenReturn(List.of(
|
AiMcpToolPreviewDto.builder()
|
.name("rsf_query_task")
|
.toolGroup("任务查询")
|
.toolPurpose("purpose")
|
.queryBoundary("boundary")
|
.exampleQuestions(List.of("q1"))
|
.build()
|
));
|
|
List<AiMcpToolPreviewDto> result = aiMcpAdminService.previewTools(mount, 1L);
|
|
assertEquals(1, result.size());
|
assertEquals("任务查询", result.get(0).getToolGroup());
|
assertEquals("purpose", result.get(0).getToolPurpose());
|
assertEquals("{schema}", result.get(0).getInputSchema());
|
}
|
}
|
