From f3b64d003bc3458af3dd434e6187d3aba23a64aa Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期四, 26 三月 2026 14:35:44 +0800
Subject: [PATCH] #

---
 src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java |   25 +++++++++++++++++++++++++
 1 files changed, 25 insertions(+), 0 deletions(-)

diff --git a/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java b/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java
index d9163cf..05f033b 100644
--- a/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java
+++ b/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java
@@ -42,12 +42,21 @@
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.ArgumentMatchers.same;
 import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.inOrder;
 import static org.mockito.Mockito.times;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 class StationOperateProcessUtilsReroutePipelineTest {
+
+    @SuppressWarnings("unchecked")
+    private void stubTaskDispatchLock(StationMoveCoordinator coordinator) {
+        when(coordinator.withTaskDispatchLock(any(), any())).thenAnswer(invocation -> {
+            java.util.function.Supplier<Object> supplier = invocation.getArgument(1);
+            return supplier == null ? null : supplier.get();
+        });
+    }
 
     @Test
     void choosesRunBlockCommandBuilderForRunBlockRerouteScene() {
@@ -181,6 +190,7 @@
         StationOperateProcessUtils utils = new StationOperateProcessUtils();
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
 
@@ -226,6 +236,7 @@
         StationOperateProcessUtils utils = new StationOperateProcessUtils();
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
 
@@ -273,6 +284,7 @@
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         BasStationOptService basStationOptService = mock(BasStationOptService.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
         ReflectionTestUtils.setField(utils, "basStationOptService", basStationOptService);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
@@ -310,6 +322,7 @@
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
         StationThread stationThread = mock(StationThread.class);
+        stubTaskDispatchLock(coordinator);
 
         ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
         ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
@@ -371,6 +384,7 @@
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
         StationThread stationThread = mock(StationThread.class);
+        stubTaskDispatchLock(coordinator);
 
         ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
         ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
@@ -479,7 +493,10 @@
     void executePlan_runBlockReroute_reissuesWhenBlockedSessionMatchesCandidatePath() {
         StationOperateProcessUtils utils = new StationOperateProcessUtils();
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
+        RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
+        ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
 
         StationCommand command = new StationCommand();
         command.setTaskNo(100);
@@ -521,6 +538,9 @@
             );
 
             assertTrue(!result.skipped());
+            org.mockito.InOrder inOrder = inOrder(coordinator);
+            inOrder.verify(coordinator).markCancelPending(100, "reroute_pending");
+            inOrder.verify(coordinator).cancelSession(100);
             verify(coordinator, times(1)).cancelSession(100);
         } finally {
             MessageQueue.clear(SlaveType.Devp, 1);
@@ -532,6 +552,7 @@
         StationOperateProcessUtils utils = new StationOperateProcessUtils();
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
 
@@ -582,6 +603,7 @@
         StationOperateProcessUtils utils = new StationOperateProcessUtils();
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
+        stubTaskDispatchLock(coordinator);
         ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
 
@@ -636,6 +658,7 @@
         WrkAnalysisService wrkAnalysisService = mock(WrkAnalysisService.class);
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         StationThread stationThread = mock(StationThread.class);
+        stubTaskDispatchLock(coordinator);
 
         ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
         ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
@@ -696,6 +719,7 @@
         NotifyUtils notifyUtils = mock(NotifyUtils.class);
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         StationThread stationThread = mock(StationThread.class);
+        stubTaskDispatchLock(coordinator);
 
         ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
         ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
@@ -749,6 +773,7 @@
         StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
         RedisUtil redisUtil = mock(RedisUtil.class);
         StationThread stationThread = mock(StationThread.class);
+        stubTaskDispatchLock(coordinator);
 
         ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
         ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);

--
Gitblit v1.9.1