package jnpf.dmsService.pdf;
|
|
import jnpf.base.ActionResult;
|
import jnpf.dms.DmsPersistentPdfApi;
|
import jnpf.dmsEntity.permission.entity.DmsFileVersionEntity;
|
import jnpf.dmsEntity.storage.entity.DmsFileObjectEntity;
|
import jnpf.dmsEntity.storage.pdf.DmsPdfGenerationTask;
|
import jnpf.dmsEntity.storage.vo.DmsPersistentPdfGenerateVO;
|
import jnpf.dmsMapper.DmsFileObjectMapper;
|
import jnpf.dmsMapper.DmsFileVersionMapper;
|
import jnpf.dmsMapper.DmsPdfGenerationTaskMapper;
|
import org.junit.jupiter.api.BeforeEach;
|
import org.junit.jupiter.api.Test;
|
|
import java.lang.reflect.Field;
|
import java.util.UUID;
|
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.anyString;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
class DmsPdfGenerationJobTest {
|
private static final String TENANT = "tenant-1";
|
private static final UUID VERSION_ID =
|
UUID.fromString("018f0000-0000-7000-8000-000000000702");
|
private static final String CHECKSUM = "a".repeat(64);
|
|
private final DmsPdfGenerationTaskMapper tasks = mock(DmsPdfGenerationTaskMapper.class);
|
private final DmsFileVersionMapper versions = mock(DmsFileVersionMapper.class);
|
private final DmsFileObjectMapper objects = mock(DmsFileObjectMapper.class);
|
private final DmsPersistentPdfApi fileApi = mock(DmsPersistentPdfApi.class);
|
private final DmsPdfGenerationCompletion completion = mock(DmsPdfGenerationCompletion.class);
|
private final DmsPdfGenerationJob job =
|
new DmsPdfGenerationJob(tasks, versions, objects, fileApi, completion);
|
|
@BeforeEach
|
void configure() throws Exception {
|
set("maxAttempts", 8);
|
set("leaseSeconds", 300);
|
}
|
|
@Test
|
void skipsTaskWhenAnotherWorkerOwnsTheLease() {
|
DmsPdfGenerationTask task = task(0);
|
when(tasks.claim(eq(TENANT), eq(VERSION_ID), anyString(), eq(8), eq(300)))
|
.thenReturn(0);
|
|
job.process(task);
|
|
verify(versions, never()).selectByTenantAndId(anyString(), any());
|
verify(fileApi, never()).generate(anyString(), any());
|
}
|
|
@Test
|
void generatesAndCompletesClaimedTask() {
|
DmsPdfGenerationTask task = task(0);
|
DmsFileVersionEntity version = version(CHECKSUM);
|
DmsFileObjectEntity object = object();
|
DmsPersistentPdfGenerateVO pdf = new DmsPersistentPdfGenerateVO();
|
pdf.setObjectKey("dms-derived/pdf/" + VERSION_ID + "/" + CHECKSUM + ".pdf");
|
pdf.setSizeBytes(128L);
|
pdf.setChecksumSha256("b".repeat(64));
|
pdf.setSourceChecksumSha256(CHECKSUM);
|
ActionResult<DmsPersistentPdfGenerateVO> response = new ActionResult<>();
|
response.setCode(200);
|
response.setData(pdf);
|
when(tasks.claim(eq(TENANT), eq(VERSION_ID), anyString(), eq(8), eq(300)))
|
.thenReturn(1);
|
when(versions.selectByTenantAndId(TENANT, VERSION_ID)).thenReturn(version);
|
when(objects.selectByTenantAndFileVersionId(TENANT, VERSION_ID)).thenReturn(object);
|
when(fileApi.generate(eq("jnpf-dms"), any())).thenReturn(response);
|
|
job.process(task);
|
|
verify(completion).succeed(eq(task), anyString(), eq(pdf));
|
verify(tasks, never()).fail(anyString(), any(), anyString(), anyString(),
|
org.mockito.ArgumentMatchers.anyInt(), org.mockito.ArgumentMatchers.anyBoolean());
|
}
|
|
@Test
|
void retriesWhenSourceChecksumChanged() {
|
DmsPdfGenerationTask task = task(0);
|
when(tasks.claim(eq(TENANT), eq(VERSION_ID), anyString(), eq(8), eq(300)))
|
.thenReturn(1);
|
when(versions.selectByTenantAndId(TENANT, VERSION_ID))
|
.thenReturn(version("c".repeat(64)));
|
when(objects.selectByTenantAndFileVersionId(TENANT, VERSION_ID)).thenReturn(object());
|
|
job.process(task);
|
|
verify(tasks).fail(eq(TENANT), eq(VERSION_ID), anyString(),
|
eq("持久化 PDF 源文件不存在或已变化"), eq(60), eq(false));
|
verify(fileApi, never()).generate(anyString(), any());
|
}
|
|
@Test
|
void marksTaskFailedAfterLastAttempt() {
|
DmsPdfGenerationTask task = task(7);
|
when(tasks.claim(eq(TENANT), eq(VERSION_ID), anyString(), eq(8), eq(300)))
|
.thenReturn(1);
|
when(versions.selectByTenantAndId(TENANT, VERSION_ID)).thenReturn(version(CHECKSUM));
|
when(objects.selectByTenantAndFileVersionId(TENANT, VERSION_ID)).thenReturn(object());
|
when(fileApi.generate(eq("jnpf-dms"), any())).thenThrow(new IllegalStateException("timeout"));
|
|
job.process(task);
|
|
verify(tasks).fail(eq(TENANT), eq(VERSION_ID), anyString(), eq("timeout"),
|
eq(21600), eq(true));
|
}
|
|
private DmsPdfGenerationTask task(int attempts) {
|
DmsPdfGenerationTask task = new DmsPdfGenerationTask();
|
task.setTenantId(TENANT);
|
task.setFileVersionId(VERSION_ID);
|
task.setSourceChecksumSha256(CHECKSUM);
|
task.setAttempts(attempts);
|
return task;
|
}
|
|
private DmsFileVersionEntity version(String checksum) {
|
DmsFileVersionEntity version = new DmsFileVersionEntity();
|
version.setId(VERSION_ID);
|
version.setTenantId(TENANT);
|
version.setFileName("quality.docx");
|
version.setMimeType("application/vnd.openxmlformats-officedocument.wordprocessingml.document");
|
version.setSizeBytes(1024L);
|
version.setChecksumSha256(checksum);
|
return version;
|
}
|
|
private DmsFileObjectEntity object() {
|
DmsFileObjectEntity object = new DmsFileObjectEntity();
|
object.setFileVersionId(VERSION_ID);
|
object.setTenantId(TENANT);
|
object.setStoragePlatform("seaweedfs-1");
|
object.setBucketName("jnpf-dms");
|
object.setBasePath("");
|
object.setObjectPath("dms/source/");
|
object.setObjectName("quality.docx");
|
return object;
|
}
|
|
private void set(String fieldName, Object value) throws Exception {
|
Field field = DmsPdfGenerationJob.class.getDeclaredField(fieldName);
|
field.setAccessible(true);
|
field.set(job, value);
|
}
|
}
|
