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
9 package org.opendaylight.openflowplugin.applications.bulk.o.matic;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.when;
15 import com.google.common.base.Optional;
16 import com.google.common.util.concurrent.Futures;
17 import java.util.ArrayList;
18 import java.util.List;
19 import java.util.concurrent.ExecutorService;
20 import org.junit.Before;
21 import org.junit.Test;
22 import org.junit.runner.RunWith;
23 import org.mockito.ArgumentMatchers;
24 import org.mockito.Mock;
25 import org.mockito.Mockito;
26 import org.mockito.runners.MockitoJUnitRunner;
27 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
28 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.AddFlowInput;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.SalFlowService;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
36 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
42 * Test for {@link FlowWriterDirectOFRpc}.
44 @RunWith(MockitoJUnitRunner.class)
45 public class FlowWriterDirectOFRpcTest {
47 private static final Logger LOG = LoggerFactory.getLogger(FlowWriterDirectOFRpcTest.class);
48 private static final int FLOWS_PER_DPN = 100;
51 private DataBroker mockDataBroker;
53 private SalFlowService mockSalFlowService;
55 private ExecutorService mockFlowPusher;
57 private ReadOnlyTransaction readOnlyTransaction;
59 private Nodes mockNodes;
61 private FlowWriterDirectOFRpc flowWriterDirectOFRpc;
64 public void setUp() throws Exception {
65 doReturn(RpcResultBuilder.success().buildFuture()).when(mockSalFlowService).addFlow(any());
67 when(mockDataBroker.newReadOnlyTransaction()).thenReturn(readOnlyTransaction);
68 NodeBuilder nodeBuilder = new NodeBuilder()
69 .setId(new NodeId("1"));
71 final List<Node> nodes = new ArrayList<>();
72 final Node node = nodeBuilder.build();
75 when(mockNodes.getNode()).thenReturn(nodes);
77 when(readOnlyTransaction.read(Mockito.any(LogicalDatastoreType.class),
78 Mockito.<InstanceIdentifier<Nodes>>any()))
79 .thenReturn(Futures.immediateCheckedFuture(Optional.of(mockNodes)));
81 Mockito.doAnswer(invocation -> {
82 ((Runnable)invocation.getArguments()[0]).run();
84 }).when(mockFlowPusher).execute(ArgumentMatchers.<Runnable>any());
86 flowWriterDirectOFRpc = new FlowWriterDirectOFRpc(mockDataBroker, mockSalFlowService, mockFlowPusher);
90 public void testRpcFlowAdd() throws Exception {
91 flowWriterDirectOFRpc.rpcFlowAdd("1", FLOWS_PER_DPN, 10);
92 Mockito.verify(mockSalFlowService, Mockito.times(FLOWS_PER_DPN)).addFlow(Mockito.<AddFlowInput>any());
96 public void testRpcFlowAddAll() throws Exception {
97 flowWriterDirectOFRpc.rpcFlowAddAll(FLOWS_PER_DPN, 10);
98 Mockito.verify(mockSalFlowService, Mockito.times(FLOWS_PER_DPN)).addFlow(Mockito.<AddFlowInput>any());