2 * Copyright (c) 2014 Brocade Communications 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.topology.manager;
10 import static org.junit.Assert.assertEquals;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.ArgumentMatchers.eq;
13 import static org.mockito.Mockito.doReturn;
14 import static org.mockito.Mockito.mock;
15 import static org.mockito.Mockito.never;
16 import static org.mockito.Mockito.verify;
17 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newDestTp;
18 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newInvNodeConnKey;
19 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newInvNodeKey;
20 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newLink;
21 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newNodeConnID;
22 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.newSourceTp;
23 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.setupStubbedSubmit;
24 import static org.opendaylight.openflowplugin.applications.topology.manager.TestUtils.waitForSubmit;
26 import java.util.Optional;
27 import java.util.concurrent.CountDownLatch;
28 import java.util.concurrent.ExecutorService;
29 import java.util.concurrent.Executors;
30 import org.junit.After;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.mockito.ArgumentCaptor;
34 import org.mockito.Mock;
35 import org.mockito.MockitoAnnotations;
36 import org.opendaylight.mdsal.binding.api.DataBroker;
37 import org.opendaylight.mdsal.binding.api.ReadWriteTransaction;
38 import org.opendaylight.mdsal.binding.api.TransactionChain;
39 import org.opendaylight.mdsal.binding.api.TransactionChainListener;
40 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.topology.discovery.rev130819.LinkDiscoveredBuilder;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.topology.discovery.rev130819.LinkRemovedBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
44 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.LinkId;
45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.LinkKey;
51 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
52 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
54 public class FlowCapableTopologyExporterTest {
56 private OperationProcessor processor;
57 private FlowCapableTopologyExporter exporter;
58 private InstanceIdentifier<Topology> topologyIID;
59 private final ExecutorService executor = Executors.newFixedThreadPool(1);
61 private DataBroker mockDataBroker;
63 private TransactionChain mockTxChain;
67 MockitoAnnotations.initMocks(this);
69 doReturn(mockTxChain).when(mockDataBroker)
70 .createTransactionChain(any(TransactionChainListener.class));
72 processor = new OperationProcessor(mockDataBroker);
74 topologyIID = InstanceIdentifier.create(NetworkTopology.class)
75 .child(Topology.class, new TopologyKey(new TopologyId("flow:1")));
76 exporter = new FlowCapableTopologyExporter(processor, topologyIID);
77 executor.execute(processor);
81 public void tearDown() {
82 executor.shutdownNow();
87 public void testOnLinkDiscovered() {
89 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
90 sourceNodeKey = newInvNodeKey("sourceNode");
91 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
92 sourceNodeConnKey = newInvNodeConnKey("sourceTP");
93 InstanceIdentifier<?> sourceConnID = newNodeConnID(sourceNodeKey, sourceNodeConnKey);
95 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
96 destNodeKey = newInvNodeKey("destNode");
97 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
98 destNodeConnKey = newInvNodeConnKey("destTP");
99 InstanceIdentifier<?> destConnID = newNodeConnID(destNodeKey, destNodeConnKey);
101 ReadWriteTransaction mockTx = mock(ReadWriteTransaction.class);
102 CountDownLatch submitLatch = setupStubbedSubmit(mockTx);
103 doReturn(mockTx).when(mockTxChain).newReadWriteTransaction();
105 exporter.onLinkDiscovered(new LinkDiscoveredBuilder().setSource(
106 new NodeConnectorRef(sourceConnID)).setDestination(
107 new NodeConnectorRef(destConnID)).build());
109 waitForSubmit(submitLatch);
111 ArgumentCaptor<Link> mergedNode = ArgumentCaptor.forClass(Link.class);
112 verify(mockTx).merge(eq(LogicalDatastoreType.OPERATIONAL), eq(topologyIID.child(
113 Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId())))),
114 mergedNode.capture(), eq(true));
115 assertEquals("Source node ID", "sourceNode",
116 mergedNode.getValue().getSource().getSourceNode().getValue());
117 assertEquals("Dest TP ID", "sourceTP",
118 mergedNode.getValue().getSource().getSourceTp().getValue());
119 assertEquals("Dest node ID", "destNode",
120 mergedNode.getValue().getDestination().getDestNode().getValue());
121 assertEquals("Dest TP ID", "destTP",
122 mergedNode.getValue().getDestination().getDestTp().getValue());
126 public void testOnLinkRemoved() {
128 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
129 sourceNodeKey = newInvNodeKey("sourceNode");
130 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
131 sourceNodeConnKey = newInvNodeConnKey("sourceTP");
132 InstanceIdentifier<?> sourceConnID = newNodeConnID(sourceNodeKey, sourceNodeConnKey);
134 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
135 destNodeKey = newInvNodeKey("destNode");
136 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
137 destNodeConnKey = newInvNodeConnKey("destTP");
138 InstanceIdentifier<?> destConnID = newNodeConnID(destNodeKey, destNodeConnKey);
140 Link link = newLink(sourceNodeConnKey.getId().getValue(), newSourceTp(sourceNodeConnKey.getId().getValue()),
141 newDestTp(destNodeConnKey.getId().getValue()));
143 ReadWriteTransaction mockTx = mock(ReadWriteTransaction.class);
144 final CountDownLatch submitLatch = setupStubbedSubmit(mockTx);
145 doReturn(mockTx).when(mockTxChain).newReadWriteTransaction();
146 doReturn(FluentFutures.immediateFluentFuture(Optional.of(link))).when(mockTx)
147 .read(LogicalDatastoreType.OPERATIONAL,
148 topologyIID.child(Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));
150 exporter.onLinkRemoved(new LinkRemovedBuilder().setSource(
151 new NodeConnectorRef(sourceConnID)).setDestination(
152 new NodeConnectorRef(destConnID)).build());
154 waitForSubmit(submitLatch);
156 verify(mockTx).delete(LogicalDatastoreType.OPERATIONAL, topologyIID.child(
157 Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));
161 public void testOnLinkRemovedLinkDoesNotExist() {
163 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
164 sourceNodeKey = newInvNodeKey("sourceNode");
165 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
166 sourceNodeConnKey = newInvNodeConnKey("sourceTP");
167 InstanceIdentifier<?> sourceConnID = newNodeConnID(sourceNodeKey, sourceNodeConnKey);
169 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey
170 destNodeKey = newInvNodeKey("destNode");
171 org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey
172 destNodeConnKey = newInvNodeConnKey("destTP");
173 InstanceIdentifier<?> destConnID = newNodeConnID(destNodeKey, destNodeConnKey);
175 ReadWriteTransaction mockTx = mock(ReadWriteTransaction.class);
176 final CountDownLatch submitLatch = setupStubbedSubmit(mockTx);
177 doReturn(mockTx).when(mockTxChain).newReadWriteTransaction();
178 doReturn(FluentFutures.immediateFluentFuture(Optional.empty())).when(mockTx)
179 .read(LogicalDatastoreType.OPERATIONAL,
180 topologyIID.child(Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));
182 exporter.onLinkRemoved(new LinkRemovedBuilder().setSource(
183 new NodeConnectorRef(sourceConnID)).setDestination(
184 new NodeConnectorRef(destConnID)).build());
186 waitForSubmit(submitLatch);
188 verify(mockTx, never()).delete(LogicalDatastoreType.OPERATIONAL, topologyIID.child(
189 Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));