pcep/tunnel/tunnel-provider/src/test/java/org/opendaylight/bgpcep/pcep/tunnel/provider/NodeChangedListenerTest.java

   1 /*
   2  * Copyright (c) 2014 Cisco 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.bgpcep.pcep.tunnel.provider;
   9
  10 import static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.junit.Assert.assertNull;
  13 import static org.opendaylight.protocol.util.CheckTestUtil.readDataOperational;
  14
  15 import com.google.common.collect.Lists;
  16 import java.util.Collections;
  17 import java.util.concurrent.ExecutionException;
  18 import org.junit.After;
  19 import org.junit.Assert;
  20 import org.junit.Before;
  21 import org.junit.Test;
  22 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
  23 import org.opendaylight.mdsal.binding.api.WriteTransaction;
  24 import org.opendaylight.mdsal.binding.dom.adapter.test.AbstractConcurrentDataBrokerTest;
  25 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressNoZone;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4AddressNoZone;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Path1;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.Path1Builder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.LspIdentifiersBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.lsp.identifiers.address.family.Ipv4CaseBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.identifiers.tlv.lsp.identifiers.address.family.ipv4._case.Ipv4Builder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.LspBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.lsp.TlvsBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.bandwidth.object.BandwidthBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.Ipv4ExtendedTunnelId;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev150820.LspId;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.Node1;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.Node1Builder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.PccSyncState;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.PathComputationClientBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.path.computation.client.ReportedLsp;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.path.computation.client.ReportedLspBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.path.computation.client.ReportedLspKey;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.path.computation.client.reported.lsp.PathBuilder;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev181109.pcep.client.attributes.path.computation.client.reported.lsp.PathKey;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  51 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  52 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  53 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  54 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  55 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  57 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
  58 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  59 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  60 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  61 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  62 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNode;
  63 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  64 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  65 import org.opendaylight.yangtools.yang.common.Uint32;
  66
  67 public class NodeChangedListenerTest extends AbstractConcurrentDataBrokerTest {
  68
  69     private static final TopologyId PCEP_TOPOLOGY_ID = new TopologyId("pcep-topology");
  70     private static final TopologyId TUNNEL_TOPOLOGY_ID = new TopologyId("tunnel-topology");
  71
  72     private static final String NODE1_IPV4 = "39.39.39.39";
  73     private static final NodeId NODE1_ID = new NodeId("pcc://" + NODE1_IPV4);
  74     private static final String LSP1_NAME = "lsp1";
  75     private static final Uint32 LSP1_ID = Uint32.ONE;
  76
  77     private static final String NODE2_IPV4 = "40.40.40.40";
  78     private static final NodeId NODE2_ID = new NodeId("pcc://" + NODE2_IPV4);
  79     private static final String LSP2_NAME = "lsp2";
  80     private static final Uint32 LSP2_ID = Uint32.TWO;
  81
  82     private static final InstanceIdentifier<Topology> PCEP_TOPO_IID = InstanceIdentifier.builder(NetworkTopology.class)
  83             .child(Topology.class, new TopologyKey(PCEP_TOPOLOGY_ID)).build();
  84     private static final InstanceIdentifier<Topology> TUNNEL_TOPO_IID = InstanceIdentifier
  85             .builder(NetworkTopology.class).child(Topology.class, new TopologyKey(TUNNEL_TOPOLOGY_ID)).build();
  86
  87     private ListenerRegistration<NodeChangedListener> listenerRegistration;
  88
  89     @Before
  90     public void setUp() throws InterruptedException, ExecutionException {
  91         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
  92         wTx.mergeParentStructurePut(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID, new TopologyBuilder()
  93                 .withKey(new TopologyKey(PCEP_TOPOLOGY_ID)).setNode(Collections.emptyList())
  94                 .setTopologyId(PCEP_TOPOLOGY_ID).build());
  95         wTx.mergeParentStructurePut(LogicalDatastoreType.OPERATIONAL, TUNNEL_TOPO_IID, new TopologyBuilder()
  96                 .withKey(new TopologyKey(TUNNEL_TOPOLOGY_ID)).setTopologyId(TUNNEL_TOPOLOGY_ID).build());
  97         wTx.commit().get();
  98         final NodeChangedListener nodeListener = new NodeChangedListener(getDataBroker(),
  99                 PCEP_TOPOLOGY_ID, TUNNEL_TOPO_IID);
 100         this.listenerRegistration = getDataBroker().registerDataTreeChangeListener(DataTreeIdentifier.create(
 101                 LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.child(Node.class)), nodeListener);
 102     }
 103
 104     @Test
 105     public void testNodeChangedListener() throws InterruptedException, ExecutionException {
 106         // add node -> create two nodes with TPs and link
 107         createNode(NODE1_ID, NODE1_IPV4, LSP1_NAME, LSP1_ID, NODE2_IPV4);
 108         final Topology tunnelTopo = readDataOperational(getDataBroker(), TUNNEL_TOPO_IID, tunnelTopo1 -> {
 109             assertNotNull(tunnelTopo1.getNode());
 110             assertEquals(2, tunnelTopo1.getNode().size());
 111             return tunnelTopo1;
 112         });
 113
 114         final NodeId srcId = new NodeId("ip://" + new IpAddress(new Ipv4Address(NODE1_IPV4)));
 115         final NodeId dstId = new NodeId("ip://" + new IpAddress(new Ipv4Address(NODE2_IPV4)));
 116
 117         final Node dst;
 118         final Node src;
 119
 120         if (tunnelTopo.getNode().get(0).getNodeId().equals(srcId)) {
 121             src = tunnelTopo.getNode().get(0);
 122             dst = tunnelTopo.getNode().get(1);
 123         } else {
 124             src = tunnelTopo.getNode().get(1);
 125             dst = tunnelTopo.getNode().get(0);
 126         }
 127
 128         Assert.assertEquals(srcId, src.getNodeId());
 129         Assert.assertEquals(dstId, dst.getNodeId());
 130
 131         Assert.assertEquals(1, dst.getTerminationPoint().size());
 132         Assert.assertEquals(1, src.getTerminationPoint().size());
 133         final TerminationPoint dstTp = dst.getTerminationPoint().get(0);
 134         final TerminationPoint srcTp = src.getTerminationPoint().get(0);
 135         final TpId dstNodeTpId = new TpId(dstId.getValue());
 136         final TpId srcNodeTpId = new TpId(srcId.getValue());
 137         Assert.assertEquals(dstNodeTpId, dstTp.getTpId());
 138         Assert.assertEquals(srcNodeTpId, srcTp.getTpId());
 139
 140         Assert.assertEquals(1, src.getSupportingNode().size());
 141         Assert.assertNull(dst.getSupportingNode());
 142         final SupportingNode sNode = src.getSupportingNode().get(0);
 143         Assert.assertEquals(NODE1_ID, sNode.key().getNodeRef());
 144
 145         Assert.assertEquals(1, tunnelTopo.getLink().size());
 146         final Link link = tunnelTopo.getLink().get(0);
 147         Assert.assertEquals(srcId, link.getSource().getSourceNode());
 148         Assert.assertEquals(srcNodeTpId, link.getSource().getSourceTp());
 149         Assert.assertEquals(dstId, link.getDestination().getDestNode());
 150         Assert.assertEquals(dstNodeTpId, link.getDestination().getDestTp());
 151
 152         // update second node -> adds supporting node and second link
 153         createNode(NODE2_ID, NODE2_IPV4, LSP2_NAME, LSP2_ID, NODE1_IPV4);
 154         readDataOperational(getDataBroker(), TUNNEL_TOPO_IID, updatedNodeTopo -> {
 155             assertNotNull(updatedNodeTopo.getNode());
 156             Assert.assertEquals(2, updatedNodeTopo.getNode().size());
 157             final Node updatedNode;
 158             if (updatedNodeTopo.getNode().get(0).getNodeId().equals(srcId)) {
 159                 updatedNode = updatedNodeTopo.getNode().get(1);
 160             } else {
 161                 updatedNode = updatedNodeTopo.getNode().get(0);
 162             }
 163
 164             assertNotNull(updatedNode.getSupportingNode());
 165             Assert.assertEquals(1, updatedNode.getSupportingNode().size());
 166             final SupportingNode sNode2 = updatedNode.getSupportingNode().get(0);
 167             Assert.assertEquals(NODE2_ID, sNode2.getNodeRef());
 168             Assert.assertEquals(2, updatedNodeTopo.getLink().size());
 169             return updatedNodeTopo;
 170
 171         });
 172
 173         readDataOperational(getDataBroker(), TUNNEL_TOPO_IID, updatedNodeTopo -> {
 174             final Link link2;
 175             if (updatedNodeTopo.getLink().get(0).getSource().getSourceNode().equals(srcId)) {
 176                 link2 = updatedNodeTopo.getLink().get(1);
 177             } else {
 178                 link2 = updatedNodeTopo.getLink().get(0);
 179             }
 180             assertEquals(dstId, link2.getSource().getSourceNode());
 181             assertEquals(dstNodeTpId, link2.getSource().getSourceTp());
 182             assertEquals(srcId, link2.getDestination().getDestNode());
 183             assertEquals(srcNodeTpId, link2.getDestination().getDestTp());
 184             return updatedNodeTopo;
 185         });
 186
 187         // remove nodes -> remove link
 188         removeNode(NODE1_ID);
 189         removeNode(NODE2_ID);
 190         readDataOperational(getDataBroker(), TUNNEL_TOPO_IID, removedNodeTopo -> {
 191             assertNull(removedNodeTopo.getNode());
 192             assertNull(removedNodeTopo.getLink());
 193             return removedNodeTopo;
 194         });
 195     }
 196
 197     @After
 198     public void tearDown() {
 199         this.listenerRegistration.close();
 200     }
 201
 202     private void createNode(final NodeId nodeId, final String ipv4Address, final String lspName, final Uint32 lspId,
 203             final String dstIpv4Address) throws InterruptedException, ExecutionException {
 204         final NodeBuilder nodeBuilder = new NodeBuilder();
 205         nodeBuilder.withKey(new NodeKey(nodeId));
 206         nodeBuilder.setNodeId(nodeId);
 207         final PathBuilder pathBuilder = new PathBuilder();
 208         pathBuilder.withKey(new PathKey(new LspId(lspId)));
 209         pathBuilder.setBandwidth(new BandwidthBuilder().setBandwidth(
 210                 new Bandwidth(new byte[]{0x00, 0x00, (byte) 0xff, (byte) 0xff})).build());
 211         pathBuilder.addAugmentation(Path1.class, new Path1Builder().setLsp(new LspBuilder().setTlvs(new TlvsBuilder()
 212                 .setLspIdentifiers(new LspIdentifiersBuilder().setAddressFamily(new Ipv4CaseBuilder().setIpv4(
 213                         new Ipv4Builder().setIpv4TunnelSenderAddress(new Ipv4AddressNoZone(ipv4Address))
 214                                 .setIpv4ExtendedTunnelId(new Ipv4ExtendedTunnelId(ipv4Address))
 215                                 .setIpv4TunnelEndpointAddress(new Ipv4AddressNoZone(dstIpv4Address))
 216                                 .build()).build()).build()).build()).setAdministrative(true)
 217                 .setDelegate(true).build()).build());
 218         final ReportedLsp reportedLps = new ReportedLspBuilder().withKey(new ReportedLspKey(lspName)).setPath(
 219                 Collections.singletonList(pathBuilder.build())).build();
 220         final Node1Builder node1Builder = new Node1Builder();
 221         node1Builder.setPathComputationClient(new PathComputationClientBuilder()
 222                 .setStateSync(PccSyncState.Synchronized)
 223                 .setReportedLsp(Lists.newArrayList(reportedLps))
 224                 .setIpAddress(new IpAddressNoZone(new Ipv4AddressNoZone(ipv4Address)))
 225                 .build());
 226         nodeBuilder.addAugmentation(Node1.class, node1Builder.build());
 227         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
 228         wTx.put(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.builder().child(Node.class,
 229                 new NodeKey(nodeId)).build(), nodeBuilder.build());
 230         wTx.commit().get();
 231     }
 232
 233     private void removeNode(final NodeId nodeId) throws InterruptedException, ExecutionException {
 234         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
 235         wTx.delete(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.builder()
 236                 .child(Node.class, new NodeKey(nodeId)).build());
 237         wTx.commit().get();
 238     }
 239 }