fix(firstleg): allow optional ocean booking customer

This commit is contained in:
2026-09-09 11:21:37 +08:00
parent 81b729929c
commit f37e9ce555
6 changed files with 144 additions and 30 deletions
@@ -39,6 +39,8 @@ class OceanBookingMapperContractTest {
.flatMap(annotation -> Arrays.stream(annotation.value()))
.reduce("", (left, right) -> left + '\n' + right);
assertThat(sql)
.contains("customer_id=#{value.customerId},customer_name=#{value.customerName}")
.doesNotContain("customer_id=COALESCE", "customer_name=COALESCE")
.contains("status=#{fromStatus} AND version=#{expectedVersion}")
.contains("status='DRAFT' AND version=#{version}")
.doesNotContain("CONFIRMED", "CHANGE_PENDING", "CANCELLED", "SUBMITTED", "FAILED");
@@ -40,11 +40,12 @@ class OceanBookingDocumentServiceTest {
@Test void uploadIsIdempotentAndDoesNotUploadASecondFileOnReplay() throws Exception {
OceanBookingEntity booking = draft(2);
booking.setCustomerId(null);
MockMultipartFile file = new MockMultipartFile("file", "booking.pdf", "application/pdf", "pdf".getBytes());
when(repository.claimOperation(eq(10L), eq(7L), eq("UPLOAD_DOCUMENT"), eq("upload-1"), anyString()))
.thenReturn(true, false);
when(repository.findForUpdate(10L)).thenReturn(booking);
when(files.upload("token", 7L, 9L, "OCEAN_BOOKING_DOCUMENT", "10", file))
when(files.upload("token", 7L, 0L, "OCEAN_BOOKING_DOCUMENT", "10", file))
.thenReturn(ApiResponse.success(metadata(91L)));
when(repository.touchDraft(10L, 2, 7L, "admin")).thenReturn(1);
when(repository.completeOperation(eq(10L), eq(7L), eq("UPLOAD_DOCUMENT"), eq("upload-1"), anyString()))
@@ -75,9 +76,9 @@ class OceanBookingDocumentServiceTest {
when(repository.claimOperation(anyLong(), eq(7L), eq("UPLOAD_DOCUMENT"), anyString(), anyString()))
.thenReturn(true);
when(repository.findForUpdate(10L)).thenReturn(draft(0), draft(1));
when(files.upload("token", 7L, 9L, "OCEAN_BOOKING_DOCUMENT", "10", confirmation))
when(files.upload("token", 7L, 0L, "OCEAN_BOOKING_DOCUMENT", "10", confirmation))
.thenReturn(ApiResponse.success(metadata(91L)));
when(files.upload("token", 7L, 9L, "OCEAN_BOOKING_DOCUMENT", "10", bill))
when(files.upload("token", 7L, 0L, "OCEAN_BOOKING_DOCUMENT", "10", bill))
.thenReturn(ApiResponse.success(metadata(92L)));
when(repository.touchDraft(eq(10L), anyInt(), eq(7L), eq("admin"))).thenReturn(1);
when(repository.completeOperation(eq(10L), eq(7L), eq("UPLOAD_DOCUMENT"), anyString(), anyString()))
@@ -101,7 +102,7 @@ class OceanBookingDocumentServiceTest {
@Test void attachRejectsCrossScopeFileBeforePersisting() {
when(repository.claimOperation(any(), anyLong(), anyString(), anyString(), anyString())).thenReturn(true);
when(repository.findForUpdate(10L)).thenReturn(draft(1));
when(files.metadata("token", 9L, "OCEAN_BOOKING_DOCUMENT", "10", 92L))
when(files.metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 92L))
.thenReturn(ApiResponse.success(new OceanBookingFileClient.Metadata(92L, 19L,
"OCEAN_BOOKING_DOCUMENT", "10", "x.pdf", "hash", "application/pdf", 3L)));
@@ -144,12 +145,12 @@ class OceanBookingDocumentServiceTest {
document.setId(20L); document.setOceanBookingId(10L); document.setFileId(91L);
document.setDocumentType("BILL_OF_LADING"); document.setValid(true);
when(repository.activeDocuments(10L)).thenReturn(List.of(document));
when(files.metadata("token", 9L, "OCEAN_BOOKING_DOCUMENT", "10", 91L))
when(files.metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L))
.thenReturn(ApiResponse.success(metadata(91L)));
service.requireActiveValidDocument(booking);
verify(files).metadata("token", 9L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
verify(files).metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
}
@Test void releaseGateRejectsWhenBothActiveFileReferencesNoLongerExist() {
@@ -161,8 +162,8 @@ class OceanBookingDocumentServiceTest {
assertThatThrownBy(() -> service.requireActiveValidDocument(booking))
.isInstanceOf(BizException.class).hasMessageContaining("有效订舱确认或提单文件");
verify(files).metadata("token", 9L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
verify(files).metadata("token", 9L, "OCEAN_BOOKING_DOCUMENT", "10", 92L);
verify(files).metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
verify(files).metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 92L);
verify(repository, never()).touchDraft(anyLong(), anyInt(), anyLong(), anyString());
}
@@ -173,13 +174,48 @@ class OceanBookingDocumentServiceTest {
return document;
}
@Test void changingOrClearingCustomerKeepsPlatformFileDownloadAndReleaseScope() {
OceanBookingEntity booking = draft(3);
OceanBookingDocumentEntity document = activeDocument(20L, 91L, "BOOKING_CONFIRMATION");
document.setFileName("booking.pdf");
when(repository.find(10L)).thenReturn(booking);
when(repository.document(10L, 20L)).thenReturn(document);
when(repository.activeDocuments(10L)).thenReturn(List.of(document));
when(files.metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L))
.thenReturn(ApiResponse.success(metadata(91L)));
when(files.download("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L))
.thenReturn(org.springframework.http.ResponseEntity.ok(new byte[]{1, 2, 3}));
for (Long customer : java.util.Arrays.asList(9L, 19L, null)) {
booking.setCustomerId(customer);
service.requireActiveValidDocument(booking);
assertThat(service.download(10L, 20L).content()).containsExactly(1, 2, 3);
}
verify(files, times(6)).metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
verify(files, times(3)).download("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 91L);
}
@Test void platformFileFromAnotherBookingIsRejected() {
when(repository.claimOperation(any(), anyLong(), anyString(), anyString(), anyString())).thenReturn(true);
OceanBookingEntity booking = draft(1);
booking.setCustomerId(null);
when(repository.findForUpdate(10L)).thenReturn(booking);
when(files.metadata("token", 0L, "OCEAN_BOOKING_DOCUMENT", "10", 92L))
.thenReturn(ApiResponse.success(new OceanBookingFileClient.Metadata(92L, 0L,
"OCEAN_BOOKING_DOCUMENT", "99", "x.pdf", "hash", "application/pdf", 3L)));
assertThatThrownBy(() -> service.attach(10L,
new com.baiyue.firstleg.oceanbooking.dto.OceanBookingDtos.DocumentRequest(1, 92L, "BILL_OF_LADING"),
"other-booking", 7L, "admin"))
.isInstanceOf(BizException.class).hasMessageContaining("归属不匹配");
verify(repository, never()).insert(any(OceanBookingDocumentEntity.class));
}
private static OceanBookingEntity draft(int version) {
OceanBookingEntity booking = new OceanBookingEntity();
booking.setId(10L); booking.setCustomerId(9L); booking.setStatus("DRAFT"); booking.setVersion(version);
return booking;
}
private static OceanBookingFileClient.Metadata metadata(long fileId) {
return new OceanBookingFileClient.Metadata(fileId, 9L, "OCEAN_BOOKING_DOCUMENT", "10",
return new OceanBookingFileClient.Metadata(fileId, 0L, "OCEAN_BOOKING_DOCUMENT", "10",
"booking.pdf", "hash", "application/pdf", 3L);
}
}
@@ -40,6 +40,7 @@ class OceanBookingServiceTest {
@Mock private com.baiyue.firstleg.client.WarehousePartnerClient partners;
@Mock private FirstlegDocumentNumberGenerator documentNumbers;
@Mock private OceanBookingDocumentService documents;
@Mock private com.baiyue.firstleg.service.IamCustomerSnapshotService customers;
@InjectMocks private OceanBookingService service;
private ObjectMapper objectMapper;
@@ -47,6 +48,8 @@ class OceanBookingServiceTest {
void setUp() throws Exception {
MockitoAnnotations.openMocks(this);
objectMapper = new ObjectMapper().findAndRegisterModules();
lenient().when(customers.active(eq(100L), any())).thenReturn(
new com.baiyue.firstleg.client.IamCustomerSnapshotClient.Customer(100L, "C100", "客户", "ACTIVE", null, null));
lenient().when(documentNumbers.next("OB")).thenReturn("OB-260818-0001");
var field = OceanBookingService.class.getDeclaredField("objectMapper");
field.setAccessible(true);
@@ -76,6 +79,70 @@ class OceanBookingServiceTest {
verifyNoInteractions(priceBooks, suppliers, warehouses);
}
@Test
void customerlessCreateSaveAndReleaseRetainsOtherRequirements() throws Exception {
saveAndReleaseWithoutCustomer(false);
}
@Test
void clearingExistingCustomerPersistsNullAndAllowsRelease() throws Exception {
saveAndReleaseWithoutCustomer(true);
}
private void saveAndReleaseWithoutCustomer(boolean initialCustomer) throws Exception {
stubMasterData();
when(suppliers.booking(81L, "订舱代理")).thenReturn(
new WarehouseSupplierClient.Snapshot(81L, "订舱代理", "BOOKING", "ACTIVE", List.of("MATSON")));
var tree = objectMapper.valueToTree(completeDraft());
((com.fasterxml.jackson.databind.node.ObjectNode) tree).putNull("customerId").putNull("customerName");
DraftData emptyCustomer = objectMapper.treeToValue(tree, DraftData.class);
AtomicReference<OceanBookingEntity> stored = stubCreatedDraft();
var created = service.create(9L, "admin", "optional-create",
new CreateRequest(0, initialCustomer ? completeDraft() : emptyCustomer));
long id = created.oceanBooking().getId();
when(repository.updateDraft(any(), eq(0), eq(9L), eq("admin"))).thenAnswer(invocation -> {
OceanBookingEntity value = invocation.getArgument(0);
assertThat(value.getCustomerId()).isNull();
assertThat(value.getCustomerName()).isNull();
value.setVersion(1);
return 1;
});
service.updateDraft(id, 9L, "admin", "optional-save", new UpdateRequest(0, emptyCustomer));
when(repository.parties(id)).thenReturn(parties());
when(repository.requirements(id)).thenReturn(List.of(requirement()));
when(repository.transition(id, "DRAFT", "RELEASED", 1, 9L, "admin")).thenAnswer(invocation -> {
stored.get().setStatus("RELEASED"); stored.get().setVersion(2); return 1;
});
var result = service.release(id, 9L, "admin", "optional-release", new ReleaseRequest(1, "确认"));
assertThat(result.oceanBooking().getStatus()).isEqualTo("RELEASED");
assertThat(result.oceanBooking().getCustomerId()).isNull();
assertThat(result.oceanBooking().getCustomerName()).isNull();
verify(documents).requireActiveValidDocument(stored.get());
verify(customers, times(initialCustomer ? 1 : 0)).active(anyLong(), any());
}
@Test
void invalidSpecifiedCustomerIsRejectedBeforeCreatingDraft() {
when(customers.active(100L, "客户")).thenThrow(new BizException(
com.baiyue.common.core.api.ResultCode.CONFLICT, "所选客户不存在或未启用"));
assertThatThrownBy(() -> service.create(9L, "admin", "bad-customer", new CreateRequest(0, completeDraft())))
.isInstanceOf(BizException.class).hasMessageContaining("客户不存在或未启用");
verify(repository, never()).insert(any(OceanBookingEntity.class));
}
@Test
void releaseRevalidatesSpecifiedCustomer() {
when(repository.find(1L)).thenReturn(completeBooking("DRAFT", 3));
when(repository.parties(1L)).thenReturn(parties());
when(repository.requirements(1L)).thenReturn(List.of(requirement()));
when(customers.active(100L, "客户")).thenThrow(new BizException(
com.baiyue.common.core.api.ResultCode.CONFLICT, "所选客户不存在或未启用"));
assertThatThrownBy(() -> service.release(1L, 9L, "admin", "inactive-customer", new ReleaseRequest(3, null)))
.isInstanceOf(BizException.class).hasMessageContaining("客户不存在或未启用");
verify(repository, never()).transition(anyLong(), anyString(), anyString(), anyInt(), anyLong(), anyString());
}
@Test
void freezesServerSnapshotsAndPersistsQuantityRequirementsOnly() {
stubMasterData();
@@ -86,6 +153,9 @@ class OceanBookingServiceTest {
assertThat(inserted.get().getChannelName()).isEqualTo("美西海运");
assertThat(inserted.get().getWarehouseName()).isEqualTo("深圳中转仓");
assertThat(inserted.get().getBookingAgentName()).isEqualTo("订舱代理");
assertThat(inserted.get().getCustomerId()).isEqualTo(100L);
assertThat(inserted.get().getCustomerName()).isEqualTo("客户");
verify(customers).active(100L, "客户");
@SuppressWarnings("unchecked")
ArgumentCaptor<List<OceanBookingContainerRequirementEntity>> captor = ArgumentCaptor.forClass(List.class);
verify(repository).replaceRequirements(eq(inserted.get().getId()), captor.capture());