提交

event-bridge-service/src/test/java/com/baiyue/eventbridge/service/OverseasEventDispatcherTest.java

@@ -21,6 +21,7 @@ class OverseasEventDispatcherTest {
     private final BridgeInboxMapper inboxMapper = mock(BridgeInboxMapper.class);
     private final BridgeDeliveryAttemptMapper attemptMapper = mock(BridgeDeliveryAttemptMapper.class);
     private final LastlegEventClient eventClient = mock(LastlegEventClient.class);
+    private final FirstlegEventClient firstlegEventClient = mock(FirstlegEventClient.class);
     private OverseasEventDispatcher dispatcher;
     private BridgeInbox event;
 
@@ -34,6 +35,7 @@ class OverseasEventDispatcherTest {
         ReflectionTestUtils.setField(dispatcher, "inboxMapper", inboxMapper);
         ReflectionTestUtils.setField(dispatcher, "attemptMapper", attemptMapper);
         ReflectionTestUtils.setField(dispatcher, "eventClient", eventClient);
+        ReflectionTestUtils.setField(dispatcher, "firstlegEventClient", firstlegEventClient);
         ReflectionTestUtils.setField(dispatcher, "properties", properties);
         event = new BridgeInbox();
         event.setEventId("event-1");
@@ -83,9 +85,24 @@ class OverseasEventDispatcherTest {
         dispatcher.dispatchOne("event-1");
 
         verifyNoInteractions(eventClient);
+        verifyNoInteractions(firstlegEventClient);
         verify(attemptMapper, never()).insert(any(BridgeDeliveryAttempt.class));
     }
 
+    @Test
+    void cnTargetIsDeliveredToFirstleg() throws Exception {
+        event.setSourceRegion("OVERSEAS");
+        event.setTargetRegion("CN");
+        when(firstlegEventClient.deliver(event)).thenReturn("200");
+        when(inboxMapper.markProcessed(eq("event-1"), anyString())).thenReturn(1);
+
+        dispatcher.dispatchOne("event-1");
+
+        verify(firstlegEventClient).deliver(event);
+        verifyNoInteractions(eventClient);
+        assertThat(updatedAttempt().getStatus()).isEqualTo("DELIVERED");
+    }
+
     private BridgeDeliveryAttempt updatedAttempt() {
         ArgumentCaptor<BridgeDeliveryAttempt> captor = ArgumentCaptor.forClass(BridgeDeliveryAttempt.class);
         verify(attemptMapper).updateById(captor.capture());