package com.zy.core.thread.impl;
|
|
import com.zy.asrs.entity.DeviceConfig;
|
import com.zy.common.utils.RedisUtil;
|
import com.zy.core.cache.MessageQueue;
|
import com.zy.core.enums.SlaveType;
|
import com.zy.core.model.Task;
|
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.network.ZyStationConnectDriver;
|
import com.zy.core.thread.impl.v5.StationV5SegmentExecutor;
|
import com.zy.core.thread.impl.v5.StationV5StatusReader;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.util.Arrays;
|
import java.util.Collections;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.timeout;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
class ZyStationV5ThreadTest {
|
|
@Test
|
void pollAndDispatchQueuedCommand_submitsQueuedMoveCommandToSegmentExecutor() {
|
DeviceConfig deviceConfig = new DeviceConfig();
|
deviceConfig.setDeviceNo(1);
|
RedisUtil redisUtil = mock(RedisUtil.class);
|
StationCommand command = new StationCommand();
|
StationV5SegmentExecutor segmentExecutor = mock(StationV5SegmentExecutor.class);
|
|
ZyStationV5Thread thread = new ZyStationV5Thread(deviceConfig, redisUtil);
|
ReflectionTestUtils.setField(thread, "segmentExecutor", segmentExecutor);
|
|
MessageQueue.init(SlaveType.Devp, 1);
|
try {
|
MessageQueue.offer(SlaveType.Devp, 1, new Task(2, command));
|
|
ReflectionTestUtils.invokeMethod(thread, "pollAndDispatchQueuedCommand");
|
|
verify(segmentExecutor, timeout(1000)).execute(command);
|
} finally {
|
MessageQueue.clear(SlaveType.Devp, 1);
|
thread.close();
|
}
|
}
|
|
@Test
|
void clearPath_delegatesPureSlotClearingToDriver() {
|
DeviceConfig deviceConfig = new DeviceConfig();
|
deviceConfig.setDeviceNo(1);
|
RedisUtil redisUtil = mock(RedisUtil.class);
|
ZyStationConnectDriver connectDriver = mock(ZyStationConnectDriver.class);
|
when(connectDriver.clearTaskBufferSlot(10, 2)).thenReturn(true);
|
|
ZyStationV5Thread thread = new ZyStationV5Thread(deviceConfig, redisUtil);
|
ReflectionTestUtils.setField(thread, "zyStationConnectDriver", connectDriver);
|
|
StationTaskBufferItem hitItem = new StationTaskBufferItem();
|
hitItem.setSlotIdx(2);
|
hitItem.setTaskNo(100);
|
hitItem.setTargetStaNo(88);
|
|
StationProtocol station20 = new StationProtocol();
|
station20.setStationId(20);
|
station20.setTaskBufferItems(Collections.emptyList());
|
|
StationProtocol station10 = new StationProtocol();
|
station10.setStationId(10);
|
station10.setTaskBufferItems(List.of(hitItem));
|
|
StationV5StatusReader statusReader = (StationV5StatusReader) ReflectionTestUtils.getField(thread, "statusReader");
|
ReflectionTestUtils.setField(statusReader, "statusList", Arrays.asList(station20, station10));
|
|
boolean result = thread.clearPath(100);
|
|
assertTrue(result);
|
verify(connectDriver).clearTaskBufferSlot(eq(10), eq(2));
|
verify(redisUtil, never()).set(org.mockito.ArgumentMatchers.anyString(), org.mockito.ArgumentMatchers.any(), org.mockito.ArgumentMatchers.anyLong());
|
verify(redisUtil, never()).del(org.mockito.ArgumentMatchers.any(String[].class));
|
assertEquals(0, hitItem.getTaskNo());
|
assertEquals(0, hitItem.getTargetStaNo());
|
}
|
}
|
