feat(fleet): enforce reviewed trips and original fee coverage for dispatch

2026-10-03 04:18:52 +08:00
parent c50e9f97d1
commit 5e683da3b9
162 changed files with 9642 additions and 218 deletions
@@ -0,0 +1,52 @@
+package com.baiyue.file.service;
+
+import com.baiyue.common.core.exception.BizException;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
+import org.springframework.mock.web.MockMultipartFile;
+import org.springframework.test.util.ReflectionTestUtils;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.nio.file.Path;
+
+import static org.assertj.core.api.Assertions.*;
+import static org.mockito.Mockito.*;
+
+class DriverLicenseFileTest {
+    @TempDir Path root;
+
+    @Test void licensesArePlatformOwnedWithPositiveDriverId() {
+        var category = BusinessFileCategory.require("FLEET_DRIVER_LICENSE");
+        category.validateOwner(0);
+        assertThatThrownBy(() -> category.validateOwner(1)).isInstanceOf(BizException.class).hasMessageContaining("平台");
+        assertThatThrownBy(() -> category.validateOwner(-1)).isInstanceOf(BizException.class);
+        assertThat(category.storageBusinessId("71")).isEqualTo("71");
+        assertThatThrownBy(() -> category.storageBusinessId("0")).isInstanceOf(BizException.class);
+    }
+
+    @Test void storesRealPdfAndRejectsWrongMimeAndSpoofedPdf() throws Exception {
+        var storage = storage();
+        var pdf = new MockMultipartFile("file", "license.pdf", "application/pdf", "%PDF-1.7\nlicense".getBytes());
+        var result = storage.store(0, "71", pdf, BusinessFileCategory.FLEET_DRIVER_LICENSE);
+        assertThat(storage.read(result.objectKey())).isEqualTo(pdf.getBytes());
+        assertThatThrownBy(() -> storage.store(0, "71", new MockMultipartFile("file", "license.txt", "text/plain", "text".getBytes()), BusinessFileCategory.FLEET_DRIVER_LICENSE))
+            .isInstanceOf(BizException.class).hasMessageContaining("仅支持");
+        assertThatThrownBy(() -> storage.store(0, "71", new MockMultipartFile("file", "license.pdf", "application/pdf", "not pdf".getBytes()), BusinessFileCategory.FLEET_DRIVER_LICENSE))
+            .isInstanceOf(BizException.class).hasMessageContaining("不是有效PDF");
+    }
+
+    @Test void rejectsMoreThanTenMibBeforeReadingBytes() {
+        var file = mock(MultipartFile.class);
+        when(file.isEmpty()).thenReturn(false);
+        when(file.getSize()).thenReturn(10485761L);
+        assertThatThrownBy(() -> storage().store(0, "71", file, BusinessFileCategory.FLEET_DRIVER_LICENSE))
+            .isInstanceOf(BizException.class).hasMessageContaining("10MB");
+    }
+
+    private BusinessFileStorage storage() {
+        var storage = new BusinessFileStorage();
+        ReflectionTestUtils.setField(storage, "configuredRoot", root);
+        storage.initializeRoot();
+        return storage;
+    }
+}
@@ -0,0 +1,25 @@
+package com.baiyue.file.service;
+
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
+import org.springframework.mock.web.MockMultipartFile;
+import org.springframework.test.util.ReflectionTestUtils;
+import java.nio.file.Path;
+import static org.assertj.core.api.Assertions.*;
+
+class VehicleCarePhotoTest {
+    @TempDir Path root;
+    @Test void recordPhotosArePlatformOwnedAndRequireRealImageBytes() throws Exception {
+        var category = BusinessFileCategory.require("FLEET_VEHICLE_RECORD_PHOTO");
+        category.validateOwner(0);
+        assertThatThrownBy(() -> category.validateOwner(7)).hasMessageContaining("平台");
+        assertThatThrownBy(() -> category.validateOwner(-1)).hasMessageContaining("平台");
+        assertThatThrownBy(() -> category.storageBusinessId("0")).hasMessageContaining("有效");
+        var storage = new BusinessFileStorage(); ReflectionTestUtils.setField(storage, "configuredRoot", root); storage.initializeRoot();
+        byte[] bytes = {(byte)0x89, 0x50, 0x4e, 0x47, 13, 10, 26, 10};
+        var result = storage.store(0, "71", new MockMultipartFile("file", "meter.png", "image/png", bytes), category);
+        assertThat(storage.read(result.objectKey())).isEqualTo(bytes);
+        assertThatThrownBy(() -> storage.store(0, "71", new MockMultipartFile("file", "fake.png", "image/png", "not-an-image".getBytes()), category)).hasMessageContaining("不是有效图片");
+        assertThatThrownBy(() -> storage.store(0, "71", new MockMultipartFile("file", "doc.pdf", "application/pdf", "%PDF-test".getBytes()), category)).hasMessageContaining("仅支持JPG");
+    }
+}