From 90402710d962aa357062ecb94649e0f277c1dfb3 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期日, 29 三月 2026 21:03:48 +0800
Subject: [PATCH] #
---
src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java | 404 ++++++++++++++++++++++++++++++++++++++++-----------------
1 files changed, 284 insertions(+), 120 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..cdc99b8 100644
--- a/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java
+++ b/src/test/java/com/zy/core/utils/StationOperateProcessUtilsReroutePipelineTest.java
@@ -2,8 +2,10 @@
import com.zy.asrs.entity.BasStationOpt;
import com.zy.asrs.entity.BasDevp;
+import com.zy.asrs.entity.BasStation;
import com.zy.asrs.entity.WrkMast;
import com.zy.asrs.service.BasDevpService;
+import com.zy.asrs.service.BasStationService;
import com.zy.asrs.service.BasStationOptService;
import com.zy.asrs.service.WrkMastService;
import com.zy.asrs.service.WrkAnalysisService;
@@ -19,16 +21,34 @@
import com.zy.core.move.StationMoveCoordinator;
import com.zy.core.move.StationMoveDispatchMode;
import com.zy.core.move.StationMoveSession;
+import com.zy.core.dispatch.StationCommandDispatchResult;
+import com.zy.core.dispatch.StationCommandDispatcher;
import com.zy.core.model.command.StationCommand;
import com.zy.core.model.protocol.StationProtocol;
import com.zy.core.model.protocol.StationTaskBufferItem;
import com.zy.core.thread.StationThread;
import com.zy.common.utils.RedisUtil;
+import com.zy.core.utils.station.StationDispatchLoadSupport;
+import com.zy.core.utils.station.StationDispatchRuntimeStateSupport;
+import com.zy.core.utils.station.StationOutboundDecisionSupport;
+import com.zy.core.utils.station.StationOutboundDispatchProcessor;
+import com.zy.core.utils.station.StationRegularDispatchProcessor;
+import com.zy.core.utils.station.StationRerouteProcessor;
+import com.zy.core.utils.station.model.DispatchLimitConfig;
+import com.zy.core.utils.station.model.LoadGuardState;
+import com.zy.core.utils.station.model.LoopHitResult;
+import com.zy.core.utils.station.model.RerouteCommandPlan;
+import com.zy.core.utils.station.model.RerouteContext;
+import com.zy.core.utils.station.model.RerouteDecision;
+import com.zy.core.utils.station.model.RerouteExecutionResult;
+import com.zy.core.utils.station.model.RerouteSceneType;
import org.junit.jupiter.api.Test;
import org.springframework.test.util.ReflectionTestUtils;
import java.util.Date;
import java.util.Collections;
+import java.util.HashMap;
+import java.util.IdentityHashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;
@@ -42,25 +62,115 @@
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 {
+ private final Map<StationOperateProcessUtils, Map<String, Object>> dependencyOverrides = new IdentityHashMap<>();
+
+ private StationOperateProcessUtils newUtils() {
+ StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ stash(utils, "stationOutboundDecisionSupport", new StationOutboundDecisionSupport());
+ wireRerouteProcessor(utils);
+ return utils;
+ }
+
+ private void wireOutboundSupport(StationOperateProcessUtils utils) {
+ StationOutboundDecisionSupport support = new StationOutboundDecisionSupport();
+ copyIfPresent(utils, support, "wrkMastService");
+ copyIfPresent(utils, support, "basDevpService");
+ copyIfPresent(utils, support, "basStationService");
+ copyIfPresent(utils, support, "navigateUtils");
+ copyIfPresent(utils, support, "redisUtil");
+ copyIfPresent(utils, support, "stationTaskLoopService");
+ copyIfPresent(utils, support, "stationMoveCoordinator");
+ StationDispatchRuntimeStateSupport runtimeStateSupport = new StationDispatchRuntimeStateSupport();
+ copyIfPresent(utils, runtimeStateSupport, "redisUtil");
+ copyIfPresent(utils, runtimeStateSupport, "basStationOptService");
+ ReflectionTestUtils.setField(support, "stationDispatchRuntimeStateSupport", runtimeStateSupport);
+ stash(utils, "stationOutboundDecisionSupport", support);
+ }
+
+ private void wireRerouteProcessor(StationOperateProcessUtils utils) {
+ StationRerouteProcessor processor = new StationRerouteProcessor();
+ copyIfPresent(utils, processor, "basDevpService");
+ copyIfPresent(utils, processor, "wrkMastService");
+ copyIfPresent(utils, processor, "commonService");
+ copyIfPresent(utils, processor, "redisUtil");
+ copyIfPresent(utils, processor, "locMastService");
+ copyIfPresent(utils, processor, "wmsOperateUtils");
+ copyIfPresent(utils, processor, "basStationOptService");
+ copyIfPresent(utils, processor, "stationMoveCoordinator");
+ Object dispatcher = readIfPresent(utils, "stationCommandDispatcher");
+ if (dispatcher == null) {
+ dispatcher = new StationCommandDispatcher(
+ (RedisUtil) readIfPresent(utils, "redisUtil"),
+ (StationMoveCoordinator) readIfPresent(utils, "stationMoveCoordinator")
+ );
+ }
+ ReflectionTestUtils.setField(processor, "stationCommandDispatcher", dispatcher);
+ Object outboundSupport = readIfPresent(utils, "stationOutboundDecisionSupport");
+ if (outboundSupport == null) {
+ wireOutboundSupport(utils);
+ outboundSupport = readIfPresent(utils, "stationOutboundDecisionSupport");
+ }
+ if (outboundSupport != null) {
+ ReflectionTestUtils.setField(processor, "stationOutboundDecisionSupport", outboundSupport);
+ }
+ StationDispatchRuntimeStateSupport runtimeStateSupport = new StationDispatchRuntimeStateSupport();
+ copyIfPresent(utils, runtimeStateSupport, "redisUtil");
+ copyIfPresent(utils, runtimeStateSupport, "basStationOptService");
+ ReflectionTestUtils.setField(processor, "stationDispatchRuntimeStateSupport", runtimeStateSupport);
+ ReflectionTestUtils.setField(utils, "stationRerouteProcessor", processor);
+ }
+
+ private void stash(StationOperateProcessUtils utils, String fieldName, Object value) {
+ dependencyOverrides.computeIfAbsent(utils, key -> new HashMap<>()).put(fieldName, value);
+ }
+
+ private Object readIfPresent(Object source, String fieldName) {
+ try {
+ return ReflectionTestUtils.getField(source, fieldName);
+ } catch (IllegalArgumentException ignore) {
+ if (source instanceof StationOperateProcessUtils) {
+ Map<String, Object> values = dependencyOverrides.get(source);
+ return values == null ? null : values.get(fieldName);
+ }
+ return null;
+ }
+ }
+
+ private void copyIfPresent(Object source, Object target, String fieldName) {
+ Object value = readIfPresent(source, fieldName);
+ if (value != null) {
+ try {
+ ReflectionTestUtils.setField(target, fieldName, value);
+ } catch (IllegalArgumentException ignore) {
+ }
+ }
+ }
+
+ @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() {
- StationOperateProcessUtils.RerouteSceneType scene =
- StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE;
+ RerouteSceneType scene = RerouteSceneType.RUN_BLOCK_REROUTE;
- assertSame(StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE, scene);
+ assertSame(RerouteSceneType.RUN_BLOCK_REROUTE, scene);
}
@Test
void resolveExecutionTarget_skipsWhenTargetEqualsCurrentStation() {
- StationOperateProcessUtils.RerouteDecision decision =
- StationOperateProcessUtils.RerouteDecision.skip("same-station");
+ RerouteDecision decision = RerouteDecision.skip("same-station");
assertTrue(decision.skip());
assertEquals("same-station", decision.skipReason());
@@ -68,7 +178,7 @@
@Test
void buildCommandPlan_usesRunBlockCommandBuilderForRunBlockScene() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationThread stationThread = mock(StationThread.class);
StationCommand command = new StationCommand();
command.setTaskNo(100);
@@ -76,8 +186,8 @@
command.setTargetStaNo(20);
when(stationThread.getRunBlockRerouteCommand(100, 10, 20, 0, 0.25d)).thenReturn(command);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.RUN_BLOCK_REROUTE,
buildBasDevp(1),
stationThread,
buildStationProtocol(10, 100, 10),
@@ -89,9 +199,9 @@
.withCancelSessionBeforeDispatch()
.withResetSegmentCommandsBeforeDispatch();
- StationOperateProcessUtils.RerouteCommandPlan plan = utils.buildRerouteCommandPlan(
+ RerouteCommandPlan plan = utils.buildRerouteCommandPlan(
context,
- StationOperateProcessUtils.RerouteDecision.proceed(20)
+ RerouteDecision.proceed(20)
);
verify(stationThread).getRunBlockRerouteCommand(100, 10, 20, 0, 0.25d);
@@ -100,7 +210,7 @@
@Test
void executePlan_skipsWhenCurrentTaskStillExistsInBuffer() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationCommand command = new StationCommand();
command.setTaskNo(100);
command.setStationId(10);
@@ -109,8 +219,8 @@
StationTaskBufferItem bufferItem = new StationTaskBufferItem();
bufferItem.setTaskNo(100);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.OUT_ORDER,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.OUT_ORDER,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(10, 100, 10, Collections.singletonList(bufferItem)),
@@ -120,11 +230,11 @@
"checkStationOutOrder"
);
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(20),
+ RerouteDecision.proceed(20),
"checkStationOutOrder"
)
);
@@ -135,11 +245,11 @@
@Test
void outOrderAndWatchCircle_shareDecisionFlow() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
WrkMast wrkMast = buildWrkMast(100, 20);
- StationOperateProcessUtils.RerouteContext outOrderContext = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.OUT_ORDER,
+ RerouteContext outOrderContext = RerouteContext.create(
+ RerouteSceneType.OUT_ORDER,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(10, 100, 10),
@@ -148,8 +258,8 @@
0.0d,
"checkStationOutOrder"
);
- StationOperateProcessUtils.RerouteContext watchCircleContext = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.WATCH_CIRCLE,
+ RerouteContext watchCircleContext = RerouteContext.create(
+ RerouteSceneType.WATCH_CIRCLE,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(10, 100, 10),
@@ -159,8 +269,8 @@
"watchCircleStation"
);
- StationOperateProcessUtils.RerouteDecision outOrderDecision = utils.resolveSharedRerouteDecision(outOrderContext);
- StationOperateProcessUtils.RerouteDecision watchCircleDecision = utils.resolveSharedRerouteDecision(watchCircleContext);
+ RerouteDecision outOrderDecision = utils.resolveSharedRerouteDecision(outOrderContext);
+ RerouteDecision watchCircleDecision = utils.resolveSharedRerouteDecision(watchCircleContext);
assertEquals(20, outOrderDecision.targetStationId());
assertEquals(20, watchCircleDecision.targetStationId());
@@ -168,7 +278,7 @@
@Test
void runBlockReroute_keepsDirectReassignAndNormalRerouteSeparate() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
WrkMast inboundWrkMast = buildWrkMast(100, 20);
inboundWrkMast.setIoType(WrkIoType.IN.id);
@@ -178,11 +288,13 @@
@Test
void idleRecover_skipsWhenLastDispatchIsTooRecent() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationMoveSession session = new StationMoveSession();
session.setStatus(StationMoveSession.STATUS_RUNNING);
@@ -196,8 +308,8 @@
command.setStationId(10);
command.setTargetStaNo(20);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.IDLE_RECOVER,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.IDLE_RECOVER,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(10, 100, 10),
@@ -207,11 +319,11 @@
"checkStationIdleRecover"
).withRecentDispatchGuard();
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(20),
+ RerouteDecision.proceed(20),
"checkStationIdleRecover"
)
);
@@ -223,11 +335,13 @@
@Test
void idleRecover_skipsWhenCurrentStationIsStillInsideRecentlyIssuedActiveRoute() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationMoveSession session = new StationMoveSession();
session.setStatus(StationMoveSession.STATUS_RUNNING);
@@ -242,8 +356,8 @@
command.setStationId(121);
command.setTargetStaNo(124);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.IDLE_RECOVER,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.IDLE_RECOVER,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(121, 10510, 121),
@@ -253,11 +367,11 @@
"checkStationIdleRecover"
).withRecentDispatchGuard();
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(124),
+ RerouteDecision.proceed(124),
"checkStationIdleRecover"
)
);
@@ -269,13 +383,15 @@
@Test
void idleRecover_skipsWhenStationCommandLogShowsRecentIssuedMove() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
BasStationOptService basStationOptService = mock(BasStationOptService.class);
RedisUtil redisUtil = mock(RedisUtil.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "basStationOptService", basStationOptService);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "basStationOptService", basStationOptService);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationMoveSession session = new StationMoveSession();
session.setStatus(StationMoveSession.STATUS_RUNNING);
@@ -304,17 +420,20 @@
@Test
void checkStationOutOrder_skipsWhenActiveSessionAlreadyOwnsCurrentStation() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
BasDevpService basDevpService = mock(BasDevpService.class);
WrkMastService wrkMastService = mock(WrkMastService.class);
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
StationThread stationThread = mock(StationThread.class);
+ stubTaskDispatchLock(coordinator);
- ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
+ stash(utils, "basDevpService", basDevpService);
ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireOutboundSupport(utils);
+ wireRerouteProcessor(utils);
BasDevp basDevp = buildBasDevp(1);
basDevp.setIsOutOrderList("[{\"deviceNo\":1,\"stationId\":145}]");
@@ -365,17 +484,20 @@
@Test
void checkStationOutOrder_restartsWhenBlockedSessionExistsButStationNoLongerRunBlock() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
BasDevpService basDevpService = mock(BasDevpService.class);
WrkMastService wrkMastService = mock(WrkMastService.class);
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
StationThread stationThread = mock(StationThread.class);
+ stubTaskDispatchLock(coordinator);
- ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
+ stash(utils, "basDevpService", basDevpService);
ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireOutboundSupport(utils);
+ wireRerouteProcessor(utils);
BasDevp basDevp = buildBasDevp(1);
basDevp.setIsOutOrderList("[{\"deviceNo\":1,\"stationId\":145}]");
@@ -427,15 +549,17 @@
@Test
void checkStationOutOrder_skipsRunBlockStationBeforePlanning() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
BasDevpService basDevpService = mock(BasDevpService.class);
WrkMastService wrkMastService = mock(WrkMastService.class);
RedisUtil redisUtil = mock(RedisUtil.class);
StationThread stationThread = mock(StationThread.class);
- ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
+ stash(utils, "basDevpService", basDevpService);
ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stash(utils, "redisUtil", redisUtil);
+ wireOutboundSupport(utils);
+ wireRerouteProcessor(utils);
BasDevp basDevp = buildBasDevp(1);
basDevp.setIsOutOrderList("[{\"deviceNo\":1,\"stationId\":145}]");
@@ -477,9 +601,13 @@
@Test
void executePlan_runBlockReroute_reissuesWhenBlockedSessionMatchesCandidatePath() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
+ RedisUtil redisUtil = mock(RedisUtil.class);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationCommand command = new StationCommand();
command.setTaskNo(100);
@@ -497,8 +625,8 @@
when(coordinator.loadSession(100)).thenReturn(session);
when(coordinator.buildPathSignature(command)).thenReturn("same-path");
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.RUN_BLOCK_REROUTE,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(145, 100, 145),
@@ -511,16 +639,19 @@
MessageQueue.init(SlaveType.Devp, 1);
try {
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(111),
+ RerouteDecision.proceed(111),
"checkStationRunBlock_reroute"
)
);
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);
@@ -529,11 +660,13 @@
@Test
void executePlan_runBlockReroute_ignoresCurrentTaskBufferAfterReset() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationCommand command = new StationCommand();
command.setTaskNo(10388);
@@ -547,8 +680,8 @@
when(redisUtil.get(anyString())).thenReturn(null);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.RUN_BLOCK_REROUTE,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(186, 10388, 189, Collections.singletonList(bufferItem)),
@@ -561,11 +694,11 @@
MessageQueue.init(SlaveType.Devp, 1);
try {
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(124),
+ RerouteDecision.proceed(124),
"checkStationRunBlock_reroute"
)
);
@@ -579,11 +712,13 @@
@Test
void executePlan_runBlockReroute_bypassesSuppressDispatchAfterReset() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stubTaskDispatchLock(coordinator);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireRerouteProcessor(utils);
StationCommand command = new StationCommand();
command.setTaskNo(10388);
@@ -595,8 +730,8 @@
when(redisUtil.get(anyString())).thenReturn(null);
when(coordinator.shouldSuppressDispatch(10388, 186, command)).thenReturn(true);
- StationOperateProcessUtils.RerouteContext context = StationOperateProcessUtils.RerouteContext.create(
- StationOperateProcessUtils.RerouteSceneType.RUN_BLOCK_REROUTE,
+ RerouteContext context = RerouteContext.create(
+ RerouteSceneType.RUN_BLOCK_REROUTE,
buildBasDevp(1),
mock(StationThread.class),
buildStationProtocol(186, 10388, 186),
@@ -610,11 +745,11 @@
MessageQueue.init(SlaveType.Devp, 1);
try {
- StationOperateProcessUtils.RerouteExecutionResult result = utils.executeReroutePlan(
+ RerouteExecutionResult result = utils.executeReroutePlan(
context,
- StationOperateProcessUtils.RerouteCommandPlan.dispatch(
+ RerouteCommandPlan.dispatch(
command,
- StationOperateProcessUtils.RerouteDecision.proceed(124),
+ RerouteDecision.proceed(124),
"checkStationRunBlock_reroute"
)
);
@@ -628,21 +763,28 @@
@Test
void stationInExecute_recordsDispatchSessionAfterIssuingMoveCommand() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
BasDevpService basDevpService = mock(BasDevpService.class);
WrkMastService wrkMastService = mock(WrkMastService.class);
CommonService commonService = mock(CommonService.class);
RedisUtil redisUtil = mock(RedisUtil.class);
WrkAnalysisService wrkAnalysisService = mock(WrkAnalysisService.class);
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
+ StationCommandDispatcher dispatcher = mock(StationCommandDispatcher.class);
+ StationDispatchLoadSupport loadSupport = mock(StationDispatchLoadSupport.class);
StationThread stationThread = mock(StationThread.class);
+ stubTaskDispatchLock(coordinator);
- ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
- ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "commonService", commonService);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
- ReflectionTestUtils.setField(utils, "wrkAnalysisService", wrkAnalysisService);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
+ StationRegularDispatchProcessor processor = new StationRegularDispatchProcessor();
+ ReflectionTestUtils.setField(processor, "basDevpService", basDevpService);
+ ReflectionTestUtils.setField(processor, "wrkMastService", wrkMastService);
+ ReflectionTestUtils.setField(processor, "commonService", commonService);
+ ReflectionTestUtils.setField(processor, "redisUtil", redisUtil);
+ ReflectionTestUtils.setField(processor, "wrkAnalysisService", wrkAnalysisService);
+ ReflectionTestUtils.setField(processor, "stationMoveCoordinator", coordinator);
+ ReflectionTestUtils.setField(processor, "stationCommandDispatcher", dispatcher);
+ ReflectionTestUtils.setField(processor, "stationDispatchLoadSupport", loadSupport);
+ ReflectionTestUtils.setField(utils, "stationRegularDispatchProcessor", processor);
BasDevp basDevp = buildBasDevp(1);
basDevp.setBarcodeStationList("[{\"deviceNo\":1,\"stationId\":101}]");
@@ -675,6 +817,18 @@
when(stationThread.getCommand(com.zy.core.enums.StationCommandType.MOVE, 500670, 101, 102, 0))
.thenReturn(command);
when(redisUtil.get(anyString())).thenReturn(null);
+ when(dispatcher.dispatch(1, command, "station-operate-process", "stationInExecute"))
+ .thenReturn(StationCommandDispatchResult.accepted("accepted", 1, "station-operate-process", "stationInExecute"));
+
+ DispatchLimitConfig baseConfig = new DispatchLimitConfig();
+ LoadGuardState loadGuardState = new LoadGuardState();
+ LoopHitResult noHit = LoopHitResult.noHit();
+ when(loadSupport.getDispatchLimitConfig(null, null)).thenReturn(baseConfig);
+ when(loadSupport.countCurrentStationTask()).thenReturn(0);
+ when(loadSupport.buildLoadGuardState(baseConfig)).thenReturn(loadGuardState);
+ when(loadSupport.getDispatchLimitConfig(101, 102)).thenReturn(baseConfig);
+ when(loadSupport.findPathLoopHit(baseConfig, 101, 102, loadGuardState)).thenReturn(noHit);
+ when(loadSupport.isDispatchBlocked(baseConfig, 0, loadGuardState, false)).thenReturn(false);
MessageQueue.init(SlaveType.Devp, 1);
SlaveConnection.put(SlaveType.Devp, 1, stationThread);
@@ -689,71 +843,81 @@
}
@Test
- void dualCrnStationOutExecute_recordsDispatchSessionAfterIssuingMoveCommand() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ void dualCrnStationOutExecute_delegatesToOutboundProcessor() {
+ StationOperateProcessUtils utils = newUtils();
+ StationOutboundDispatchProcessor processor = mock(StationOutboundDispatchProcessor.class);
+ ReflectionTestUtils.setField(utils, "stationOutboundDispatchProcessor", processor);
+
+ utils.dualCrnStationOutExecute();
+
+ verify(processor, times(1)).dualCrnStationOutExecute();
+ }
+
+ @Test
+ void stationOutExecuteFinish_attemptsClearPathBeforeCompletingTask() {
+ StationOperateProcessUtils utils = newUtils();
WrkMastService wrkMastService = mock(WrkMastService.class);
- RedisUtil redisUtil = mock(RedisUtil.class);
+ BasStationService basStationService = mock(BasStationService.class);
+ WrkAnalysisService wrkAnalysisService = mock(WrkAnalysisService.class);
NotifyUtils notifyUtils = mock(NotifyUtils.class);
+ RedisUtil redisUtil = mock(RedisUtil.class);
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
StationThread stationThread = mock(StationThread.class);
- ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
- ReflectionTestUtils.setField(utils, "notifyUtils", notifyUtils);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
+ StationRegularDispatchProcessor processor = new StationRegularDispatchProcessor();
+ ReflectionTestUtils.setField(processor, "wrkMastService", wrkMastService);
+ ReflectionTestUtils.setField(processor, "basStationService", basStationService);
+ ReflectionTestUtils.setField(processor, "wrkAnalysisService", wrkAnalysisService);
+ ReflectionTestUtils.setField(processor, "notifyUtils", notifyUtils);
+ ReflectionTestUtils.setField(processor, "redisUtil", redisUtil);
+ ReflectionTestUtils.setField(processor, "stationMoveCoordinator", coordinator);
+ ReflectionTestUtils.setField(utils, "stationRegularDispatchProcessor", processor);
WrkMast wrkMast = buildWrkMast(10335, 145);
- wrkMast.setDualCrnNo(1);
- wrkMast.setWrkSts(WrkStsType.OUTBOUND_RUN_COMPLETE.sts);
+ wrkMast.setWrkSts(WrkStsType.STATION_RUN.sts);
when(wrkMastService.list(any(com.baomidou.mybatisplus.core.conditions.Wrapper.class)))
.thenReturn(Collections.singletonList(wrkMast));
when(wrkMastService.updateById(wrkMast)).thenReturn(true);
- when(redisUtil.get(anyString())).thenAnswer(invocation -> {
- String key = invocation.getArgument(0);
- if (Objects.equals(key, RedisKeyType.DUAL_CRN_OUT_TASK_STATION_INFO.key + 10335)) {
- return "{\"deviceNo\":1,\"stationId\":198}";
- }
- return null;
- });
+ BasStation basStation = new BasStation();
+ basStation.setStationId(145);
+ basStation.setDeviceNo(1);
+ when(basStationService.getOne(any(com.baomidou.mybatisplus.core.conditions.Wrapper.class)))
+ .thenReturn(basStation);
- StationProtocol stationProtocol = buildStationProtocol(198, 0, 198);
- stationProtocol.setAutoing(true);
- stationProtocol.setLoading(true);
- when(stationThread.getStatusMap()).thenReturn(Map.of(198, stationProtocol));
+ StationProtocol stationProtocol = buildStationProtocol(145, 10335, 145);
+ when(stationThread.getStatusMap()).thenReturn(Map.of(145, stationProtocol));
+ when(stationThread.clearPath(10335)).thenReturn(true);
- StationCommand command = new StationCommand();
- command.setTaskNo(10335);
- command.setStationId(198);
- command.setTargetStaNo(145);
- when(stationThread.getCommand(eq(com.zy.core.enums.StationCommandType.MOVE), eq(10335), eq(198), eq(145), eq(0), eq(0.0d)))
- .thenReturn(command);
-
- MessageQueue.init(SlaveType.Devp, 1);
SlaveConnection.put(SlaveType.Devp, 1, stationThread);
try {
- utils.dualCrnStationOutExecute();
+ utils.stationOutExecuteFinish();
- verify(coordinator, times(1)).recordDispatch(eq(10335), eq(198), eq("dualCrnStationOutExecute"), same(command), eq(false));
+ verify(stationThread, times(1)).clearPath(10335);
+ verify(coordinator, times(1)).finishSession(10335);
+ verify(wrkMastService, times(1)).updateById(wrkMast);
+ assertEquals(WrkStsType.STATION_RUN_COMPLETE.sts, wrkMast.getWrkSts());
} finally {
- MessageQueue.clear(SlaveType.Devp, 1);
SlaveConnection.remove(SlaveType.Devp, 1);
}
}
@Test
void watchCircleStation_usesSessionArrivalStateWhenLegacyCommandMissing() {
- StationOperateProcessUtils utils = new StationOperateProcessUtils();
+ StationOperateProcessUtils utils = newUtils();
BasDevpService basDevpService = mock(BasDevpService.class);
WrkMastService wrkMastService = mock(WrkMastService.class);
StationMoveCoordinator coordinator = mock(StationMoveCoordinator.class);
RedisUtil redisUtil = mock(RedisUtil.class);
StationThread stationThread = mock(StationThread.class);
+ stubTaskDispatchLock(coordinator);
- ReflectionTestUtils.setField(utils, "basDevpService", basDevpService);
+ stash(utils, "basDevpService", basDevpService);
ReflectionTestUtils.setField(utils, "wrkMastService", wrkMastService);
- ReflectionTestUtils.setField(utils, "stationMoveCoordinator", coordinator);
- ReflectionTestUtils.setField(utils, "redisUtil", redisUtil);
+ stash(utils, "stationMoveCoordinator", coordinator);
+ stash(utils, "redisUtil", redisUtil);
+ wireOutboundSupport(utils);
+ wireRerouteProcessor(utils);
BasDevp basDevp = buildBasDevp(1);
when(basDevpService.list(any(com.baomidou.mybatisplus.core.conditions.Wrapper.class)))
--
Gitblit v1.9.1