提交

lastleg-service/src/test/java/com/baiyue/lastleg/dispatch/controller/DispatchQueryControllerTest.java

@@ -0,0 +1,53 @@
+package com.baiyue.lastleg.dispatch.controller;
+
+import com.baiyue.common.core.api.PageResult;
+import com.baiyue.common.security.AuthenticatedUser;
+import com.baiyue.lastleg.dispatch.dto.DispatchRequests;
+import com.baiyue.lastleg.dispatch.service.DispatchService;
+import org.junit.jupiter.api.BeforeEach;
+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.test.util.ReflectionTestUtils;
+import org.springframework.web.server.ResponseStatusException;
+
+import java.util.List;
+import java.util.Set;
+
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class DispatchQueryControllerTest {
+    @Mock DispatchService service;
+    private DispatchQueryController controller;
+
+    @BeforeEach
+    void setUp() {
+        controller = new DispatchQueryController();
+        ReflectionTestUtils.setField(controller, "service", service);
+    }
+
+    @Test
+    void platformPageQueriesAllCustomersWithoutCustomerId() {
+        DispatchRequests.Query query = new DispatchRequests.Query(1, 20, null, null, null, null);
+        when(service.page(null, query)).thenReturn(new PageResult<>(List.of(), 1, 20, 0, 0));
+
+        var response = controller.page(platform(), query);
+
+        verify(service).page(null, query);
+        assertThat(response.code()).isEqualTo("200");
+        assertThat(response.data().records()).isEmpty();
+        assertThatThrownBy(() -> controller.page(null, query))
+            .isInstanceOf(ResponseStatusException.class)
+            .satisfies(error -> org.assertj.core.api.Assertions.assertThat(
+                ((ResponseStatusException) error).getStatusCode().value()).isEqualTo(401));
+    }
+
+    private static AuthenticatedUser platform() {
+        return new AuthenticatedUser(7L, 0L, 0L, 8L, "operator", "PLATFORM", "WEB", Set.of());
+    }
+}