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

This commit is contained in:
2026-06-01 05:40:59 +08:00
parent eb64af9d50
commit 91e05a26cd
5 changed files with 282 additions and 0 deletions

View File

@@ -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"));
}
}

View File

@@ -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));
}
}

View File

@@ -6,6 +6,8 @@ import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity; import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ExceptionHandler; import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice; import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;
import org.springframework.web.bind.MissingServletRequestParameterException;
@Slf4j @Slf4j
@RestControllerAdvice @RestControllerAdvice
@@ -17,6 +19,15 @@ public class GlobalExceptionHandler {
return buildResponse(HttpStatus.BAD_REQUEST, exception.getMessage()); 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) @ExceptionHandler(Exception.class)
public ResponseEntity<RequestResult<Void>> handleException(Exception exception) { public ResponseEntity<RequestResult<Void>> handleException(Exception exception) {
log.error("GlobalExceptionHandler.handleException", exception); log.error("GlobalExceptionHandler.handleException", exception);

View File

@@ -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());
}
}

View File

@@ -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));
}
}