pcep/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
   9 package org.opendaylight.bgpcep.pcep.tunnel.provider;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.collect.Lists;
  13 import org.junit.After;
  14 import org.junit.Assert;
  15 import org.junit.Before;
  16 import org.junit.Test;
  17 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  18 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
  19 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  20 import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
  21 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
  22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  24 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Path1;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.Path1Builder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.LspIdentifiersBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.lsp.identifiers.address.family.Ipv4CaseBuilder;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.lsp.identifiers.address.family.ipv4._case.Ipv4Builder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.BandwidthBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.Ipv4ExtendedTunnelId;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.LspId;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.Node1;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.Node1Builder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.PccSyncState;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.PathComputationClientBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLsp;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLspBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.ReportedLspKey;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.PathBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.topology.pcep.rev131024.pcep.client.attributes.path.computation.client.reported.lsp.PathKey;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  51 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  52 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
  53 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  54 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Link;
  55 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  56 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  57 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  58 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  59 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNode;
  60 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  61 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  62
  63 public class NodeChangedListenerTest extends AbstractDataBrokerTest {
  64
  65     private static final TopologyId PCEP_TOPOLOGY_ID = new TopologyId("pcep-topology");
  66     private static final TopologyId TUNNEL_TOPOLOGY_ID = new TopologyId("tunnel-topology");
  67
  68     private static final String NODE1_IPV4 = "39.39.39.39";
  69     private static final NodeId NODE1_ID = new NodeId("pcc://" + NODE1_IPV4);
  70     private static final String LSP1_NAME = "lsp1";
  71     private static final long LSP1_ID = 1;
  72
  73     private static final String NODE2_IPV4 = "40.40.40.40";
  74     private static final NodeId NODE2_ID = new NodeId("pcc://" + NODE2_IPV4);
  75     private static final String LSP2_NAME = "lsp2";
  76     private static final long LSP2_ID = 2;
  77
  78     private static final InstanceIdentifier<Topology> PCEP_TOPO_IID = InstanceIdentifier.builder(NetworkTopology.class).child(Topology.class, new TopologyKey(PCEP_TOPOLOGY_ID)).toInstance();
  79     private static final InstanceIdentifier<Topology> TUNNEL_TOPO_IID = InstanceIdentifier.builder(NetworkTopology.class).child(Topology.class, new TopologyKey(TUNNEL_TOPOLOGY_ID)).toInstance();
  80
  81     private ListenerRegistration<DataChangeListener> listenerRegistration;
  82
  83     @Before
  84     public void setUp() throws TransactionCommitFailedException {
  85         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
  86         wTx.put(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID, new TopologyBuilder().setKey(new TopologyKey(PCEP_TOPOLOGY_ID)).setNode(Lists.<Node>newArrayList()).setTopologyId(PCEP_TOPOLOGY_ID).build(), true);
  87         wTx.put(LogicalDatastoreType.OPERATIONAL, TUNNEL_TOPO_IID, new TopologyBuilder().setKey(new TopologyKey(TUNNEL_TOPOLOGY_ID)).setTopologyId(TUNNEL_TOPOLOGY_ID).build(), true);
  88         wTx.submit().checkedGet();
  89         final NodeChangedListener nodeListener = new NodeChangedListener(getDataBroker(), TUNNEL_TOPO_IID);
  90         this.listenerRegistration = getDataBroker().registerDataChangeListener(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.child(Node.class), nodeListener, DataChangeScope.SUBTREE);
  91     }
  92
  93     @Test
  94     public void testNodeChangedListener() throws ReadFailedException, TransactionCommitFailedException {
  95         // add node -> create two nodes with TPs and link
  96         createNode(NODE1_ID, NODE1_IPV4, LSP1_NAME, LSP1_ID, NODE2_IPV4);
  97         final Topology tunnelTopo = readTunnelTopology().get();
  98         Assert.assertEquals(2, tunnelTopo.getNode().size());
  99         final Node dst = tunnelTopo.getNode().get(0);
 100         final Node src = tunnelTopo.getNode().get(1);
 101         final NodeId srcId = new NodeId("ip://" + new IpAddress(new Ipv4Address(NODE1_IPV4)));
 102         final NodeId dstId = new NodeId("ip://" + new IpAddress(new Ipv4Address(NODE2_IPV4)));
 103         Assert.assertEquals(srcId, src.getNodeId());
 104         Assert.assertEquals(dstId, dst.getNodeId());
 105
 106         Assert.assertEquals(1, dst.getTerminationPoint().size());
 107         Assert.assertEquals(1, src.getTerminationPoint().size());
 108         final TerminationPoint dstTp = dst.getTerminationPoint().get(0);
 109         final TerminationPoint srcTp = src.getTerminationPoint().get(0);
 110         final TpId dstNodeTpId = new TpId(dstId.getValue());
 111         final TpId srcNodeTpId = new TpId(srcId.getValue());
 112         Assert.assertEquals(dstNodeTpId, dstTp.getTpId());
 113         Assert.assertEquals(srcNodeTpId, srcTp.getTpId());
 114
 115         Assert.assertEquals(1, src.getSupportingNode().size());
 116         Assert.assertNull(dst.getSupportingNode());
 117         final SupportingNode sNode = src.getSupportingNode().get(0);
 118         Assert.assertEquals(NODE1_ID, sNode.getKey().getNodeRef());
 119
 120         Assert.assertEquals(1, tunnelTopo.getLink().size());
 121         final Link link = tunnelTopo.getLink().get(0);
 122         Assert.assertEquals(srcId, link.getSource().getSourceNode());
 123         Assert.assertEquals(srcNodeTpId, link.getSource().getSourceTp());
 124         Assert.assertEquals(dstId, link.getDestination().getDestNode());
 125         Assert.assertEquals(dstNodeTpId, link.getDestination().getDestTp());
 126
 127         // update second node -> adds supporting node and second link
 128         createNode(NODE2_ID, NODE2_IPV4, LSP2_NAME, LSP2_ID, NODE1_IPV4);
 129         final Topology updatedNodeTopo = readTunnelTopology().get();
 130         final Node updatedNode = updatedNodeTopo.getNode().get(0);
 131         Assert.assertEquals(1, updatedNode.getSupportingNode().size());
 132         final SupportingNode sNode2 = updatedNode.getSupportingNode().get(0);
 133         Assert.assertEquals(NODE2_ID, sNode2.getNodeRef());
 134
 135         Assert.assertEquals(2, updatedNodeTopo.getLink().size());
 136         final Link link2 = updatedNodeTopo.getLink().get(0);
 137         Assert.assertEquals(dstId, link2.getSource().getSourceNode());
 138         Assert.assertEquals(dstNodeTpId, link2.getSource().getSourceTp());
 139         Assert.assertEquals(srcId, link2.getDestination().getDestNode());
 140         Assert.assertEquals(srcNodeTpId, link2.getDestination().getDestTp());
 141
 142         // remove nodes -> remove link
 143         removeNode(NODE1_ID);
 144         removeNode(NODE2_ID);
 145         final Topology removedNodeTopo = readTunnelTopology().get();
 146         Assert.assertEquals(0, removedNodeTopo.getNode().size());
 147         Assert.assertEquals(0, removedNodeTopo.getLink().size());
 148     }
 149
 150     @After
 151     public void tearDown() {
 152         this.listenerRegistration.close();
 153     }
 154
 155     private void createNode(final NodeId nodeId, final String ipv4Address, final String lspName, final long lspId, final String dstIpv4Address) throws TransactionCommitFailedException {
 156         final NodeBuilder nodeBuilder = new NodeBuilder();
 157         nodeBuilder.setKey(new NodeKey(nodeId));
 158         nodeBuilder.setNodeId(nodeId);
 159         final PathBuilder pathBuilder = new PathBuilder();
 160         pathBuilder.setKey(new PathKey(new LspId(lspId)));
 161         pathBuilder.setBandwidth(new BandwidthBuilder().setBandwidth(new Bandwidth(new byte[] {0x00, 0x00, (byte) 0xff, (byte) 0xff})).build());
 162         pathBuilder.addAugmentation(Path1.class, new Path1Builder().setLsp(new LspBuilder().setTlvs(new TlvsBuilder().setLspIdentifiers(new LspIdentifiersBuilder().setAddressFamily(new Ipv4CaseBuilder().setIpv4(new Ipv4Builder().setIpv4TunnelSenderAddress(new Ipv4Address(ipv4Address)).setIpv4ExtendedTunnelId(new Ipv4ExtendedTunnelId(ipv4Address)).setIpv4TunnelEndpointAddress(new Ipv4Address(dstIpv4Address)).build()).build()).build()).build()).setAdministrative(true).setDelegate(true).build()).build());
 163         final ReportedLsp reportedLps = new ReportedLspBuilder().setKey(new ReportedLspKey(lspName)).setPath(Lists.newArrayList(pathBuilder.build())).build();
 164         final Node1Builder node1Builder = new Node1Builder();
 165         node1Builder.setPathComputationClient(new PathComputationClientBuilder().setStateSync(PccSyncState.Synchronized).setReportedLsp(Lists.newArrayList(reportedLps)).setIpAddress(new IpAddress(new Ipv4Address(ipv4Address))).build());
 166         nodeBuilder.addAugmentation(Node1.class, node1Builder.build());
 167         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
 168         wTx.put(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.builder().child(Node.class, new NodeKey(nodeId)).toInstance(), nodeBuilder.build());
 169         wTx.submit().checkedGet();
 170     }
 171
 172     private void removeNode(final NodeId nodeId) throws TransactionCommitFailedException {
 173         final WriteTransaction wTx = getDataBroker().newWriteOnlyTransaction();
 174         wTx.delete(LogicalDatastoreType.OPERATIONAL, PCEP_TOPO_IID.builder().child(Node.class, new NodeKey(nodeId)).toInstance());
 175         wTx.submit().checkedGet();
 176     }
 177
 178     private Optional<Topology> readTunnelTopology() throws ReadFailedException {
 179         final ReadTransaction rTx = getDataBroker().newReadOnlyTransaction();
 180         return rTx.read(LogicalDatastoreType.OPERATIONAL, TUNNEL_TOPO_IID).checkedGet();
 181     }
 182 }