renderers/vpp/src/test/java/org/opendaylight/groupbasedpolicy/renderer/vpp/policy/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.policy;
  10
  11 import java.util.Arrays;
  12 import java.util.Collection;
  13 import java.util.Collections;
  14 import java.util.concurrent.ExecutionException;
  15
  16 import javax.annotation.Nonnull;
  17
  18 import org.junit.Assert;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.mockito.Mockito;
  22 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  23 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  24 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  25 import org.opendaylight.groupbasedpolicy.renderer.vpp.iface.InterfaceManager;
  26 import org.opendaylight.groupbasedpolicy.renderer.vpp.lisp.LispStateManager;
  27 import org.opendaylight.groupbasedpolicy.renderer.vpp.lisp.loopback.LoopbackManager;
  28 import org.opendaylight.groupbasedpolicy.renderer.vpp.lisp.flat.overlay.FlatOverlayManager;
  29 import org.opendaylight.groupbasedpolicy.renderer.vpp.nat.NatManager;
  30 import org.opendaylight.groupbasedpolicy.renderer.vpp.policy.acl.AclManager;
  31 import org.opendaylight.groupbasedpolicy.renderer.vpp.routing.RoutingManager;
  32 import org.opendaylight.groupbasedpolicy.renderer.vpp.util.VppIidFactory;
  33 import org.opendaylight.groupbasedpolicy.test.CustomDataBrokerTest;
  34 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.v3po.rev170315.VxlanVni;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.NodeVbridgeAugment;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.NodeVbridgeAugmentBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyTypesVbridgeAugment;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyTypesVbridgeAugmentBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyVbridgeAugment;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.TopologyVbridgeAugmentBuilder;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.network.topology.topology.node.BridgeMemberBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.topology.rev160129.network.topology.topology.topology.types.VbridgeTopologyBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.tunnel.vxlan.rev170327.TunnelTypeVxlan;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.vbridge.tunnel.vxlan.rev170327.network.topology.topology.tunnel.parameters.VxlanTunnelParametersBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
  51 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
  52 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  53 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  55 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.TopologyTypesBuilder;
  56 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.node.attributes.SupportingNodeBuilder;
  57
  58 import com.google.common.base.Optional;
  59
  60 public class BridgeDomainManagerImplTest extends CustomDataBrokerTest {
  61
  62     private static final TopologyId SUPPORTING_TOPOLOGY_NETCONF = new TopologyId("topology-netconf");
  63     private final static String BRIDGE_DOMAIN_ID = "bridge-domain-test";
  64     private final static TopologyId BASE_TOPOLOGY_ID = new TopologyId(BRIDGE_DOMAIN_ID);
  65     private final static VxlanVni BRIDGE_DOMAIN_VNI = new VxlanVni(10L);
  66     private final static boolean BRIDGE_DOMAIN_FLOOD = true;
  67     private final static boolean BRIDGE_DOMAIN_FORWARD = true;
  68     private final static boolean BRIDGE_DOMAIN_LEARN = true;
  69     private final static boolean BRIDGE_DOMAIN_UNICAST_FLOOD = true;
  70     private final static boolean BRIDGE_DOMAIN_ARP = false;
  71     private final static NodeId VPP_NODE_ID = new NodeId("vppNode");
  72     private final static Topology BASE_TOPOLOGY = new TopologyBuilder().setTopologyId(BASE_TOPOLOGY_ID)
  73         .setNode(Collections.singletonList(new NodeBuilder().setNodeId(VPP_NODE_ID)
  74                 .setSupportingNode(Collections.singletonList(new SupportingNodeBuilder()
  75                         .setTopologyRef(SUPPORTING_TOPOLOGY_NETCONF)
  76                         .setNodeRef(VPP_NODE_ID)
  77                         .build()))
  78                 .build()))
  79         .setTopologyTypes(new TopologyTypesBuilder()
  80             .addAugmentation(TopologyTypesVbridgeAugment.class, new TopologyTypesVbridgeAugmentBuilder()
  81                 .setVbridgeTopology(new VbridgeTopologyBuilder().build()).build())
  82             .build())
  83         .addAugmentation(TopologyVbridgeAugment.class,
  84                 new TopologyVbridgeAugmentBuilder().setTunnelType(TunnelTypeVxlan.class)
  85                     .setTunnelParameters(new VxlanTunnelParametersBuilder().setVni(BRIDGE_DOMAIN_VNI).build())
  86                     .setForward(BRIDGE_DOMAIN_FORWARD)
  87                     .setUnknownUnicastFlood(BRIDGE_DOMAIN_UNICAST_FLOOD)
  88                     .setLearn(BRIDGE_DOMAIN_LEARN)
  89                     .setArpTermination(BRIDGE_DOMAIN_ARP)
  90                     .setFlood(BRIDGE_DOMAIN_FLOOD)
  91                     .build())
  92         .build();
  93
  94     private DataBroker dataBroker;
  95     private BridgeDomainManagerImpl bridgeDomainManager;
  96
  97     @Override
  98     @Nonnull
  99     public Collection<Class<?>> getClassesFromModules() {
 100         return Arrays.asList(NetworkTopology.class, Topology.class, TopologyVbridgeAugment.class,
 101                 TunnelTypeVxlan.class);
 102     }
 103
 104     @Before
 105     public void init() {
 106         dataBroker = getDataBroker();
 107         bridgeDomainManager = new BridgeDomainManagerImpl(dataBroker);
 108         final InterfaceManager interfaceManager = Mockito.mock(InterfaceManager.class);
 109         final AclManager aclManager = Mockito.mock(AclManager.class);
 110         final NatManager natManager = Mockito.mock(NatManager.class);
 111         final RoutingManager routingManager = Mockito.mock(RoutingManager.class);
 112         final LispStateManager lispStateManager = Mockito.mock(LispStateManager.class);
 113         final FlatOverlayManager flatOverlayManager = Mockito.mock(FlatOverlayManager.class);
 114         final LoopbackManager loopbackManager = Mockito.mock(LoopbackManager.class);
 115         final ForwardingManager fwManager =
 116             new ForwardingManager(interfaceManager, aclManager, natManager, routingManager, bridgeDomainManager,
 117                 lispStateManager, loopbackManager, flatOverlayManager, dataBroker);
 118         fwManager.setTimer((byte) 1);
 119     }
 120
 121     @Test
 122     public void testCreateVxlanBridgeDomainOnVppNode() throws Exception {
 123         bridgeDomainManager.createVxlanBridgeDomainOnVppNode(BRIDGE_DOMAIN_ID, BRIDGE_DOMAIN_VNI, VPP_NODE_ID);
 124         // simulates VBD - when BD is created a node is stored to OPER DS
 125         Thread vbdThread = new Thread(() -> {
 126             WriteTransaction wTx = dataBroker.newWriteOnlyTransaction();
 127             wTx.put(LogicalDatastoreType.OPERATIONAL, VppIidFactory
 128                 .getNodeIid(BASE_TOPOLOGY.getKey(), new NodeKey(VPP_NODE_ID)),
 129                     new NodeBuilder().setNodeId(VPP_NODE_ID)
 130                         .addAugmentation(NodeVbridgeAugment.class, new NodeVbridgeAugmentBuilder()
 131                             .setBridgeMember(new BridgeMemberBuilder().build()).build())
 132                         .build(),
 133                     true);
 134             try {
 135                 wTx.submit().get();
 136             } catch (InterruptedException | ExecutionException e) {
 137                 Assert.fail();
 138             }
 139         });
 140         vbdThread.join();
 141
 142         Optional<Topology> topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 143                 VppIidFactory.getTopologyIid(BASE_TOPOLOGY.getKey()), dataBroker.newReadOnlyTransaction());
 144         Assert.assertTrue(topologyOptional.isPresent());
 145
 146         Topology topology = topologyOptional.get();
 147         Assert.assertEquals(BASE_TOPOLOGY.getAugmentation(TopologyVbridgeAugment.class),
 148                 topology.getAugmentation(TopologyVbridgeAugment.class));
 149     }
 150
 151     @Test
 152     public void testRemoveBridgeDomainFromVppNode() {
 153         WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
 154
 155         writeTransaction.put(LogicalDatastoreType.CONFIGURATION, VppIidFactory.getTopologyIid(BASE_TOPOLOGY.getKey()),
 156                 BASE_TOPOLOGY, true);
 157         boolean result = DataStoreHelper.submitToDs(writeTransaction);
 158         Assert.assertTrue(result);
 159
 160         Optional<Node> topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 161                 VppIidFactory.getNodeIid(new TopologyKey(BASE_TOPOLOGY_ID), new NodeKey(VPP_NODE_ID)),
 162                 dataBroker.newReadOnlyTransaction());
 163         Assert.assertTrue(topologyOptional.isPresent());
 164
 165         bridgeDomainManager.removeBridgeDomainFromVppNode(BRIDGE_DOMAIN_ID, VPP_NODE_ID);
 166
 167         topologyOptional = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 168                 VppIidFactory.getNodeIid(new TopologyKey(BASE_TOPOLOGY_ID), new NodeKey(VPP_NODE_ID)),
 169                 dataBroker.newReadOnlyTransaction());
 170         Assert.assertFalse(topologyOptional.isPresent());
 171     }
 172
 173 }