package jnpf.bizcommon.onlyoffice.service.impl;
|
|
import jnpf.dmsEntity.preview.model.DmsFilePreviewTicket;
|
import jnpf.exception.DataException;
|
import jnpf.util.JsonUtil;
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
import org.apache.pdfbox.pdmodel.PDPage;
|
import org.junit.Test;
|
import org.mockito.ArgumentCaptor;
|
|
import java.io.ByteArrayOutputStream;
|
import java.util.UUID;
|
|
import static org.junit.Assert.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
public class DmsOnlyOfficePageCountServiceTest {
|
private static final UUID VERSION = UUID.randomUUID();
|
private final OnlyOfficeDmsPrintRenderer renderer = mock(OnlyOfficeDmsPrintRenderer.class);
|
|
@Test
|
public void countsRenderedPagesFromPreparedPreviewSource() throws Exception {
|
DmsFilePreviewTicket ticket = ticket();
|
byte[] pdf;
|
try (PDDocument document = new PDDocument();
|
ByteArrayOutputStream output = new ByteArrayOutputStream()) {
|
for (int page = 0; page < 19; page++) document.addPage(new PDPage());
|
document.save(output);
|
pdf = output.toByteArray();
|
}
|
when(renderer.render(any(), eq("preview-ticket"))).thenReturn(pdf);
|
DmsOnlyOfficePageCountService service = service(ticket);
|
|
assertEquals(19, service.getPageCount(VERSION, "preview-ticket"));
|
|
ArgumentCaptor<DmsFilePreviewTicket> source = ArgumentCaptor.forClass(DmsFilePreviewTicket.class);
|
verify(renderer).render(source.capture(), eq("preview-ticket"));
|
assertEquals(ticket, source.getValue());
|
verify(renderer, never()).renderBase(any(), anyString());
|
}
|
|
@Test
|
public void rejectsInvalidTicketsBeforeRendering() {
|
for (int scenario = 0; scenario < 4; scenario++) {
|
DmsFilePreviewTicket ticket = ticket();
|
switch (scenario) {
|
case 0: ticket.setExpiresAt(System.currentTimeMillis() - 1); break;
|
case 1: ticket.setScope(DmsFilePreviewTicket.PRINT_SCOPE); break;
|
case 2: ticket.setFileVersionId(UUID.randomUUID()); break;
|
case 3: ticket.setFileName("manual.txt"); break;
|
}
|
assertRejected(service(ticket));
|
}
|
assertRejected(new DmsOnlyOfficePageCountService(renderer, value -> null));
|
verifyNoInteractions(renderer);
|
}
|
|
@Test
|
public void rejectsFailedConversionAndInvalidPdf() {
|
when(renderer.render(any(), anyString())).thenThrow(new DataException("转换失败"));
|
assertRejected(service(ticket()));
|
reset(renderer);
|
when(renderer.render(any(), anyString())).thenReturn(new byte[]{1, 2, 3});
|
assertRejected(service(ticket()));
|
}
|
|
private void assertRejected(DmsOnlyOfficePageCountService service) {
|
try {
|
service.getPageCount(VERSION, "preview-ticket");
|
fail("expected page count failure");
|
} catch (DataException expected) {
|
assertNotNull(expected.getMessage());
|
}
|
}
|
|
private DmsOnlyOfficePageCountService service(DmsFilePreviewTicket ticket) {
|
return new DmsOnlyOfficePageCountService(renderer, value -> JsonUtil.getObjectToString(ticket));
|
}
|
|
private DmsFilePreviewTicket ticket() {
|
DmsFilePreviewTicket ticket = new DmsFilePreviewTicket();
|
ticket.setScope(DmsFilePreviewTicket.SCOPE);
|
ticket.setTenantId("tenant-1");
|
ticket.setUserId("user-1");
|
ticket.setFileVersionId(VERSION);
|
ticket.setFileName("玻思韬DMS调研报告.docx");
|
ticket.setStoragePlatform("seaweedfs-1");
|
ticket.setBucketName("jnpf-dms");
|
ticket.setObjectPath("prepared-preview/");
|
ticket.setObjectName("prepared.docx");
|
ticket.setExpiresAt(System.currentTimeMillis() + 60_000);
|
return ticket;
|
}
|
}
|
