コード例 #1
0
 private ConnectionContext prepareConnectionContext() {
   ConnectionContext mockedConnectionContext = mock(ConnectionContext.class);
   FeaturesReply mockedFeaturesReply = mock(FeaturesReply.class);
   when(mockedFeaturesReply.getAuxiliaryId()).thenReturn(DUMMY_AUXILIARY_ID);
   when(mockedConnectionContext.getFeatures()).thenReturn(mockedFeaturesReply);
   return mockedConnectionContext;
 }
コード例 #2
0
  @Test
  public void registerServiceTest() {

    final RpcContext mockedRpcContext = mock(RpcContext.class);
    final DeviceContext mockedDeviceContext = mock(DeviceContext.class);
    final ConnectionContext mockedConnectionContext = mock(ConnectionContext.class);

    final FeaturesReply mockedFeatures = mock(FeaturesReply.class);
    when(mockedConnectionContext.getFeatures()).thenReturn(mockedFeatures);

    final BigInteger mockedDataPathId = mock(BigInteger.class);
    when(mockedFeatures.getDatapathId()).thenReturn(mockedDataPathId);

    when(mockedDeviceContext.getPrimaryConnectionContext()).thenReturn(mockedConnectionContext);
    MdSalRegistratorUtils.registerServices(mockedRpcContext, mockedDeviceContext);
    verify(mockedRpcContext, times(NUMBER_OF_RPC_SERVICE_REGISTRATION))
        .registerRpcServiceImplementation(any(RpcService.class.getClass()), any(RpcService.class));
  }