renderers/vpp/src/test/java/org/opendaylight/groupbasedpolicy/renderer/vpp/manager/BridgeDomainManagerImplTest.java

   1 /*
   2  * Copyright (c) 2016 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.groupbasedpolicy.renderer.vpp.manager;
  10
  11 import java.util.Arrays;
  12 import java.util.Collection;
  13
  14 import org.junit.Assert;
  15 import org.junit.Before;
  16 import org.junit.Test;
  17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  18 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  19 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  20 import org.opendaylight.groupbasedpolicy.renderer.vpp.util.VppIidFactory;
  21 import org.opendaylight.groupbasedpolicy.test.CustomDataBrokerTest;
  22 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev150105.VxlanVni;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyTypesVbridgeAugment;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyTypesVbridgeAugmentBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyVbridgeAugment;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyVbridgeAugmentBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.network.topology.topology.topology.types.VbridgeTopologyBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.tunnel.vxlan.rev160429.TunnelTypeVxlan;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.tunnel.vxlan.rev160429.network.topology.topology.tunnel.parameters.VxlanTunnelParametersBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
  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.Node;
  38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  39 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.TopologyTypesBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNodeBuilder;
  42
  43 import com.google.common.base.Optional;
  44 import com.google.common.util.concurrent.ListenableFuture;
  45
  46 public class BridgeDomainManagerImplTest extends CustomDataBrokerTest {
  47
  48     private static final TopologyId SUPPORTING_TOPOLOGY_NETCONF = new TopologyId("topology-netconf");
  49     private final static String BRIDGE_DOMAIN_ID = "bridge-domain-test";
  50     private final static TopologyId BASE_TOPOLOGY_ID = new TopologyId(BRIDGE_DOMAIN_ID);
  51     private final static VxlanVni BRIDGE_DOMAIN_VNI = new VxlanVni(10L);
  52     private final static boolean BRIDGE_DOMAIN_FLOOD = true;
  53     private final static boolean BRIDGE_DOMAIN_FORWARD = true;
  54     private final static boolean BRIDGE_DOMAIN_LEARN = true;
  55     private final static boolean BRIDGE_DOMAIN_UNICAST_FLOOD = true;
  56     private final static boolean BRIDGE_DOMAIN_ARP = false;
  57     private final static NodeId VPP_NODE_ID = new NodeId("vppNode");
  58     private final static Topology BASE_TOPOLOGY = new TopologyBuilder().setTopologyId(BASE_TOPOLOGY_ID)
  59         .setNode(Arrays.asList(new NodeBuilder().setNodeId(VPP_NODE_ID)
  60             .setSupportingNode(Arrays.asList(new SupportingNodeBuilder().setTopologyRef(SUPPORTING_TOPOLOGY_NETCONF)
  61                 .setNodeRef(VPP_NODE_ID)
  62                 .build()))
  63             .build()))
  64         .setTopologyTypes(new TopologyTypesBuilder()
  65             .addAugmentation(TopologyTypesVbridgeAugment.class, new TopologyTypesVbridgeAugmentBuilder()
  66                 .setVbridgeTopology(new VbridgeTopologyBuilder().build()).build())
  67             .build())
  68         .addAugmentation(TopologyVbridgeAugment.class,
  69                 new TopologyVbridgeAugmentBuilder().setTunnelType(TunnelTypeVxlan.class)
  70                     .setTunnelParameters(new VxlanTunnelParametersBuilder().setVni(BRIDGE_DOMAIN_VNI).build())
  71                     .setForward(BRIDGE_DOMAIN_FORWARD)
  72                     .setUnknownUnicastFlood(BRIDGE_DOMAIN_UNICAST_FLOOD)
  73                     .setLearn(BRIDGE_DOMAIN_LEARN)
  74                     .setArpTermination(BRIDGE_DOMAIN_ARP)
  75                     .setFlood(BRIDGE_DOMAIN_FLOOD)
  76                     .build())
  77         .build();
  78
  79     private DataBroker dataBroker;
  80     private BridgeDomainManagerImpl bridgeDomainManager;
  81
  82     @Override
  83     public Collection<Class<?>> getClassesFromModules() {
  84         return Arrays.asList(NetworkTopology.class, Topology.class, TopologyVbridgeAugment.class,
  85                 TunnelTypeVxlan.class);
  86     }
  87
  88     @Before
  89     public void init() {
  90         dataBroker = getDataBroker();
  91         bridgeDomainManager = new BridgeDomainManagerImpl(dataBroker);
  92     }
  93
  94     @Test
  95     public void testCreateVxlanBridgeDomainOnVppNode() {
  96         ListenableFuture<Void> registered =
  97                 bridgeDomainManager.createVxlanBridgeDomainOnVppNode(BRIDGE_DOMAIN_ID, BRIDGE_DOMAIN_VNI, VPP_NODE_ID);
  98         Assert.assertTrue(registered.isDone());
  99
 100         Optional<Topology> topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 101                 VppIidFactory.getTopologyIid(BASE_TOPOLOGY.getKey()), dataBroker.newReadOnlyTransaction());
 102         Assert.assertTrue(topologyOptional.isPresent());
 103
 104         Topology topology = topologyOptional.get();
 105         Assert.assertEquals(BASE_TOPOLOGY, topology);
 106     }
 107
 108     @Test
 109     public void testRemoveBridgeDomainFromVppNode() {
 110         WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
 111
 112         writeTransaction.put(LogicalDatastoreType.CONFIGURATION, VppIidFactory.getTopologyIid(BASE_TOPOLOGY.getKey()),
 113                 BASE_TOPOLOGY, true);
 114         boolean result = DataStoreHelper.submitToDs(writeTransaction);
 115         Assert.assertTrue(result);
 116
 117         Optional<Node> topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 118                 VppIidFactory.getNodeIid(new TopologyKey(BASE_TOPOLOGY_ID), new NodeKey(VPP_NODE_ID)),
 119                 dataBroker.newReadOnlyTransaction());
 120         Assert.assertTrue(topologyOptional.isPresent());
 121
 122         ListenableFuture<Void> deleted =
 123                 bridgeDomainManager.removeBridgeDomainFromVppNode(BRIDGE_DOMAIN_ID, VPP_NODE_ID);
 124         Assert.assertTrue(deleted.isDone());
 125
 126         topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 127                 VppIidFactory.getNodeIid(new TopologyKey(BASE_TOPOLOGY_ID), new NodeKey(VPP_NODE_ID)),
 128                 dataBroker.newReadOnlyTransaction());
 129         Assert.assertFalse(topologyOptional.isPresent());
 130     }
 131
 132 }