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.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;
  25
  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;
  53
  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         MockitoAnnotations.initMocks(this);
  68
  69         doReturn(mockTxChain).when(mockDataBroker)
  70                 .createTransactionChain(any(TransactionChainListener.class));
  71
  72         processor = new OperationProcessor(mockDataBroker);
  73
  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);
  78     }
  79
  80     @After
  81     public void tearDown() {
  82         executor.shutdownNow();
  83     }
  84
  85
  86     @Test
  87     public void testOnLinkDiscovered() {
  88
  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);
  94
  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);
 100
 101         ReadWriteTransaction mockTx = mock(ReadWriteTransaction.class);
 102         CountDownLatch submitLatch = setupStubbedSubmit(mockTx);
 103         doReturn(mockTx).when(mockTxChain).newReadWriteTransaction();
 104
 105         exporter.onLinkDiscovered(new LinkDiscoveredBuilder().setSource(
 106                 new NodeConnectorRef(sourceConnID)).setDestination(
 107                         new NodeConnectorRef(destConnID)).build());
 108
 109         waitForSubmit(submitLatch);
 110
 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());
 123     }
 124
 125     @Test
 126     public void testOnLinkRemoved() {
 127
 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);
 133
 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);
 139
 140         Link link = newLink(sourceNodeConnKey.getId().getValue(), newSourceTp(sourceNodeConnKey.getId().getValue()),
 141                 newDestTp(destNodeConnKey.getId().getValue()));
 142
 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()))));
 149
 150         exporter.onLinkRemoved(new LinkRemovedBuilder().setSource(
 151                 new NodeConnectorRef(sourceConnID)).setDestination(
 152                 new NodeConnectorRef(destConnID)).build());
 153
 154         waitForSubmit(submitLatch);
 155
 156         verify(mockTx).delete(LogicalDatastoreType.OPERATIONAL, topologyIID.child(
 157                 Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));
 158     }
 159
 160     @Test
 161     public void testOnLinkRemovedLinkDoesNotExist() {
 162
 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);
 168
 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);
 174
 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()))));
 181
 182         exporter.onLinkRemoved(new LinkRemovedBuilder().setSource(
 183                 new NodeConnectorRef(sourceConnID)).setDestination(
 184                 new NodeConnectorRef(destConnID)).build());
 185
 186         waitForSubmit(submitLatch);
 187
 188         verify(mockTx, never()).delete(LogicalDatastoreType.OPERATIONAL, topologyIID.child(
 189                 Link.class, new LinkKey(new LinkId(sourceNodeConnKey.getId()))));
 190     }
 191 }