feat(firstleg): close phase3 outbound customs and schedule gates

2026-08-08 22:36:41 +08:00
parent 789d97675b
commit 66d1aa0d3d
81 changed files with 5196 additions and 36 deletions
@@ -0,0 +1,58 @@
+package com.baiyue.file.service;
+
+import com.baiyue.common.core.exception.BizException;
+import com.baiyue.file.entity.FileObjectEntity;
+import com.baiyue.file.mapper.FileObjectMapper;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.*;
+import org.springframework.mock.web.MockMultipartFile;
+
+import static org.assertj.core.api.Assertions.*;
+import static org.mockito.ArgumentMatchers.*;
+import static org.mockito.Mockito.*;
+
+class InternalBusinessFileServiceTest {
+    @Mock FileObjectMapper files;
+    @Mock BusinessFileStorage storage;
+    @InjectMocks InternalBusinessFileService service;
+
+    @BeforeEach void setUp() { MockitoAnnotations.openMocks(this); }
+
+    @Test void customsDocumentUploadPersistsExactBusinessOwnership() {
+        MockMultipartFile file = new MockMultipartFile("file", "invoice.txt", "text/plain", "invoice".getBytes());
+        when(storage.storeLoading(10001L, 41L, file)).thenReturn(new BusinessFileStorage.Stored(
+            "business-files/10001/41/key.txt", "text/plain", 7, "a".repeat(64), "invoice.txt"));
+        when(files.insert(any(FileObjectEntity.class))).thenAnswer(invocation -> {
+            ((FileObjectEntity) invocation.getArgument(0)).setId(71L); return 1;
+        });
+
+        var result = service.upload(10001L, "CUSTOMS_DOCUMENT", "41", 7L, file);
+
+        assertThat(result.fileId()).isEqualTo(71L);
+        assertThat(result.businessType()).isEqualTo("CUSTOMS_DOCUMENT");
+        assertThat(result.businessId()).isEqualTo("41");
+        assertThat(result.contentType()).isEqualTo("text/plain");
+    }
+
+    @Test void unsupportedBusinessTypeFailsClosed() {
+        assertThatThrownBy(() -> service.upload(10001L, "ARBITRARY", "41", 7L,
+            new MockMultipartFile("file", "a.txt", "text/plain", "x".getBytes())))
+            .isInstanceOf(BizException.class).hasMessageContaining("不支持");
+        verifyNoInteractions(files, storage);
+    }
+
+    @Test void downloadUsesValidatedMetadataObjectKey() {
+        FileObjectEntity value = new FileObjectEntity(); value.setId(71L); value.setTenantId(10001L);
+        value.setBusinessType("CUSTOMS_RECEIPT"); value.setBusinessId("61"); value.setOriginalName("release.txt");
+        value.setObjectKey("business-files/10001/61/release.txt"); value.setContentType("text/plain");
+        when(files.selectOne(any())).thenReturn(value);
+        when(storage.read(value.getObjectKey())).thenReturn("released".getBytes());
+
+        var result = service.download(10001L, "CUSTOMS_RECEIPT", "61", 71L);
+
+        assertThat(result.fileName()).isEqualTo("release.txt");
+        assertThat(result.content()).isEqualTo("released".getBytes());
+        verify(storage).read(value.getObjectKey());
+    }
+}