-
- @Test
- public void testDeviceConnectionCheck_WORKING() throws Exception {
- final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
- Assert.assertNull(deviceConnectionCheckResult);
- }
-
- @Test
- public void testDeviceConnectionCheck_RIP() throws Exception {
- Mockito.reset(mockedConnectionContext);
- when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.RIP);
- final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
- Assert.assertNotNull(deviceConnectionCheckResult);
- Assert.assertTrue(deviceConnectionCheckResult.isDone());
- try {
- deviceConnectionCheckResult.get();
- Assert.fail("connection in state RIP should have caused exception here");
- } catch (final Exception e) {
- LOG.debug("expected behavior for RIP connection achieved");
- Assert.assertTrue(e instanceof ExecutionException);
- }
- }
-
- @Test
- public void testDeviceConnectionCheck_HANSHAKING() throws Exception {
- Mockito.reset(mockedConnectionContext);
- when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.HANDSHAKING);
- final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
- Assert.assertNotNull(deviceConnectionCheckResult);
- Assert.assertTrue(deviceConnectionCheckResult.isDone());
- try {
- final Boolean checkPositive = deviceConnectionCheckResult.get();
- Assert.assertTrue(checkPositive);
- } catch (final Exception e) {
- Assert.fail("connection in state HANDSHAKING should NOT have caused exception here");
- }
- }