.thenReturn(Futures.immediateFuture(null));
Mockito.when(rpcContext.stopClusterServices()).thenReturn(Futures.immediateFuture(null));
Mockito.when(statisticsContext.stopClusterServices()).thenReturn(Futures.immediateFuture(null));
- Mockito.when(statisticsContext.initialGatherDynamicData()).thenReturn(Futures.immediateFuture(null));
+ Mockito.when(statisticsContext.gatherDynamicData()).thenReturn(Futures.immediateFuture(null));
Mockito.when(connectionContext.getDeviceInfo()).thenReturn(deviceInfo);
contextChain = new ContextChainImpl(connectionContext);
contextChain.isMastered(ContextChainMastershipState.INITIAL_SUBMIT);
contextChain.isMastered(ContextChainMastershipState.MASTER_ON_DEVICE);
contextChain.isMastered(ContextChainMastershipState.INITIAL_FLOW_REGISTRY_FILL);
+ contextChain.isMastered(ContextChainMastershipState.RPC_REGISTRATION);
contextChain.connectionDropped();
Mockito.verify(deviceContext).stopClusterServices();
Mockito.verify(rpcContext).stopClusterServices();