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