[alerter] support AlertsController unit test case (#442) (#587)

alerter/src/test/java/com/usthe/alert/controller/AlertsControllerTest.java

@@ -1,40 +1,168 @@
 package com.usthe.alert.controller;
 
+import com.usthe.alert.dto.AlertSummary;
+import com.usthe.alert.service.AlertService;
+import com.usthe.common.entity.alerter.Alert;
+import com.usthe.common.entity.dto.AlertReport;
+import com.usthe.common.util.CommonConstants;
+import com.usthe.common.util.GsonUtil;
 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.Mockito;
+import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.data.domain.Page;
+import org.springframework.data.domain.PageImpl;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.domain.Sort;
+import org.springframework.data.jpa.domain.Specification;
+import org.springframework.http.MediaType;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 
-import static org.junit.jupiter.api.Assertions.*;
+import java.util.Collections;
+import java.util.List;
+import java.util.stream.Collectors;
+import java.util.stream.LongStream;
+
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;
 
 /**
  * Test case for {@link AlertsController}
  */
+@ExtendWith(MockitoExtension.class)
 class AlertsControllerTest {
 
+    private MockMvc mockMvc;
+
+    @InjectMocks
+    private AlertsController alertsController;
+
+    @Mock
+    private AlertService alertService;
+
+    private List<Long> ids;
+
+
     @BeforeEach
     void setUp() {
+        this.mockMvc = MockMvcBuilders.standaloneSetup(alertsController).build();
+        ids = LongStream.rangeClosed(1, 10).boxed().collect(Collectors.toList());
     }
 
     @Test
-    void getAlerts() {
+    void getAlerts() throws Exception {
+
+        //定义要用到的测试值
+        String sortField = "id";
+        String orderType = "asc";
+        int pageIndex = 0;
+        int pageSize = 10;
+        PageRequest pageRequest = PageRequest.of(pageIndex, pageSize, Sort.by(new Sort.Order(Sort.Direction.fromString(orderType), sortField)));
+        Page<Alert> alertPage = new PageImpl<>(Collections.singletonList(Alert.builder().build()));
+
+
+        //打桩
+        Mockito.when(
+                        alertService.getAlerts(
+                                Mockito.any(Specification.class)
+                                , Mockito.argThat(
+                                        argument ->
+                                                argument.getPageNumber() == pageRequest.getPageNumber()
+                                                        && argument.getPageSize() == pageRequest.getPageSize()
+                                                        && argument.getSort().equals(pageRequest.getSort())
+                                )
+                        )
+                )
+                .thenReturn(alertPage);
+
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .get("/api/alerts")
+                                .param("ids", ids.stream().map(String::valueOf).collect(Collectors.joining(",")))
+                                .param("monitorId", "1")
+                                .param("priority", "1")
+                                .param("status", "1")
+                                .param("content", "test")
+                                .param("sort", sortField)
+                                .param("order", orderType)
+                                .param("pageIndex", String.valueOf(pageIndex))
+                                .param("pageSize", String.valueOf(pageSize))
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(jsonPath("$.data.content.length()").value(1))
+                .andReturn();
     }
 
     @Test
-    void deleteAlerts() {
+    void deleteAlerts() throws Exception {
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .delete("/api/alerts")
+                                .param("ids", ids.stream().map(String::valueOf).collect(Collectors.joining(",")))
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(content().json("{\"data\":null,\"msg\":null,\"code\":0}"))
+                .andReturn();
     }
 
     @Test
-    void clearAllAlerts() {
+    void clearAllAlerts() throws Exception {
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .delete("/api/alerts/clear")
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(content().json("{\"data\":null,\"msg\":null,\"code\":0}"))
+                .andReturn();
     }
 
     @Test
-    void applyAlertDefinesStatus() {
+    void applyAlertDefinesStatus() throws Exception {
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .put("/api/alerts/status/1")
+                                .param("ids", ids.stream().map(String::valueOf).collect(Collectors.joining(",")))
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(content().json("{\"data\":null,\"msg\":null,\"code\":0}"))
+                .andReturn();
     }
 
     @Test
-    void getAlertsSummary() {
+    void getAlertsSummary() throws Exception {
+        //打桩
+        Mockito.when(alertService.getAlertsSummary()).thenReturn(new AlertSummary());
+
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .get("/api/alerts/summary")
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(content().json("{\"data\":{\"total\":0,\"dealNum\":0,\"rate\":0.0,\"priorityWarningNum\":0,\"priorityCriticalNum\":0,\"priorityEmergencyNum\":0},\"msg\":null,\"code\":0}"))
+                .andReturn();
     }
 
     @Test
-    void addNewAlertReport() {
+    void addNewAlertReport() throws Exception {
+
+        mockMvc.perform(
+                        MockMvcRequestBuilders
+                                .post("/api/alerts/report")
+                                .contentType(MediaType.APPLICATION_JSON)
+                                .content(GsonUtil.toJson(AlertReport.builder().build()))
+                )
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value((int) CommonConstants.SUCCESS_CODE))
+                .andExpect(content().json("{\"data\":null,\"msg\":\"Add report success\",\"code\":0}"))
+                .andReturn();
     }
 }