applications/topology-manager/src/test/java/org/opendaylight/openflowplugin/applications/topology/manager/FlowCapableTopologyExporterTest.java

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