package com.ruoyi.system.service.impl;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.eq;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
import java.util.List;
|
import java.util.function.Function;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.Mockito;
|
import com.battery.device.api.DeviceAdapter;
|
import com.battery.device.discovery.InMemoryDeviceDiscoveryCandidateStore;
|
import com.battery.device.model.ChannelIdentity;
|
import com.battery.device.model.DetectedChannel;
|
import com.battery.device.registry.DeviceAdapterRegistry;
|
import com.ruoyi.system.mapper.TDetectionCabinetMapper;
|
import com.ruoyi.system.mapper.TTestMapper;
|
|
class DeviceDiscoveryPersistenceServiceTest
|
{
|
@Test
|
void discoveryPersistsCandidatesWithoutReceivingTest()
|
{
|
InMemoryDeviceDiscoveryCandidateStore candidateStore = new InMemoryDeviceDiscoveryCandidateStore();
|
TDetectionCabinetMapper cabinetMapper = Mockito.mock(TDetectionCabinetMapper.class);
|
TTestMapper testMapper = Mockito.mock(TTestMapper.class);
|
DeviceAdapterRegistry registry = Mockito.mock(DeviceAdapterRegistry.class);
|
DeviceAdapter adapter = Mockito.mock(DeviceAdapter.class);
|
ChannelIdentity identity = new ChannelIdentity("lab-bts", 22, 1, 1, 1);
|
when(testMapper.hasReceivingTestForAdapter("lab-bts")).thenReturn(0);
|
when(adapter.discoverChannels()).thenReturn(List.of(new DetectedChannel(identity, "127.0.0.1", true)));
|
when(registry.execute(eq("lab-bts"), any())).thenAnswer(invocation -> {
|
@SuppressWarnings("unchecked")
|
Function<DeviceAdapter, Object> operation = (Function<DeviceAdapter, Object>) invocation.getArgument(1);
|
return operation.apply(adapter);
|
});
|
|
DeviceDiscoveryPersistenceService service = new DeviceDiscoveryPersistenceService(candidateStore,
|
cabinetMapper, testMapper);
|
|
var result = service.discoverForReception(registry,
|
new DeviceDiscoveryPersistenceService.DiscoverySource("lab-bts", 7L, "一号 BTS", 1000));
|
|
assertEquals(1, result.cabinetCount());
|
assertEquals(1, candidateStore.snapshot(java.time.Instant.now()).cabinets().size());
|
verify(adapter).discoverChannels();
|
}
|
}
|
