优化

2026-08-07 17:14:40 +08:00
parent f4c3a95772
commit 91f31d9567
20 changed files with 354 additions and 7 deletions
@@ -0,0 +1,89 @@
+package com.baiyue.firstleg.service;
+
+import com.baiyue.common.core.api.ResultCode;
+import com.baiyue.common.core.exception.BizException;
+import com.baiyue.firstleg.client.PrintPdfClient;
+import com.baiyue.firstleg.dto.InboundCartonLabels;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.ArgumentCaptor;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.test.util.ReflectionTestUtils;
+
+import java.math.BigDecimal;
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+class InboundCartonLabelPrintServiceTest {
+    private final InboundCartonLabelService labels = mock(InboundCartonLabelService.class);
+    private final PrintPdfClient print = mock(PrintPdfClient.class);
+    private InboundCartonLabelPrintService service;
+
+    @BeforeEach
+    void setUp() {
+        service = new InboundCartonLabelPrintService(labels, print, new ObjectMapper());
+        ReflectionTestUtils.setField(service, "internalToken", "internal-token");
+    }
+
+    @Test
+    void forwardsStableLabelsAndOnlySafePdfHeaders() {
+        when(labels.list(10001L, 70001L, List.of(1))).thenReturn(batch());
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_PDF);
+        headers.setContentDispositionFormData("attachment", "labels.pdf");
+        headers.set("X-Print-Page-Count", "1");
+        headers.set("X-Internal-Service-Token", "must-not-leak");
+        when(print.generateInboundMarks(org.mockito.ArgumentMatchers.anyString(),
+            org.mockito.ArgumentMatchers.anyString(), org.mockito.ArgumentMatchers.any()))
+            .thenReturn(new ResponseEntity<>(new byte[] {1, 2, 3}, headers,
+                org.springframework.http.HttpStatus.OK));
+
+        ResponseEntity<byte[]> response = service.generate(10001L, 70001L, List.of(1), "request-1");
+
+        ArgumentCaptor<PrintPdfClient.InboundMarkPdfRequest> request =
+            ArgumentCaptor.forClass(PrintPdfClient.InboundMarkPdfRequest.class);
+        verify(print).generateInboundMarks(org.mockito.ArgumentMatchers.eq("internal-token"),
+            org.mockito.ArgumentMatchers.eq("request-1"), request.capture());
+        assertThat(request.getValue().downloadFileName()).isEqualTo("PCHK2608040174-19件-箱唛.pdf");
+        assertThat(request.getValue().labels()).singleElement().satisfies(label -> {
+            assertThat(label.qrContent()).isEqualTo("PCHK2608040174#19-1");
+            assertThat(label.cartonText()).isEqualTo("CTNS:19-1");
+        });
+        assertThat(response.getBody()).containsExactly(1, 2, 3);
+        assertThat(response.getHeaders().getFirst("X-Print-Page-Count")).isEqualTo("1");
+        assertThat(response.getHeaders()).doesNotContainKey("X-Internal-Service-Token");
+    }
+
+    @Test
+    void convertsPrintJsonBusinessFailureInsteadOfReturningItAsPdf() throws Exception {
+        when(labels.list(10001L, 70001L, null)).thenReturn(batch());
+        byte[] error = "{\"code\":\"A0422\",\"message\":\"标签字段超出模板限制\","
+            .concat("\"data\":null,\"requestId\":\"req-2\",\"traceId\":\"trace-2\"}")
+            .getBytes(java.nio.charset.StandardCharsets.UTF_8);
+        when(print.generateInboundMarks(org.mockito.ArgumentMatchers.anyString(),
+            org.mockito.ArgumentMatchers.isNull(), org.mockito.ArgumentMatchers.any()))
+            .thenReturn(ResponseEntity.ok().contentType(MediaType.APPLICATION_JSON).body(error));
+
+        assertThatThrownBy(() -> service.generate(10001L, 70001L, null, null))
+            .isInstanceOf(BizException.class)
+            .hasMessageContaining("标签字段超出模板限制")
+            .extracting(value -> ((BizException) value).getResultCode())
+            .isEqualTo(ResultCode.VALIDATION_FAILED);
+    }
+
+    private InboundCartonLabels.Batch batch() {
+        InboundCartonLabels.Label label = new InboundCartonLabels.Label(
+            1L, 2L, 1, "音箱", 19, "PCHK2608040174#19-1", 19, 1,
+            "50x40x30cm", new BigDecimal("50"), new BigDecimal("40"), new BigDecimal("30"),
+            "PCHK2608040174", "MCC1", "CTNS:19-1", "洛杉矶普船海卡");
+        return new InboundCartonLabels.Batch(70001L, 80001L, "PCHK2608040174", 19, List.of(label));
+    }
+}