package jnpf.fileStorage.dms;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
import jnpf.base.ActionResult;
|
import jnpf.dms.DmsResourceApi;
|
import jnpf.dmsEntity.storage.approval.*;
|
import jnpf.dmsEntity.storage.form.*;
|
import jnpf.exception.DataException;
|
import org.junit.Test;
|
import org.mockito.ArgumentCaptor;
|
import org.springframework.web.multipart.MultipartFile;
|
import java.util.*;
|
import java.util.function.Function;
|
import static org.junit.Assert.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.*;
|
|
public class DmsWorkflowFileCreateServiceTest {
|
DmsResourceApi api = mock(DmsResourceApi.class);
|
DmsFileStorageService storage = mock(DmsFileStorageService.class);
|
DmsCommonFileReader reader = mock(DmsCommonFileReader.class);
|
DmsApprovalPageMergeService approvalPages = mock(DmsApprovalPageMergeService.class);
|
DmsQianpiyeDocxWriter qianpiyeWriter = mock(DmsQianpiyeDocxWriter.class);
|
MultipartFile file = mock(MultipartFile.class);
|
MultipartFile approvalPage = mock(MultipartFile.class);
|
MultipartFile merged = mock(MultipartFile.class);
|
DmsWorkflowFileCreateService service = new DmsWorkflowFileCreateService(
|
api, storage, reader, approvalPages, qianpiyeWriter, new ObjectMapper());
|
DmsApprovedItem item(String key, String type) {
|
DmsApprovedItem i = new DmsApprovedItem(); i.setSourceKey(key); i.setChangeType(type); i.setDocumentName(key);
|
i.setFileJson("[{\"url\":\"/api/file/Image/annex/" + key + ".docx\"}]");
|
i.setDocumentId(UUID.randomUUID()); i.setFileVersionId(UUID.randomUUID()); return i;
|
}
|
DmsApprovedBatch batch(DmsApprovedItem... items) {
|
DmsApprovedBatch b = new DmsApprovedBatch(); b.setId(UUID.randomUUID()); b.setFlowId("flow"); b.setTaskId("task"); b.setAuditType("xinzeng_shenhe");
|
b.setSourceDigest("digest"); b.setItems(Arrays.asList(items)); b.setStatus("WAITING"); return b;
|
}
|
ActionResult<DmsApprovedBatch> success(DmsApprovedBatch batch) {
|
ActionResult<DmsApprovedBatch> r = new ActionResult<>(); r.setCode(200); r.setData(batch); return r;
|
}
|
@SuppressWarnings("unchecked") void storing() {
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv -> {
|
Function<MultipartFile, DmsFileUploadRegisterForm> operation = inv.getArgument(3); return operation.apply(file);
|
});
|
when(storage.storeWorkflowContent(any(), any(), eq(file))).thenAnswer(inv -> {
|
DmsFileUploadRegisterForm c = new DmsFileUploadRegisterForm(); c.setDocumentId(inv.getArgument(0));
|
c.setFileVersionId(inv.getArgument(1)); c.setFileObjectId(UUID.randomUUID()); return c;
|
});
|
}
|
@Test public void consumesDmsPushedSetAndCallsOnlyRegistrationAfterStoringAllChanges() {
|
DmsApprovedItem main = item("main", "REVISE"), record = item("record", "CREATE"), reuse = item("reuse", "REUSE");
|
DmsApprovedBatch b = batch(main, record, reuse); storing();
|
when(api.registerWorkflowUpload(eq("jnpf-file"), eq("sign"), any())).thenReturn(success(b));
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), "sign"));
|
ArgumentCaptor<DmsApprovedRegistrationCallback> callback = ArgumentCaptor.forClass(DmsApprovedRegistrationCallback.class);
|
org.mockito.InOrder order = inOrder(api, storage);
|
order.verify(storage).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), file);
|
order.verify(storage).storeWorkflowContent(record.getDocumentId(), record.getFileVersionId(), file);
|
order.verify(api).registerWorkflowUpload(eq("jnpf-file"), eq("sign"), callback.capture());
|
assertEquals(b.getId(), callback.getValue().getBatchId());
|
assertEquals(2, callback.getValue().getFiles().size());
|
assertEquals("digest", callback.getValue().getSourceDigest());
|
verify(api, never()).prepareWorkflowFile(any(), any());
|
verify(api, never()).activateWorkflowFile(any(), any());
|
}
|
@Test public void secondStorageFailureNeverCallsRegistrationOrConfirmation() {
|
DmsApprovedBatch b = batch(item("main", "CREATE"), item("record", "CREATE")); storing();
|
when(storage.storeWorkflowContent(any(), any(), eq(file))).thenReturn(new DmsFileUploadRegisterForm()).thenThrow(new DataException("storage failed"));
|
assertThrows(DataException.class, () -> service.process(DmsApprovedRegistrationForm.from(b), null));
|
verifyNoInteractions(api);
|
}
|
@Test public void missingProgressIdentityCannotStartFileProcessing() {
|
DmsApprovedRegistrationForm command = DmsApprovedRegistrationForm.from(batch(item("main", "CREATE"))); command.setBatchId(null);
|
assertThrows(DataException.class, () -> service.process(command, null)); verifyNoInteractions(api, reader, storage);
|
}
|
@Test public void allReusedFilesStillConfirmTheBatchWithoutAnySourceRead() {
|
DmsApprovedBatch b = batch(item("main", "REUSE"));
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), null)); verifyNoInteractions(reader, storage);
|
verify(api, never()).prepareWorkflowFile(any(), any());
|
}
|
@Test public void missingWrongOrIncompleteCallbackCannotReportSuccess() {
|
DmsApprovedBatch b = batch(item("main", "REUSE")); DmsApprovedRegistrationForm form = DmsApprovedRegistrationForm.from(b);
|
assertThrows(DataException.class, () -> service.process(form, null));
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
b.setTaskId("other"); assertThrows(DataException.class, () -> service.process(form, null));
|
b.setTaskId("task"); b.setId(UUID.randomUUID()); assertThrows(DataException.class, () -> service.process(form, null));
|
}
|
@Test public void rejectsChoosingOnlyOneAttachmentFromMultipleFiles() {
|
assertThrows(DataException.class, () -> service.singleFileUrl("[{\"url\":\"a\"},{\"url\":\"b\"}]"));
|
assertEquals("a", service.singleFileUrl("[{\"url\":\"a\"}]"));
|
}
|
|
@Test public void prependsConfiguredApprovalPageBeforeStoringTheNewMainDocument() {
|
DmsApprovedItem main = item("main", "CREATE");
|
main.setApprovalPageFileJson("[{\"url\":\"/api/file/Image/annex/approval.pdf\"}]");
|
DmsApprovedBatch b = batch(main);
|
when(file.getOriginalFilename()).thenReturn("main.docx");
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv -> {
|
Function<MultipartFile, DmsFileUploadRegisterForm> operation = inv.getArgument(3);
|
return operation.apply(inv.<String>getArgument(0).contains("approval.pdf") ? approvalPage : file);
|
});
|
when(approvalPages.replaceOrPrepend(file, approvalPage)).thenReturn(merged);
|
when(storage.storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged))
|
.thenReturn(new DmsFileUploadRegisterForm());
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), null));
|
|
verify(approvalPages).replaceOrPrepend(file, approvalPage);
|
verify(storage).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged);
|
verify(storage, never()).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), file);
|
}
|
|
@Test public void configuredApprovalPageDoesNotBlockImageRegistration() {
|
DmsApprovedItem main = item("PIC-SMP100-01000", "CREATE");
|
main.setApprovalPageFileJson("[{\"url\":\"/api/file/Image/annex/approval.docx\"}]");
|
DmsApprovedBatch b = batch(main);
|
when(file.getOriginalFilename()).thenReturn("图片测试PIC-SMP100-01000.png");
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv -> {
|
Function<MultipartFile, DmsFileUploadRegisterForm> operation = inv.getArgument(3);
|
return operation.apply(file);
|
});
|
when(storage.storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), file))
|
.thenReturn(new DmsFileUploadRegisterForm());
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), null));
|
|
verify(reader, times(1)).withFile(any(), any(), any(), any());
|
verifyNoInteractions(approvalPages, qianpiyeWriter);
|
verify(storage).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), file);
|
}
|
|
@Test public void originalDocxTagsAreRenderedWithoutAConfiguredApprovalPage() {
|
DmsApprovedItem main = item("main", "CREATE");
|
main.setApprovalValuesByTag(java.util.Map.of("dms.qianpiye.form.wenjian_bianhao", "DOC-001"));
|
main.setApprovalFlowVersionId("flow-version");
|
DmsApprovedBatch b = batch(main);
|
when(file.getOriginalFilename()).thenReturn("main.docx");
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv ->
|
inv.<Function<MultipartFile, DmsFileUploadRegisterForm>>getArgument(3).apply(file));
|
when(qianpiyeWriter.render(file, main.getApprovalValuesByTag(), "flow-version")).thenReturn(merged);
|
when(storage.storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged))
|
.thenReturn(new DmsFileUploadRegisterForm());
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), null));
|
|
verify(qianpiyeWriter).render(file, main.getApprovalValuesByTag(), "flow-version");
|
verifyNoInteractions(approvalPages);
|
verify(storage).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged);
|
verify(reader, times(1)).withFile(any(), any(), any(), any());
|
}
|
|
@Test public void configuredApprovalPageReadFailureNeverStoresTheOriginalOrCallsBackDms() {
|
DmsApprovedItem main = item("main", "CREATE");
|
main.setApprovalPageFileJson("[{\"url\":\"/api/file/Image/annex/missing.pdf\"}]");
|
DmsApprovedRegistrationForm form = DmsApprovedRegistrationForm.from(batch(main));
|
when(file.getOriginalFilename()).thenReturn("main.docx");
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv -> {
|
String url = inv.getArgument(0);
|
if (url.contains("missing.pdf")) throw new DataException("file-common文件不存在");
|
Function<MultipartFile, DmsFileUploadRegisterForm> operation = inv.getArgument(3);
|
return operation.apply(file);
|
});
|
|
assertThrows(DataException.class, () -> service.process(form, null));
|
|
verify(approvalPages, never()).replaceOrPrepend(any(), any());
|
verifyNoInteractions(storage, api);
|
}
|
|
@Test public void taggedRevisionReplacesTheExistingApprovalPage() {
|
DmsApprovedItem main = item("main", "REVISE");
|
main.setApprovalPageFileJson("[{\"url\":\"/api/file/Image/annex/approval.docx\"}]");
|
DmsApprovedBatch b = batch(main);
|
when(file.getOriginalFilename()).thenReturn("main.docx");
|
when(approvalPage.getOriginalFilename()).thenReturn("approval.docx");
|
when(reader.withFile(any(), any(), any(), any())).thenAnswer(inv -> {
|
Function<MultipartFile, DmsFileUploadRegisterForm> operation = inv.getArgument(3);
|
return operation.apply(inv.<String>getArgument(0).contains("approval.docx") ? approvalPage : file);
|
});
|
main.setApprovalValuesByTag(Collections.singletonMap(
|
"dms.qianpiye.flow.approve-1.approver", "张三"));
|
main.setApprovalFlowVersionId("flow-version");
|
when(qianpiyeWriter.render(approvalPage, main.getApprovalValuesByTag(),
|
main.getApprovalFlowVersionId())).thenReturn(merged);
|
when(approvalPages.withoutApprovalPages(file)).thenReturn(file);
|
when(qianpiyeWriter.render(file, main.getApprovalValuesByTag(),
|
main.getApprovalFlowVersionId())).thenReturn(file);
|
when(approvalPages.replaceOrPrepend(file, merged)).thenReturn(merged);
|
when(storage.storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged))
|
.thenReturn(new DmsFileUploadRegisterForm());
|
when(api.registerWorkflowUpload(eq("jnpf-file"), isNull(), any())).thenReturn(success(b));
|
|
assertSame(b, service.process(DmsApprovedRegistrationForm.from(b), null));
|
|
verify(qianpiyeWriter).render(approvalPage, main.getApprovalValuesByTag(),
|
main.getApprovalFlowVersionId());
|
verify(qianpiyeWriter).render(file, main.getApprovalValuesByTag(),
|
main.getApprovalFlowVersionId());
|
verify(approvalPages).replaceOrPrepend(file, merged);
|
verify(storage).storeWorkflowContent(main.getDocumentId(), main.getFileVersionId(), merged);
|
verify(reader, times(2)).withFile(any(), any(), any(), any());
|
}
|
}
|
