test(api): 补齐剩余业务模块控制器接口验证

This commit is contained in:
2026-06-01 05:44:38 +08:00
parent 91e05a26cd
commit 15808b8569
5 changed files with 320 additions and 0 deletions

View File

@@ -0,0 +1,68 @@
package com.bruce.observability.controller;
import com.bruce.common.handler.GlobalExceptionHandler;
import com.bruce.observability.service.IObservabilityExportService;
import com.bruce.observability.service.IObservabilityRunService;
import com.bruce.observability.service.IObservabilityTraceService;
import com.bruce.observability.vo.ObservabilityTraceVO;
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 java.math.BigDecimal;
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;
/**
* 验证运行观测 Trace 接口的路径参数绑定和返回结构。
*/
@ExtendWith(MockitoExtension.class)
class ObservabilityTraceControllerTests {
private MockMvc mockMvc;
@Mock
private IObservabilityRunService observabilityRunService;
@Mock
private IObservabilityTraceService observabilityTraceService;
@Mock
private IObservabilityExportService observabilityExportService;
@InjectMocks
private ObservabilityTraceController observabilityTraceController;
@BeforeEach
void setUp() {
mockMvc = MockMvcBuilders.standaloneSetup(observabilityTraceController)
.setControllerAdvice(new GlobalExceptionHandler())
.build();
}
@Test
void traceShouldReturnStructuredTraceView() throws Exception {
ObservabilityTraceVO trace = new ObservabilityTraceVO();
trace.setRequestId("req-1001");
trace.setWorkflowStatus("SUCCESS");
trace.setModelCallCount(2);
trace.setEstimatedCost(BigDecimal.valueOf(0.18));
when(observabilityTraceService.getTrace("req-1001")).thenReturn(trace);
mockMvc.perform(get("/api/observability/runs/req-1001"))
.andExpect(status().isOk())
.andExpect(jsonPath("$.resultcode").value("0"))
.andExpect(jsonPath("$.data.requestId").value("req-1001"))
.andExpect(jsonPath("$.data.workflowStatus").value("SUCCESS"))
.andExpect(jsonPath("$.data.modelCallCount").value(2));
}
}