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.impl.datastore.MultipartWriterProviderFactory;
32 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManager;
33 import org.opendaylight.openflowplugin.openflow.md.core.sal.convertor.ConvertorManagerFactory;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.MultipartType;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
36 import org.opendaylight.yangtools.yang.common.RpcResult;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
41 @RunWith(MockitoJUnitRunner.class)
42 public class StatisticsContextImplTest extends StatisticsContextImpMockInitiation {
44 private static final Logger LOG = LoggerFactory.getLogger(StatisticsContextImplTest.class);
46 private static final Long TEST_XID = 55L;
47 private StatisticsContextImpl<MultipartReply> statisticsContext;
48 private ConvertorManager convertorManager;
51 public void setUp() throws Exception {
52 convertorManager = ConvertorManagerFactory.createDefaultManager();
53 when(mockedDeviceInfo.reserveXidForDeviceMessage()).thenReturn(TEST_XID);
54 Mockito.when(mockedDeviceContext.getDeviceState()).thenReturn(mockedDeviceState);
55 initStatisticsContext();
58 private void initStatisticsContext() {
59 statisticsContext = new StatisticsContextImpl<MultipartReply>(
60 true, mockedDeviceContext, convertorManager, mockedStatisticsManager,
61 MultipartWriterProviderFactory.createDefaultProvider(mockedDeviceContext),
64 statisticsContext.setStatisticsGatheringService(mockedStatisticsGatheringService);
65 statisticsContext.setStatisticsGatheringOnTheFlyService(mockedStatisticsOnFlyGatheringService);
69 public void testCreateRequestContext() {
70 final RequestContext<Object> requestContext = statisticsContext.createRequestContext();
71 assertNotNull(requestContext);
72 assertEquals(TEST_XID, requestContext.getXid().getValue());
73 Assert.assertFalse(requestContext.getFuture().isDone());
77 * There is nothing to check in close method.
80 public void testClose() throws Exception {
81 final StatisticsContextImpl<MultipartReply> statisticsContext = new StatisticsContextImpl<MultipartReply>(
82 true, mockedDeviceContext, convertorManager, mockedStatisticsManager,
83 MultipartWriterProviderFactory.createDefaultProvider(mockedDeviceContext),
86 final RequestContext<Object> requestContext = statisticsContext.createRequestContext();
87 statisticsContext.close();
89 Assert.assertTrue(requestContext.getFuture().isDone());
90 final RpcResult<?> rpcResult = requestContext.getFuture().get();
91 Assert.assertFalse(rpcResult.isSuccessful());
92 } catch (final ExecutionException e) {
93 LOG.error("request future value should be finished", e);
94 Assert.fail("request context closing failed");
99 public void testGatherDynamicData_none() throws Exception {
100 final ListenableFuture<Boolean> gatheringResult = statisticsContext.gatherDynamicData();
101 Assert.assertTrue(gatheringResult.isDone());
102 Assert.assertTrue(gatheringResult.get());
103 Mockito.verifyNoMoreInteractions(mockedStatisticsGatheringService, mockedStatisticsOnFlyGatheringService);
107 public void testGatherDynamicData_all() throws Exception {
108 Mockito.reset(mockedDeviceState);
109 when(mockedDeviceState.isTableStatisticsAvailable()).thenReturn(Boolean.TRUE);
110 when(mockedDeviceState.isFlowStatisticsAvailable()).thenReturn(Boolean.TRUE);
111 when(mockedDeviceState.isGroupAvailable()).thenReturn(Boolean.TRUE);
112 when(mockedDeviceState.isMetersAvailable()).thenReturn(Boolean.TRUE);
113 when(mockedDeviceState.isPortStatisticsAvailable()).thenReturn(Boolean.TRUE);
114 when(mockedDeviceState.isQueueStatisticsAvailable()).thenReturn(Boolean.TRUE);
115 when(mockedDeviceInfo.getNodeInstanceIdentifier()).thenReturn(DUMMY_NODE_ID);
116 initStatisticsContext();
118 when(mockedStatisticsGatheringService
119 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class)))
121 Futures.immediateFuture(RpcResultBuilder.success(Collections.<MultipartReply>emptyList()).build())
123 when(mockedStatisticsOnFlyGatheringService
124 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class)))
126 Futures.immediateFuture(RpcResultBuilder.success(Collections.<MultipartReply>emptyList()).build())
129 final ListenableFuture<Boolean> gatheringResult = statisticsContext.gatherDynamicData();
130 Assert.assertTrue(gatheringResult.isDone());
131 Assert.assertTrue(gatheringResult.get());
132 verify(mockedStatisticsGatheringService, times(7))
133 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class));
134 verify(mockedStatisticsOnFlyGatheringService)
135 .getStatisticsOfType(Matchers.any(EventIdentifier.class), Matchers.any(MultipartType.class));
136 Mockito.verifyNoMoreInteractions(mockedStatisticsGatheringService, mockedStatisticsOnFlyGatheringService);
140 public void testDeviceConnectionCheckWorking() throws Exception {
141 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
142 Assert.assertTrue(deviceConnectionCheckResult.get());
146 public void testDeviceConnectionCheckRip() throws Exception {
147 Mockito.reset(mockedConnectionContext);
148 when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.RIP);
149 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
152 deviceConnectionCheckResult.get();
153 Assert.fail("connection in state RIP should have caused exception here");
154 } catch (final ExecutionException e) {
155 LOG.debug("expected behavior for RIP connection achieved");
156 Assert.assertTrue(e instanceof ExecutionException);
161 public void testDeviceConnectionCheckHandshaking() throws Exception {
162 Mockito.reset(mockedConnectionContext);
163 when(mockedConnectionContext.getConnectionState()).thenReturn(ConnectionContext.CONNECTION_STATE.HANDSHAKING);
164 final ListenableFuture<Boolean> deviceConnectionCheckResult = statisticsContext.deviceConnectionCheck();
167 final Boolean checkPositive = deviceConnectionCheckResult.get();
168 Assert.assertTrue(checkPositive);
169 } catch (final ExecutionException e) {
170 Assert.fail("connection in state HANDSHAKING should NOT have caused exception here");