package com.zy.core.thread.impl.v5;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
import com.zy.common.utils.RedisUtil;
|
import com.zy.core.enums.RedisKeyType;
|
import com.zy.core.model.command.StationCommand;
|
import com.zy.core.service.StationTaskLoopService;
|
import org.junit.jupiter.api.Test;
|
|
import java.util.HashSet;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertSame;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.when;
|
|
class StationV5RunBlockReroutePlannerTest {
|
|
@Test
|
void plan_prefersLongerCandidateWhenShortestPathIsOverusedInsideTriggeredLoop() {
|
RedisUtil redisUtil = mock(RedisUtil.class);
|
String stateKey = RedisKeyType.STATION_RUN_BLOCK_REROUTE_STATE_.key + "100_10";
|
when(redisUtil.get(stateKey)).thenReturn(JSON.toJSONString(seedState(
|
100,
|
10,
|
2,
|
List.of(),
|
Map.of("10->11->20", 2)
|
)));
|
|
StationV5RunBlockReroutePlanner planner = new StationV5RunBlockReroutePlanner(redisUtil);
|
StationCommand shortest = moveCommand(100, 10, 20, 10, 11, 20);
|
StationCommand longer = moveCommand(100, 10, 20, 10, 31, 32, 20);
|
|
StationTaskLoopService.LoopIdentitySnapshot loopIdentity =
|
new StationTaskLoopService.LoopIdentitySnapshot("10|11|20", new HashSet<>(List.of(10, 11, 20)), 3, 3, "wholeLoop");
|
StationTaskLoopService.LoopEvaluation loopEvaluation =
|
new StationTaskLoopService.LoopEvaluation(100, 10, loopIdentity, 2, 3, true);
|
|
StationV5RunBlockReroutePlanner.PlanResult result = planner.plan(
|
100,
|
10,
|
loopEvaluation,
|
List.of(shortest, longer)
|
);
|
|
assertSame(longer, result.getCommand());
|
assertEquals(3, result.getPlanCount());
|
}
|
|
@Test
|
void plan_resetsIssuedRoutesWhenAllCandidatesHaveBeenTried() {
|
RedisUtil redisUtil = mock(RedisUtil.class);
|
String stateKey = RedisKeyType.STATION_RUN_BLOCK_REROUTE_STATE_.key + "100_10";
|
when(redisUtil.get(stateKey)).thenReturn(JSON.toJSONString(seedState(
|
100,
|
10,
|
1,
|
List.of(List.of(10, 11, 20), List.of(10, 31, 32, 20)),
|
Map.of("10->11->20", 1, "10->31->32->20", 1)
|
)));
|
|
StationV5RunBlockReroutePlanner planner = new StationV5RunBlockReroutePlanner(redisUtil);
|
StationCommand first = moveCommand(100, 10, 20, 10, 11, 20);
|
StationCommand second = moveCommand(100, 10, 20, 10, 31, 32, 20);
|
|
StationTaskLoopService.LoopEvaluation loopEvaluation =
|
new StationTaskLoopService.LoopEvaluation(100, 10, StationTaskLoopService.LoopIdentitySnapshot.empty(), 0, 0, false);
|
|
StationV5RunBlockReroutePlanner.PlanResult result = planner.plan(
|
100,
|
10,
|
loopEvaluation,
|
List.of(first, second)
|
);
|
|
assertSame(first, result.getCommand());
|
assertEquals(2, result.getPlanCount());
|
}
|
|
private static StationCommand moveCommand(Integer taskNo,
|
Integer stationId,
|
Integer targetStationId,
|
Integer... path) {
|
StationCommand command = new StationCommand();
|
command.setTaskNo(taskNo);
|
command.setStationId(stationId);
|
command.setTargetStaNo(targetStationId);
|
command.setNavigatePath(List.of(path));
|
return command;
|
}
|
|
private static JSONObject seedState(Integer taskNo,
|
Integer blockStationId,
|
Integer planCount,
|
List<List<Integer>> issuedRoutePathList,
|
Map<String, Integer> routeIssueCountMap) {
|
JSONObject state = new JSONObject();
|
state.put("taskNo", taskNo);
|
state.put("blockStationId", blockStationId);
|
state.put("planCount", planCount);
|
state.put("issuedRoutePathList", issuedRoutePathList);
|
state.put("routeIssueCountMap", new HashMap<>(routeIssueCountMap));
|
return state;
|
}
|
}
|
