test(api): 补齐聚合接口控制器测试并修复参数异常语义

2026-06-01 05:40:59 +08:00
parent eb64af9d50
commit 91e05a26cd
5 changed files with 282 additions and 0 deletions
--- a/common-agent-agent/src/test/java/com/bruce/agent/controller/AgentSessionControllerTests.java
+++ b/common-agent-agent/src/test/java/com/bruce/agent/controller/AgentSessionControllerTests.java
@@ -0,0 +1,69 @@
+package com.bruce.agent.controller;
+
+import com.bruce.agent.service.IAgentMessageService;
+import com.bruce.agent.service.IAgentSessionService;
+import com.bruce.agent.service.IAgentWorkspaceService;
+import com.bruce.agent.vo.AgentWorkspaceVO;
+import com.bruce.common.handler.GlobalExceptionHandler;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+/**
+ * 验证 Agent 工作台聚合接口的查询参数绑定和返回结构。
+ */
+@ExtendWith(MockitoExtension.class)
+class AgentSessionControllerTests {
+
+    private MockMvc mockMvc;
+
+    @Mock
+    private IAgentSessionService agentSessionService;
+
+    @Mock
+    private IAgentMessageService agentMessageService;
+
+    @Mock
+    private IAgentWorkspaceService agentWorkspaceService;
+
+    @InjectMocks
+    private AgentSessionController agentSessionController;
+
+    @BeforeEach
+    void setUp() {
+        mockMvc = MockMvcBuilders.standaloneSetup(agentSessionController)
+                .setControllerAdvice(new GlobalExceptionHandler())
+                .build();
+    }
+
+    @Test
+    void workspaceShouldReturnStructuredAggregateView() throws Exception {
+        AgentWorkspaceVO workspace = new AgentWorkspaceVO();
+        workspace.setAgentId(1001L);
+        workspace.setAgentCode("presale_agent");
+        workspace.setAgentName("售前问答 Agent");
+        workspace.setSessionId(2001L);
+        workspace.setSessionCode("session_001");
+        workspace.setLatestRequestId("req-1001");
+        workspace.setCitationCount(2);
+
+        when(agentWorkspaceService.getWorkspace(1001L, null)).thenReturn(workspace);
+
+        mockMvc.perform(get("/api/agent-sessions/workspace").param("agentId", "1001"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.resultcode").value("0"))
+                .andExpect(jsonPath("$.data.agentId").value(1001))
+                .andExpect(jsonPath("$.data.agentName").value("售前问答 Agent"))
+                .andExpect(jsonPath("$.data.latestRequestId").value("req-1001"));
+    }
+}
--- a/common-agent-boot/src/test/java/com/bruce/dashboard/StudioDashboardControllerTests.java
+++ b/common-agent-boot/src/test/java/com/bruce/dashboard/StudioDashboardControllerTests.java
@@ -0,0 +1,67 @@
+package com.bruce.dashboard;
+
+import com.bruce.common.handler.GlobalExceptionHandler;
+import com.bruce.dashboard.controller.StudioDashboardController;
+import com.bruce.dashboard.service.IStudioDashboardService;
+import com.bruce.dashboard.vo.StudioDashboardMetricsVO;
+import com.bruce.dashboard.vo.StudioDashboardVO;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+/**
+ * 验证 Studio 首页聚合接口响应结构，确保前端首页可稳定消费。
+ */
+@ExtendWith(MockitoExtension.class)
+class StudioDashboardControllerTests {
+
+    private MockMvc mockMvc;
+
+    @Mock
+    private IStudioDashboardService studioDashboardService;
+
+    @InjectMocks
+    private StudioDashboardController studioDashboardController;
+
+    @BeforeEach
+    void setUp() {
+        mockMvc = MockMvcBuilders.standaloneSetup(studioDashboardController)
+                .setControllerAdvice(new GlobalExceptionHandler())
+                .build();
+    }
+
+    @Test
+    void detailShouldReturnStructuredDashboardView() throws Exception {
+        StudioDashboardMetricsVO metrics = new StudioDashboardMetricsVO();
+        metrics.setTodayRunCount(12);
+        metrics.setSuccessRate(98.5D);
+        metrics.setP50Latency("1.28s");
+        metrics.setEstimatedCost("¥4.82");
+
+        StudioDashboardVO dashboard = new StudioDashboardVO();
+        dashboard.setProjectName("Common Agent Studio");
+        dashboard.setEnvironment("Dev");
+        dashboard.setPublishStatus("DRAFT");
+        dashboard.setMetrics(metrics);
+        dashboard.setWarningTitle("发布前仍需补齐模型路由");
+
+        when(studioDashboardService.getDashboard()).thenReturn(dashboard);
+
+        mockMvc.perform(get("/api/studio/dashboard"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.resultcode").value("0"))
+                .andExpect(jsonPath("$.data.projectName").value("Common Agent Studio"))
+                .andExpect(jsonPath("$.data.environment").value("Dev"))
+                .andExpect(jsonPath("$.data.metrics.todayRunCount").value(12));
+    }
+}
--- a/common-agent-common/src/main/java/com/bruce/common/handler/GlobalExceptionHandler.java
+++ b/common-agent-common/src/main/java/com/bruce/common/handler/GlobalExceptionHandler.java
@@ -6,6 +6,8 @@ import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.ExceptionHandler;
 import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;
+import org.springframework.web.bind.MissingServletRequestParameterException;

@Slf4j
@RestControllerAdvice
@@ -17,6 +19,15 @@ public class GlobalExceptionHandler {
        return buildResponse(HttpStatus.BAD_REQUEST, exception.getMessage());
    }

+    @ExceptionHandler({
+            MissingServletRequestParameterException.class,
+            MethodArgumentTypeMismatchException.class
+    })
+    public ResponseEntity<RequestResult<Void>> handleBadRequest(Exception exception) {
+        log.warn("GlobalExceptionHandler.handleBadRequest, message={}", exception.getMessage(), exception);
+        return buildResponse(HttpStatus.BAD_REQUEST, "请求参数不合法");
+    }
+
    @ExceptionHandler(Exception.class)
    public ResponseEntity<RequestResult<Void>> handleException(Exception exception) {
        log.error("GlobalExceptionHandler.handleException", exception);
--- a/common-agent-common/src/test/java/com/bruce/common/controller/SysEnumControllerTests.java
+++ b/common-agent-common/src/test/java/com/bruce/common/controller/SysEnumControllerTests.java
@@ -0,0 +1,76 @@
+package com.bruce.common.controller;
+
+import com.bruce.common.dto.response.SysEnumResponse;
+import com.bruce.common.handler.GlobalExceptionHandler;
+import com.bruce.common.service.ISysEnumService;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.springframework.http.MediaType;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+import java.util.List;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+/**
+ * 验证系统枚举控制器的基础接口契约，确保前端依赖的 RequestResult 结构稳定。
+ */
+@ExtendWith(MockitoExtension.class)
+class SysEnumControllerTests {
+
+    private MockMvc mockMvc;
+
+    @Mock
+    private ISysEnumService sysEnumService;
+
+    @BeforeEach
+    void setUp() {
+        SysEnumController controller = new SysEnumController();
+        ReflectionTestUtils.setField(controller, "sysEnumService", sysEnumService);
+        mockMvc = MockMvcBuilders.standaloneSetup(controller)
+                .setControllerAdvice(new GlobalExceptionHandler())
+                .build();
+    }
+
+    @Test
+    void queryForManagementShouldReturnStructuredResult() throws Exception {
+        SysEnumResponse response = new SysEnumResponse();
+        response.setId(1L);
+        response.setCatalog("common");
+        response.setType("enable_status");
+        response.setName("启用");
+        response.setValue(1);
+        response.setStrvalue("ENABLED");
+
+        when(sysEnumService.listForManagement(any())).thenReturn(List.of(response));
+
+        mockMvc.perform(post("/api/sys-enum/queryForManagement")
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content("""
+                                {
+                                  "catalog": "common",
+                                  "type": "enable_status"
+                                }
+                                """))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.resultcode").value("0"))
+                .andExpect(jsonPath("$.data[0].catalog").value("common"))
+                .andExpect(jsonPath("$.data[0].strvalue").value("ENABLED"));
+    }
+
+    @Test
+    void detailShouldReturnBadRequestWhenIdMissing() throws Exception {
+        mockMvc.perform(get("/api/sys-enum/detail"))
+                .andExpect(status().isBadRequest());
+    }
+}
--- a/common-agent-rag/src/test/java/com/bruce/rag/controller/KnowledgeWorkspaceControllerTests.java
+++ b/common-agent-rag/src/test/java/com/bruce/rag/controller/KnowledgeWorkspaceControllerTests.java
@@ -0,0 +1,59 @@
+package com.bruce.rag.controller;
+
+import com.bruce.common.handler.GlobalExceptionHandler;
+import com.bruce.rag.service.IKnowledgeWorkspaceService;
+import com.bruce.rag.vo.KnowledgeWorkspaceVO;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+import static org.mockito.Mockito.when;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+/**
+ * 验证知识工作台聚合接口的路径参数绑定和响应结构。
+ */
+@ExtendWith(MockitoExtension.class)
+class KnowledgeWorkspaceControllerTests {
+
+    private MockMvc mockMvc;
+
+    @Mock
+    private IKnowledgeWorkspaceService knowledgeWorkspaceService;
+
+    @InjectMocks
+    private KnowledgeWorkspaceController knowledgeWorkspaceController;
+
+    @BeforeEach
+    void setUp() {
+        mockMvc = MockMvcBuilders.standaloneSetup(knowledgeWorkspaceController)
+                .setControllerAdvice(new GlobalExceptionHandler())
+                .build();
+    }
+
+    @Test
+    void getWorkspaceShouldReturnStructuredWorkspaceView() throws Exception {
+        KnowledgeWorkspaceVO workspace = new KnowledgeWorkspaceVO();
+        workspace.setStoreId(1001L);
+        workspace.setStoreCode("PROD_DOC");
+        workspace.setStoreName("产品制度库");
+        workspace.setDocumentCount(9);
+        workspace.setHealthScore(88);
+
+        when(knowledgeWorkspaceService.getWorkspace(1001L)).thenReturn(workspace);
+
+        mockMvc.perform(get("/api/knowledge/workspaces/1001"))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.resultcode").value("0"))
+                .andExpect(jsonPath("$.data.storeId").value("1001"))
+                .andExpect(jsonPath("$.data.storeName").value("产品制度库"))
+                .andExpect(jsonPath("$.data.healthScore").value(88));
+    }
+}