- Mockito.verify(serviceControlRegistration).close();
- }
-
- @Test
- public void testCalculateTimerDelay() throws Exception {
- final TimeCounter timeCounter = Mockito.mock(TimeCounter.class);
- when(timeCounter.getAverageTimeBetweenMarks()).thenReturn(2000L, (Long)4000L);
-
- statisticsManager.calculateTimerDelay(timeCounter);
- Assert.assertEquals(3000L, StatisticsManagerImpl.getCurrentTimerDelay());
- statisticsManager.calculateTimerDelay(timeCounter);
- Assert.assertEquals(6000L, StatisticsManagerImpl.getCurrentTimerDelay());
- }
-
- @Test
- public void testPollStatistics() throws Exception {
- final StatisticsContext statisticsContext = Mockito.mock(StatisticsContext.class);
- final TimeCounter mockTimerCounter = Mockito.mock(TimeCounter.class);
-
- statisticsManager.pollStatistics(mockedDeviceContext.getDeviceState(), statisticsContext, mockTimerCounter, mockedDeviceInfo);
- verify(mockedDeviceContext).getDeviceState();
-
- when(mockedDeviceContext.getDeviceState().isValid()).thenReturn(true);
- statisticsManager.pollStatistics(mockedDeviceContext.getDeviceState(), statisticsContext, mockTimerCounter, mockedDeviceInfo);
-
- when(mockedDeviceContext.getDeviceState().isStatisticsPollingEnabled()).thenReturn(true);
- statisticsManager.pollStatistics(mockedDeviceContext.getDeviceState(), statisticsContext, mockTimerCounter, mockedDeviceInfo);
-
- when(statisticsContext.gatherDynamicData()).thenReturn(Futures.immediateCheckedFuture(Boolean.TRUE));
- when(statisticsContext.isSchedulingEnabled()).thenReturn(Boolean.TRUE);
- statisticsManager.pollStatistics(mockedDeviceContext.getDeviceState(), statisticsContext, mockTimerCounter, mockedDeviceInfo);
- Mockito.verify(mockTimerCounter).markStart();
- Mockito.verify(mockTimerCounter).addTimeMark();
-
- when(statisticsContext.gatherDynamicData()).thenReturn(Futures.immediateFailedFuture(new Throwable("error msg")));
- statisticsManager.pollStatistics(mockedDeviceContext.getDeviceState(), statisticsContext, mockTimerCounter, mockedDeviceInfo);
- Mockito.verify(mockTimerCounter,times(2)).addTimeMark();