Examples with BfdSession org.openkilda.model.BfdSession used on opensource projects

Example 11 with BfdSession

use of org.openkilda.model.BfdSession 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);
}