优化

firstleg/src/test/java/com/baiyue/firstleg/terminaldrayage/service/TerminalDrayageMailDispatcherTest.java

@@ -7,6 +7,7 @@ import org.springframework.mail.SimpleMailMessage;
 import org.springframework.mail.javamail.JavaMailSenderImpl;
 import org.springframework.test.util.ReflectionTestUtils;
 import java.util.List;
+import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.Mockito.*;
 import static org.mockito.ArgumentMatchers.*;
 
@@ -16,7 +17,7 @@ class TerminalDrayageMailDispatcherTest {
     TerminalDrayageMailDispatcher worker = spy(new TerminalDrayageMailDispatcher(mapper));
 
     void setup() {
-        ReflectionTestUtils.setField(worker,"from","operations@example.com");
+        ReflectionTestUtils.setField(worker,"username","operations@example.com");
         doReturn(sender).when(worker).sender();
         when(mapper.pendingMails()).thenReturn(List.of(new TerminalDrayageMapper.MailTask(
             1L,11L,2L,"Fleet","fleet@example.com","REQUEST_PICKUP","Pickup","Body","PENDING",0,null,null,null)));
@@ -27,8 +28,10 @@ class TerminalDrayageMailDispatcherTest {
     @Test void onlyClaimsOnceAndMarksSentAfterSmtpAccepted() {
         setup(); worker.dispatch();
         var ordered = inOrder(mapper,sender);
+        var sent = org.mockito.ArgumentCaptor.forClass(SimpleMailMessage.class);
         ordered.verify(mapper).claimMail(1L);
-        ordered.verify(sender).send(any(SimpleMailMessage.class));
+        ordered.verify(sender).send(sent.capture());
+        assertThat(sent.getValue().getFrom()).isEqualTo("operations@example.com");
         ordered.verify(mapper).completeMail(1L);
         when(mapper.claimMail(1L)).thenReturn(0);
         worker.dispatch();

firstleg/src/test/java/com/baiyue/firstleg/terminaldrayage/service/TerminalDrayageMailServiceTest.java

@@ -27,6 +27,8 @@ class TerminalDrayageMailServiceTest {
     }
 
     @Test void replacementQueuesCancellationThenRequestWithSeparateFrozenRecipients() {
+        when(mapper.lastPickupMail(11L, 1L)).thenReturn(new TerminalDrayageMapper.MailTask(
+            90L,11L,1L,"Old Fleet","old@example.com","REQUEST_PICKUP","subject","body","SENT",1,null,null,null));
         service.replaceFleet(11L, "MSCU1234567", 1L, 2L, 7L);
         var order = inOrder(mapper);
         order.verify(mapper).insertMail(anyLong(), eq(11L), eq(1L), eq("Old Fleet"), eq("old@example.com"),
@@ -35,6 +37,16 @@ class TerminalDrayageMailServiceTest {
             eq("REQUEST_PICKUP"), eq("Port pickup request - MSCU1234567"), contains("Please arrange"), eq(7L));
     }
 
+    @Test void replacementWithoutPreviousPickupMailQueuesOnlyNewRequest() {
+        service.replaceFleet(11L, "MSCU1234567", 1L, 2L, 7L);
+
+        verify(mapper, never()).insertMail(anyLong(), eq(11L), eq(1L), anyString(), anyString(),
+            eq("CANCEL_PICKUP"), anyString(), anyString(), eq(7L));
+        verify(mapper).insertMail(anyLong(), eq(11L), eq(2L), eq("New Fleet"), eq("new@example.com"),
+            eq("REQUEST_PICKUP"), eq("Port pickup request - MSCU1234567"), contains("Please arrange"), eq(7L));
+        verify(suppliers, never()).truckingContact("test-token", 1L, false);
+    }
+
     @Test void cancellationUsesOriginalRecipientEvenIfSupplierEmailChanged() {
         when(mapper.lastPickupMail(11L, 1L)).thenReturn(new TerminalDrayageMapper.MailTask(
             90L,11L,1L,"Frozen Fleet","original@example.com","REQUEST_PICKUP","subject","body","SENT",1,null,null,null));

firstleg/src/test/java/com/baiyue/firstleg/terminaldrayage/service/TerminalDrayageServiceTest.java

@@ -374,18 +374,23 @@ class TerminalDrayageServiceTest {
     }
 
     @Test
-    void manualPickupEmailUsesLockedVersionAndRejectsStaleClick() {
+    void manualPickupEmailRejectsStaleClickAndAllowsRepeatWithLatestVersion() {
         Context context = context("ARRIVED_DESTINATION_PORT",null);
         when(mapper.lockContext(11L)).thenReturn(context);
         when(mapper.context(11L)).thenReturn(context);
         TerminalDrayageEntity row = row(3);
         when(mapper.lock(11L)).thenReturn(row);
         when(mapper.find(11L)).thenReturn(row);
-        when(mapper.updateAggregate(any(),anyInt(),anyLong(),anyString())).thenAnswer(invocation -> {row.setVersion(4);return 1;});
+        when(mapper.updateAggregate(any(),anyInt(),anyLong(),anyString())).thenAnswer(invocation -> {
+            row.setVersion(invocation.getArgument(1, Integer.class) + 1);
+            return 1;
+        });
         var request = new com.baiyue.firstleg.terminaldrayage.dto.TerminalDrayageDtos.EmailRequest(3);
         service.sendPickupEmail(11L,7L,"admin","mail-1",request);
         assertThatThrownBy(() -> service.sendPickupEmail(11L,7L,"admin","mail-2",request)).isInstanceOf(BizException.class);
-        verify(mails,times(1)).request(11L,"MSCU1234567",91L,7L);
+        service.sendPickupEmail(11L,7L,"admin","mail-3",
+            new com.baiyue.firstleg.terminaldrayage.dto.TerminalDrayageDtos.EmailRequest(4));
+        verify(mails,times(2)).request(11L,"MSCU1234567",91L,7L);
     }
 
     private Context context(String node, LocalDateTime unloadedAt) {