2 * Copyright (c) 2016, 2017 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
8 package org.opendaylight.openflowplugin.applications.bulk.o.matic;
10 import static org.mockito.ArgumentMatchers.any;
11 import static org.mockito.Mockito.doReturn;
12 import static org.mockito.Mockito.when;
14 import java.util.ArrayList;
15 import java.util.List;
16 import java.util.Optional;
17 import java.util.concurrent.ExecutorService;
18 import org.junit.Before;
19 import org.junit.Test;
20 import org.junit.runner.RunWith;
21 import org.mockito.ArgumentMatchers;
22 import org.mockito.Mock;
23 import org.mockito.Mockito;
24 import org.mockito.junit.MockitoJUnitRunner;
25 import org.opendaylight.mdsal.binding.api.DataBroker;
26 import org.opendaylight.mdsal.binding.api.ReadTransaction;
27 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInput;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.SalFlowService;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
34 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
35 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
40 * Test for {@link FlowWriterDirectOFRpc}.
42 @RunWith(MockitoJUnitRunner.class)
43 public class FlowWriterDirectOFRpcTest {
45 private static final Logger LOG = LoggerFactory.getLogger(FlowWriterDirectOFRpcTest.class);
46 private static final int FLOWS_PER_DPN = 100;
49 private DataBroker mockDataBroker;
51 private SalFlowService mockSalFlowService;
53 private ExecutorService mockFlowPusher;
55 private ReadTransaction readOnlyTransaction;
57 private Nodes mockNodes;
59 private FlowWriterDirectOFRpc flowWriterDirectOFRpc;
62 public void setUp() throws Exception {
63 doReturn(RpcResultBuilder.success().buildFuture()).when(mockSalFlowService).addFlow(any());
65 when(mockDataBroker.newReadOnlyTransaction()).thenReturn(readOnlyTransaction);
66 NodeBuilder nodeBuilder = new NodeBuilder()
67 .setId(new NodeId("1"));
69 final List<Node> nodes = new ArrayList<>();
70 final Node node = nodeBuilder.build();
73 when(mockNodes.getNode()).thenReturn(nodes);
75 doReturn(FluentFutures.immediateFluentFuture(Optional.of(mockNodes))).when(readOnlyTransaction)
76 .read(any(LogicalDatastoreType.class), any());
78 Mockito.doAnswer(invocation -> {
79 ((Runnable)invocation.getArguments()[0]).run();
81 }).when(mockFlowPusher).execute(ArgumentMatchers.<Runnable>any());
83 flowWriterDirectOFRpc = new FlowWriterDirectOFRpc(mockDataBroker, mockSalFlowService, mockFlowPusher);
87 public void testRpcFlowAdd() throws Exception {
88 flowWriterDirectOFRpc.rpcFlowAdd("1", FLOWS_PER_DPN, 10);
89 Mockito.verify(mockSalFlowService, Mockito.times(FLOWS_PER_DPN)).addFlow(Mockito.<AddFlowInput>any());
93 public void testRpcFlowAddAll() throws Exception {
94 flowWriterDirectOFRpc.rpcFlowAddAll(FLOWS_PER_DPN, 10);
95 Mockito.verify(mockSalFlowService, Mockito.times(FLOWS_PER_DPN)).addFlow(Mockito.<AddFlowInput>any());