use of org.openkilda.northbound.dto.v1.switches.SwitchLocationDto in project open-kilda by telstra.
the class JsonSerializationTest method switchDtoTest.
@Test
public void switchDtoTest() throws IOException {
SwitchDto dto = new SwitchDto(new SwitchId(SWITCH_ID), "address-test", 37040, "host", "desc", SwitchChangeType.ACTIVATED, false, "of_version", "manufacturer", "hardware", "software", "serial_number", "pop", new SwitchLocationDto(48.860611, 2.337633, "street", "city", "country"));
assertEquals(dto, pass(dto, SwitchDto.class));
}
use of org.openkilda.northbound.dto.v1.switches.SwitchLocationDto in project open-kilda by telstra.
the class StubServiceFactory method getNorthboundStub.
/**
* Get a stub for {@link NorthboundService}. The instance is tied to the factory state.
*/
public NorthboundService getNorthboundStub() {
NorthboundService serviceMock = mock(NorthboundService.class);
when(serviceMock.getActiveSwitches()).thenAnswer(invocation -> topologyDefinition.getActiveSwitches().stream().map(sw -> new SwitchDto(sw.getDpId(), "", 0, "", "", SwitchChangeType.ACTIVATED, false, "", "", "", "", "", "", new SwitchLocationDto(0.0, 0.0, "", "", ""))).collect(toList()));
when(serviceMock.getActiveLinks()).thenAnswer(invocation -> topologyDefinition.getIslsForActiveSwitches().stream().flatMap(link -> Stream.of(IslInfoData.builder().latency(0).source(new PathNode(link.getSrcSwitch().getDpId(), link.getSrcPort(), 0)).destination(new PathNode(link.getDstSwitch().getDpId(), link.getDstPort(), 1)).speed(link.getMaxBandwidth()).state(IslChangeType.DISCOVERED).availableBandwidth(0).build(), IslInfoData.builder().latency(0).source(new PathNode(link.getDstSwitch().getDpId(), link.getDstPort(), 0)).destination(new PathNode(link.getSrcSwitch().getDpId(), link.getSrcPort(), 1)).speed(link.getMaxBandwidth()).state(IslChangeType.DISCOVERED).availableBandwidth(0).build())).collect(toList()));
when(serviceMock.getAllFlows()).thenReturn(new ArrayList<>(flowPayloads.values()));
when(serviceMock.getFlow(any())).thenAnswer(invocation -> {
String flowId = (String) invocation.getArguments()[0];
return flowPayloads.containsKey(flowId) ? SerializationUtils.clone(flowPayloads.get(flowId)) : null;
});
when(serviceMock.getFlowStatus(any())).thenAnswer(invocation -> {
String flowId = (String) invocation.getArguments()[0];
return flows.containsKey(flowId) ? new FlowIdStatusPayload(flowId, FlowState.UP) : null;
});
when(serviceMock.addFlow(any())).thenAnswer(invocation -> {
FlowPayload result = SerializationUtils.clone(((FlowPayload) invocation.getArguments()[0]));
result.setLastUpdated(LocalTime.now().toString());
result.setStatus(FlowState.IN_PROGRESS.toString());
putFlow(result.getId(), result);
return result;
});
when(serviceMock.updateFlow(any(), any())).thenAnswer(invocation -> {
String flowId = (String) invocation.getArguments()[0];
FlowPayload result = SerializationUtils.clone(((FlowPayload) invocation.getArguments()[1]));
result.setLastUpdated(LocalTime.now().toString());
putFlow(flowId, result);
return result;
});
when(serviceMock.deleteFlow(any())).thenAnswer(invocation -> {
String flowId = (String) invocation.getArguments()[0];
flows.remove(flowId);
return flowPayloads.remove(flowId);
});
when(serviceMock.synchronizeSwitchRules(any())).thenReturn(new RulesSyncResult(emptyList(), emptyList(), emptyList(), emptyList()));
when(serviceMock.validateSwitchRules(any())).thenReturn(new RulesValidationResult(emptyList(), emptyList(), emptyList()));
return serviceMock;
}
Aggregations