use of org.openkilda.wfm.topology.network.model.BfdSessionData in project open-kilda by telstra.
the class NetworkBfdSessionServiceTest method enableDisable.
@Test
public void enableDisable() throws UnknownHostException {
when(bfdSessionRepository.findBySwitchIdAndPort(alphaLogicalEndpoint.getDatapath(), alphaLogicalEndpoint.getPortNumber())).thenReturn(Optional.empty());
mockSwitchLookup(alphaSwitch);
mockSwitchLookup(betaSwitch);
doAnswer(invocation -> invocation.getArgument(0)).when(bfdSessionRepository).add(any());
// handle enable/update request
when(carrier.sendWorkerBfdSessionCreateRequest(any(NoviBfdSession.class))).thenReturn(setupRequestKey);
switchOnlineStatusMonitor.update(alphaEndpoint.getDatapath(), true);
service.enableUpdate(alphaLogicalEndpoint, alphaEndpoint.getPortNumber(), new BfdSessionData(alphaToBetaIslRef, genericBfdProperties));
ArgumentCaptor<NoviBfdSession> createBfdSessionRequestCaptor = ArgumentCaptor.forClass(NoviBfdSession.class);
verify(carrier).sendWorkerBfdSessionCreateRequest(createBfdSessionRequestCaptor.capture());
NoviBfdSession setupBfdSessionPayload = createBfdSessionRequestCaptor.getValue();
ArgumentCaptor<BfdSession> bfdSessionCreateArgument = ArgumentCaptor.forClass(BfdSession.class);
verify(bfdSessionRepository).add(bfdSessionCreateArgument.capture());
BfdSession bfdSessionDb = bfdSessionCreateArgument.getValue();
Assert.assertNotNull(bfdSessionDb.getDiscriminator());
Assert.assertNull(bfdSessionDb.getInterval());
Assert.assertEquals(Short.valueOf((short) 0), bfdSessionDb.getMultiplier());
// speaker response
when(bfdSessionRepository.findBySwitchIdAndPort(alphaLogicalEndpoint.getDatapath(), alphaLogicalEndpoint.getPortNumber())).thenReturn(Optional.of(bfdSessionCreateArgument.getValue()));
service.speakerResponse(alphaLogicalEndpoint, setupRequestKey, new BfdSessionResponse(setupBfdSessionPayload, null));
endpointStatusMonitor.update(alphaLogicalEndpoint, LinkStatus.UP);
verify(bfdSessionRepository).add(bfdSessionCreateArgument.capture());
bfdSessionDb = bfdSessionCreateArgument.getValue();
Assert.assertEquals(alphaLogicalEndpoint.getDatapath(), bfdSessionDb.getSwitchId());
Assert.assertEquals((Integer) alphaLogicalEndpoint.getPortNumber(), bfdSessionDb.getPort());
Assert.assertNotNull(bfdSessionDb.getDiscriminator());
ArgumentCaptor<NoviBfdSession> bfdSessionCreateSpeakerArgument = ArgumentCaptor.forClass(NoviBfdSession.class);
verify(carrier).sendWorkerBfdSessionCreateRequest(bfdSessionCreateSpeakerArgument.capture());
NoviBfdSession speakerBfdSetup = bfdSessionCreateSpeakerArgument.getValue();
Assert.assertEquals(alphaEndpoint.getDatapath(), speakerBfdSetup.getTarget().getDatapath());
Assert.assertEquals(InetAddress.getByName(alphaAddress), speakerBfdSetup.getTarget().getInetAddress());
Assert.assertEquals(betaEndpoint.getDatapath(), speakerBfdSetup.getRemote().getDatapath());
Assert.assertEquals(InetAddress.getByName(betaAddress), speakerBfdSetup.getRemote().getInetAddress());
Assert.assertEquals(alphaEndpoint.getPortNumber(), speakerBfdSetup.getPhysicalPortNumber());
Assert.assertEquals(alphaLogicalEndpoint.getPortNumber(), speakerBfdSetup.getLogicalPortNumber());
Assert.assertEquals(bfdSessionDb.getDiscriminator(), (Integer) speakerBfdSetup.getDiscriminator());
Assert.assertEquals(genericBfdProperties.getInterval(), bfdSessionDb.getInterval());
Assert.assertEquals(Short.valueOf(genericBfdProperties.getMultiplier()), bfdSessionDb.getMultiplier());
Assert.assertTrue(speakerBfdSetup.isKeepOverDisconnect());
verify(carrier).bfdUpNotification(alphaEndpoint);
verifyNoMoreInteractions(carrier);
resetCarrier();
reset(bfdSessionRepository);
// remove BFD session
when(carrier.sendWorkerBfdSessionDeleteRequest(any(NoviBfdSession.class))).thenReturn(removeRequestKey);
service.disable(alphaEndpoint);
verify(carrier).bfdKillNotification(alphaEndpoint);
ArgumentCaptor<NoviBfdSession> bfdSessionRemoveSpeakerArgument = ArgumentCaptor.forClass(NoviBfdSession.class);
verify(carrier).sendWorkerBfdSessionDeleteRequest(bfdSessionRemoveSpeakerArgument.capture());
NoviBfdSession speakerBfdRemove = bfdSessionRemoveSpeakerArgument.getValue();
Assert.assertEquals(speakerBfdSetup, speakerBfdRemove);
verifyNoMoreInteractions(carrier);
verifyNoMoreInteractions(bfdSessionRepository);
resetCarrier();
// remove confirmation
when(bfdSessionRepository.findBySwitchIdAndPort(alphaLogicalEndpoint.getDatapath(), alphaLogicalEndpoint.getPortNumber())).thenReturn(Optional.of(bfdSessionDb));
BfdSessionResponse speakerResponse = new BfdSessionResponse(speakerBfdRemove, null);
service.speakerResponse(alphaLogicalEndpoint, removeRequestKey, speakerResponse);
verify(carrier).sessionRotateRequest(alphaLogicalEndpoint, false);
verify(carrier).sessionCompleteNotification(alphaEndpoint);
verify(bfdSessionRepository).findBySwitchIdAndPort(alphaLogicalEndpoint.getDatapath(), alphaLogicalEndpoint.getPortNumber());
verify(bfdSessionRepository).remove(bfdSessionDb);
verifyNoMoreInteractions(carrier);
verifyNoMoreInteractions(bfdSessionRepository);
}
use of org.openkilda.wfm.topology.network.model.BfdSessionData in project open-kilda by telstra.
the class NetworkBfdSessionServiceTest method createOperationalSession.
private NoviBfdSession createOperationalSession() {
switchOnlineStatusMonitor.update(alphaLogicalEndpoint.getDatapath(), true);
doAnswer(invocation -> invocation.getArgument(0)).when(bfdSessionRepository).add(any());
mockSwitchLookup(alphaSwitch);
mockSwitchLookup(betaSwitch);
mockMissingBfdSession(alphaLogicalEndpoint);
when(carrier.sendWorkerBfdSessionCreateRequest(any(NoviBfdSession.class))).thenReturn(setupRequestKey);
service.enableUpdate(alphaLogicalEndpoint, alphaEndpoint.getPortNumber(), new BfdSessionData(alphaToBetaIslRef, genericBfdProperties));
ArgumentCaptor<NoviBfdSession> speakerBfdSetupRequestArgument = ArgumentCaptor.forClass(NoviBfdSession.class);
verify(carrier).sendWorkerBfdSessionCreateRequest(speakerBfdSetupRequestArgument.capture());
NoviBfdSession speakerBfdSession = speakerBfdSetupRequestArgument.getValue();
verify(bfdSessionRepository).add(any(BfdSession.class));
resetCarrier();
reset(bfdSessionRepository);
// speaker response
BfdSessionResponse response = new BfdSessionResponse(speakerBfdSession, null);
service.speakerResponse(alphaLogicalEndpoint, setupRequestKey, response);
verifyNoMoreInteractions(carrier);
resetCarrier();
// port up
endpointStatusMonitor.update(alphaLogicalEndpoint, LinkStatus.UP);
verify(carrier).bfdUpNotification(alphaEndpoint);
verifyNoMoreInteractions(carrier);
resetCarrier();
return speakerBfdSession;
}
Aggregations