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