2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
9 package org.opendaylight.openflowplugin.impl.statistics;
11 import static org.junit.Assert.assertEquals;
12 import static org.junit.Assert.assertNotNull;
13 import static org.mockito.Mockito.times;
14 import static org.mockito.Mockito.verify;
15 import static org.mockito.Mockito.when;
17 import com.google.common.util.concurrent.Futures;
18 import com.google.common.util.concurrent.ListenableFuture;
19 import java.util.Collections;
20 import java.util.concurrent.ExecutionException;
21 import org.junit.Assert;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.junit.runner.RunWith;
25 import org.mockito.Matchers;
26 import org.mockito.Mockito;
27 import org.mockito.runners.MockitoJUnitRunner;
28 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
29 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
30 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.EventIdentifier;
31 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
32 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
35 import org.opendaylight.yangtools.yang.common.RpcResult;
36 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
37 import org.slf4j.Logger;
38 import org.slf4j.LoggerFactory;
40 @RunWith(MockitoJUnitRunner.class)
41 public class StatisticsContextImplTest extends StatisticsContextImpMockInitiation {
43 private static final Logger LOG = LoggerFactory.getLogger(StatisticsContextImplTest.class);
45 private static final Long TEST_XID = 55L;
46 private StatisticsContextImpl statisticsContext;
47 private ConvertorManager convertorManager;
50 public void setUp() throws Exception {
51 convertorManager = ConvertorManagerFactory.createDefaultManager();
52 when(mockedDeviceInfo.reserveXidForDeviceMessage()).thenReturn(TEST_XID);
53 Mockito.when(lifecycleService.getDeviceContext()).thenReturn(mockedDeviceContext);
54 Mockito.when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
55 initStatisticsContext();
58 private void initStatisticsContext() {
59 statisticsContext = new StatisticsContextImpl(mockedDeviceInfo, false, lifecycleService, convertorManager, mockedStatisticsManager);
60 statisticsContext.setStatisticsGatheringService(mockedStatisticsGatheringService);
61 statisticsContext.setStatisticsGatheringOnTheFlyService(mockedStatisticsOnFlyGatheringService);
65 public void testCreateRequestContext() {
66 final RequestContext<Object> requestContext = statisticsContext.createRequestContext();
67 assertNotNull(requestContext);
68 assertEquals(TEST_XID, requestContext.getXid().getValue());
69 Assert.assertFalse(requestContext.getFuture().isDone());
73 * There is nothing to check in close method
76 public void testClose() throws Exception {
77 final StatisticsContextImpl statisticsContext = new StatisticsContextImpl(mockedDeviceInfo, false, lifecycleService, convertorManager, mockedStatisticsManager);
78 final RequestContext<Object> requestContext = statisticsContext.createRequestContext();
79 statisticsContext.close();
81 Assert.assertTrue(requestContext.getFuture().isDone());
82 final RpcResult<?> rpcResult = requestContext.getFuture().get();
83 Assert.assertFalse(rpcResult.isSuccessful());
84 Assert.assertFalse(rpcResult.isSuccessful());
85 } catch (final Exception e) {
86 LOG.error("request future value should be finished", e);
87 Assert.fail("request context closing failed");
92 public void testGatherDynamicData_none() throws Exception {
93 final ListenableFuture<Boolean> gatheringResult = statisticsContext.gatherDynamicData();
94 Assert.assertTrue(gatheringResult.isDone());
95 Assert.assertTrue(gatheringResult.get());
96 Mockito.verifyNoMoreInteractions(mockedStatisticsGatheringService, mockedStatisticsOnFlyGatheringService);
100 public void testGatherDynamicData_all() throws Exception {
101 Mockito.reset(mockedDeviceState);
102 when(mockedDeviceState.isTableStatisticsAvailable()).thenReturn(Boolean.TRUE);
103 when(mockedDeviceState.isFlowStatisticsAvailable()).thenReturn(Boolean.TRUE);
104 when(mockedDeviceState.isGroupAvailable()).thenReturn(Boolean.TRUE);
105 when(mockedDeviceState.isMetersAvailable()).thenReturn(Boolean.TRUE);
106 when(mockedDeviceState.isPortStatisticsAvailable()).thenReturn(Boolean.TRUE);
107 when(mockedDeviceState.isQueueStatisticsAvailable()).thenReturn(Boolean.TRUE);
108 when(mockedDeviceInfo.getNodeInstanceIdentifier()).thenReturn(dummyNodeII);
109 initStatisticsContext();
111 when(mockedStatisticsGatheringService.getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class)))
113 Futures.immediateFuture(RpcResultBuilder.success(Collections.<MultipartReply>emptyList()).build())
115 when(mockedStatisticsOnFlyGatheringService.getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class)))
117 Futures.immediateFuture(RpcResultBuilder.success(Collections.<MultipartReply>emptyList()).build())
120 final ListenableFuture<Boolean> gatheringResult = statisticsContext.gatherDynamicData();
121 Assert.assertTrue(gatheringResult.isDone());
122 Assert.assertTrue(gatheringResult.get());
123 verify(mockedStatisticsGatheringService, times(7))
124 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class));
125 verify(mockedStatisticsOnFlyGatheringService)
126 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class));
127 Mockito.verifyNoMoreInteractions(mockedStatisticsGatheringService, mockedStatisticsOnFlyGatheringService);
131 public void testDeviceConnectionCheck_WORKING() throws Exception {
132 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
133 Assert.assertNull(deviceConnectionCheckResult);
137 public void testDeviceConnectionCheck_RIP() throws Exception {
138 Mockito.reset(mockedConnectionContext);
139 when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.RIP);
140 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
141 Assert.assertNotNull(deviceConnectionCheckResult);
142 Assert.assertTrue(deviceConnectionCheckResult.isDone());
144 deviceConnectionCheckResult.get();
145 Assert.fail("connection in state RIP should have caused exception here");
146 } catch (final Exception e) {
147 LOG.debug("expected behavior for RIP connection achieved");
148 Assert.assertTrue(e instanceof ExecutionException);
153 public void testDeviceConnectionCheck_HANSHAKING() throws Exception {
154 Mockito.reset(mockedConnectionContext);
155 when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.HANDSHAKING);
156 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
157 Assert.assertNotNull(deviceConnectionCheckResult);
158 Assert.assertTrue(deviceConnectionCheckResult.isDone());
160 final Boolean checkPositive = deviceConnectionCheckResult.get();
161 Assert.assertTrue(checkPositive);
162 } catch (final Exception e) {
163 Assert.fail("connection in state HANDSHAKING should NOT have caused exception here");