提交

lastleg-service/src/test/java/com/baiyue/lastleg/controller/LastlegOrderControllerTest.java

@@ -18,6 +18,8 @@ import org.springframework.web.server.ResponseStatusException;
 import java.util.Set;
 
 import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.ArgumentMatchers.isNull;
@@ -50,6 +52,32 @@ class LastlegOrderControllerTest {
         verify(service).page(eq("200"), eq(query));
     }
 
+    @Test
+    void exportBuildsCsvBeforeReturningTheResponse() {
+        byte[] csv = "csv".getBytes(java.nio.charset.StandardCharsets.UTF_8);
+        when(service.exportCsv("200", query)).thenReturn(csv);
+
+        var response = controller.export(user(0L, "PLATFORM"), "200", query);
+
+        assertArrayEquals(csv, response.getBody());
+        verify(service).exportCsv("200", query);
+    }
+
+    @Test
+    void platformOperatorCanExportAllCustomersWithoutFilters() {
+        byte[] csv = "all customers".getBytes(java.nio.charset.StandardCharsets.UTF_8);
+        LastlegOrderRequests.Query emptyQuery = new LastlegOrderRequests.Query(
+            null, null, null, null, null, null, null, null, null, null, null, null, null, null, null);
+        when(service.exportCsv(null, emptyQuery)).thenReturn(csv);
+
+        var response = controller.export(user(0L, "PLATFORM"), null, emptyQuery);
+
+        assertEquals(Long.valueOf(1), emptyQuery.page());
+        assertEquals(Long.valueOf(20), emptyQuery.size());
+        assertArrayEquals(csv, response.getBody());
+        verify(service).exportCsv(isNull(), eq(emptyQuery));
+    }
+
     @Test
     void iamSecurityEventsAreRejectedBeforeLoadingCompletedConsumer() {
         EventEnvelope event = mock(EventEnvelope.class);