package jnpf.fileStorage.dms;
|
|
import jnpf.dmsEntity.preview.enums.DmsPreviewFormat;
|
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
|
import jnpf.dmsEntity.storage.model.DmsPersistentPdfSource;
|
import jnpf.exception.DataException;
|
import jnpf.onlyoffice.DmsPreviewPdfApi;
|
import org.dromara.x.file.storage.core.FileInfo;
|
import org.dromara.x.file.storage.core.FileStorageService;
|
import org.dromara.x.file.storage.core.platform.MinioFileStorage;
|
import org.junit.After;
|
import org.junit.Test;
|
|
import java.util.Collections;
|
import java.util.UUID;
|
import java.util.concurrent.Executors;
|
|
import static org.junit.Assert.assertEquals;
|
import static org.junit.Assert.assertThrows;
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
public class DmsOnlyOfficePreparedFileServiceTest {
|
private final FileStorageService storage = mock(FileStorageService.class);
|
private final DmsPreviewPdfApi previewPdfApi = mock(DmsPreviewPdfApi.class);
|
private final DmsOnlyOfficePreparedFileService service = new DmsOnlyOfficePreparedFileService(
|
storage, previewPdfApi, Executors.newSingleThreadScheduledExecutor(),
|
ignored -> "source-ticket", ignored -> { });
|
|
@After
|
public void tearDown() {
|
service.shutdown();
|
}
|
|
@Test
|
public void acceptsExplicitPdfPreviewWithoutCustomBarcode() {
|
DmsFilePreviewTicket ticket = validTicket();
|
ticket.setPreviewFormat(DmsPreviewFormat.PDF);
|
MinioFileStorage minio = mock(MinioFileStorage.class);
|
when(minio.getBucketName()).thenReturn("jnpf-dms");
|
when(storage.getFileStorageVerify("seaweedfs-1")).thenReturn(minio);
|
when(previewPdfApi.render(any())).thenThrow(new DataException("render-called"));
|
|
DataException error = assertThrows(DataException.class, () -> service.prepare(ticket));
|
|
assertEquals("render-called", error.getMessage());
|
verify(previewPdfApi).render(any());
|
}
|
|
@Test
|
public void usesExistingPersistentPdfWithoutCallingRenderer() {
|
DmsFilePreviewTicket ticket = validTicket();
|
ticket.setPreviewFormat(DmsPreviewFormat.PDF);
|
DmsPersistentPdfSource pdf = new DmsPersistentPdfSource();
|
pdf.setObjectKey("dms-derived/pdf/018f0000/result.pdf");
|
pdf.setSizeBytes(321L);
|
ticket.setPdfSource(pdf);
|
MinioFileStorage minio = mock(MinioFileStorage.class);
|
when(minio.getBucketName()).thenReturn("jnpf-dms");
|
when(storage.getFileStorageVerify("seaweedfs-1")).thenReturn(minio);
|
when(storage.exists(any(FileInfo.class))).thenReturn(true);
|
|
DmsFilePreviewTicket prepared = service.prepare(ticket);
|
|
assertEquals("result.pdf", prepared.getObjectName());
|
assertEquals("dms-derived/pdf/018f0000/", prepared.getObjectPath());
|
assertEquals("quality.pdf", prepared.getFileName());
|
assertEquals(Long.valueOf(321L), prepared.getSizeBytes());
|
verify(previewPdfApi, never()).render(any());
|
}
|
|
@Test
|
public void missingPersistentPdfFallsBackToExistingRenderer() {
|
DmsFilePreviewTicket ticket = validTicket();
|
ticket.setPreviewFormat(DmsPreviewFormat.PDF);
|
DmsPersistentPdfSource pdf = new DmsPersistentPdfSource();
|
pdf.setObjectKey("dms-derived/pdf/018f0000/missing.pdf");
|
pdf.setSizeBytes(321L);
|
pdf.setSourceChecksumSha256("a".repeat(64));
|
ticket.setPdfSource(pdf);
|
MinioFileStorage minio = mock(MinioFileStorage.class);
|
when(minio.getBucketName()).thenReturn("jnpf-dms");
|
when(storage.getFileStorageVerify("seaweedfs-1")).thenReturn(minio);
|
when(storage.exists(any(FileInfo.class))).thenReturn(false);
|
when(previewPdfApi.render(any())).thenThrow(new DataException("fallback-render-called"));
|
|
DmsPersistentPdfRecoveryReporter reporter = mock(DmsPersistentPdfRecoveryReporter.class);
|
DmsOnlyOfficePreparedFileService reportingService = new DmsOnlyOfficePreparedFileService(
|
storage, previewPdfApi, reporter, Executors.newSingleThreadScheduledExecutor(),
|
ignored -> "source-ticket", ignored -> { });
|
DataException error;
|
try {
|
error = assertThrows(DataException.class, () -> reportingService.prepare(ticket));
|
} finally {
|
reportingService.shutdown();
|
}
|
|
assertEquals("fallback-render-called", error.getMessage());
|
verify(previewPdfApi).render(any());
|
verify(reporter).report(ticket, "持久化 PDF 对象不存在");
|
}
|
|
@Test
|
public void rejectsSourcePreviewWithoutCustomBarcode() {
|
DmsFilePreviewTicket ticket = validTicket();
|
|
assertThrows(DataException.class, () -> service.prepare(ticket));
|
|
verify(previewPdfApi, never()).render(any());
|
}
|
|
private DmsFilePreviewTicket validTicket() {
|
DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
|
ticket.setFileVersionId(UUID.fromString("018f0000-0000-7000-8000-000000000702"));
|
ticket.setTenantId("tenant-1");
|
ticket.setUserId("user-1");
|
ticket.setFileName("quality.docx");
|
ticket.setStoragePlatform("seaweedfs-1");
|
ticket.setBucketName("jnpf-dms");
|
ticket.setObjectPath("2026/09/quality/");
|
ticket.setObjectName("quality.docx");
|
ticket.setSizeBytes(1024L);
|
ticket.setWatermarks(Collections.emptyList());
|
ticket.setExpiresAt(System.currentTimeMillis() + 300_000L);
|
return ticket;
|
}
|
}
|
