neutron-vpp-mapper/src/test/java/org/opendaylight/groupbasedpolicy/neutron/vpp/mapper/processors/NetworkAwareTest.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.neutron.vpp.mapper.processors;
  10
  11 import static org.junit.Assert.assertEquals;
  12 import static org.junit.Assert.assertNull;
  13 import static org.junit.Assert.assertFalse;
  14 import static org.junit.Assert.assertTrue;
  15
  16 import java.util.List;
  17 import java.util.UUID;
  18
  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.ReadOnlyTransaction;
  24 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  25 import org.opendaylight.controller.md.sal.binding.test.AbstractDataBrokerTest;
  26 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  27 import org.opendaylight.groupbasedpolicy.util.DataStoreHelper;
  28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.vpp_renderer.rev160425.FlatNetwork;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.vpp_renderer.rev160425.VlanNetwork;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.vpp_renderer.rev160425.config.GbpBridgeDomain;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.vpp_renderer.rev160425.bridge.domain.base.attributes.PhysicalLocationRef;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.l2.types.rev130827.VlanId;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeFlat;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.NetworkTypeVlan;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.Network;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.networks.rev150712.networks.attributes.networks.NetworkBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtension;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.provider.ext.rev150712.NetworkProviderExtensionBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
  42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  43 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
  44 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  46 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
  49
  50 import com.google.common.base.Optional;
  51 import com.google.common.collect.ImmutableList;
  52
  53 public class NetworkAwareTest extends AbstractDataBrokerTest {
  54
  55     private DataBroker dataBroker;
  56     private NetworkAware networkAware;
  57     private NetworkProviderExtensionBuilder netExtBuilder;
  58
  59     @Before
  60     public void init() {
  61         dataBroker = Mockito.spy(getDataBroker());
  62         networkAware = new NetworkAware(dataBroker);
  63         netExtBuilder = new NetworkProviderExtensionBuilder();
  64         netExtBuilder.setPhysicalNetwork("physicalNet");
  65     }
  66
  67     @Test
  68     public void testProcessCreatedNeutronDto_flat() {
  69         netExtBuilder.setNetworkType(NetworkTypeFlat.class);
  70         Network network = createTestNetwork("net", netExtBuilder.build());
  71         networkAware.processCreatedNeutronDto(network);
  72         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
  73         Optional<GbpBridgeDomain> optBrDomain = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
  74                 networkAware.getGbpBridgeDomainIid(network.getUuid().getValue()), rTx);
  75         assertTrue(optBrDomain.isPresent());
  76     }
  77
  78     @Test
  79     public void testProcessUpdatedNeutronDto() {
  80         netExtBuilder.setNetworkType(NetworkTypeFlat.class);
  81         Network network1 = createTestNetwork("net1", netExtBuilder.build());
  82         Network network2 = new NetworkBuilder(network1).setName("net2")
  83             .addAugmentation(NetworkProviderExtension.class, netExtBuilder.build())
  84             .build();
  85         networkAware.processUpdatedNeutronDto(network1, network2);
  86         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
  87         Optional<GbpBridgeDomain> optBrDomain = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
  88                 networkAware.getGbpBridgeDomainIid(network2.getUuid().getValue()), rTx);
  89         assertTrue(optBrDomain.isPresent());
  90         assertEquals(optBrDomain.get().getDescription(), "net2");
  91     }
  92
  93     @Test
  94     public void testProcessDeletedNeutronDto() {
  95         netExtBuilder.setNetworkType(NetworkTypeFlat.class);
  96         Network network = createTestNetwork("net1", netExtBuilder.build());
  97         networkAware.processDeletedNeutronDto(network);
  98         ReadOnlyTransaction rTx = dataBroker.newReadOnlyTransaction();
  99         Optional<GbpBridgeDomain> optBrDomain = DataStoreHelper.readFromDs(LogicalDatastoreType.CONFIGURATION,
 100                 networkAware.getGbpBridgeDomainIid(network.getUuid().getValue()), rTx);
 101         assertFalse(optBrDomain.isPresent());
 102     }
 103
 104     @Test
 105     public void testCreateBridgeDomain_vlanNetwork() {
 106         netExtBuilder.setNetworkType(NetworkTypeVlan.class);
 107         netExtBuilder.setSegmentationId("2016");
 108         Network vlanNetwork = createTestNetwork("VlanNet", netExtBuilder.build());
 109         GbpBridgeDomain bridgeDomain = networkAware.createGbpBridgeDomain(vlanNetwork);
 110         assertEquals(bridgeDomain.getId(), vlanNetwork.getUuid().getValue());
 111         assertEquals(bridgeDomain.getDescription(), vlanNetwork.getName());
 112         assertEquals(bridgeDomain.getType(), VlanNetwork.class);
 113         assertEquals(bridgeDomain.getVlan(), new VlanId(Integer.valueOf(2016)));
 114     }
 115
 116     @Test
 117     public void testCreateBridgeDomain_flatNetwork() {
 118         netExtBuilder.setNetworkType(NetworkTypeFlat.class);
 119         Network flatNetwork = createTestNetwork("FlatNet", netExtBuilder.build());
 120         GbpBridgeDomain bridgeDomain = networkAware.createGbpBridgeDomain(flatNetwork);
 121         assertEquals(bridgeDomain.getId(), flatNetwork.getUuid().getValue());
 122         assertEquals(bridgeDomain.getDescription(), flatNetwork.getName());
 123         assertEquals(bridgeDomain.getType(), FlatNetwork.class);
 124     }
 125
 126     @Test
 127     public void testCreateBridgeDomain_noPhysicalNetwork() {
 128         netExtBuilder.setNetworkType(NetworkTypeFlat.class);
 129         netExtBuilder.setPhysicalNetwork(null);
 130         Network flatNetwork = createTestNetwork("FlatNet", netExtBuilder.build());
 131         GbpBridgeDomain bridgeDomain = networkAware.createGbpBridgeDomain(flatNetwork);
 132         assertNull(bridgeDomain.getPhysicalLocationRef());
 133     }
 134
 135     @Test
 136     public void testCreateBridgeDomain_noNetworkType() {
 137         Network vlanNetwork = createTestNetwork("noTypeNet", new NetworkProviderExtensionBuilder().build());
 138         GbpBridgeDomain bridgeDomain = networkAware.createGbpBridgeDomain(vlanNetwork);
 139         assertNull(bridgeDomain);
 140     }
 141
 142     @Test
 143     public void testResolveDomainLocations() {
 144         NodeId nodeId = new NodeId("node1");
 145         TpId tpId = new TpId("tp1");
 146         TopologyId topologyId = new TopologyId("physicalNet");
 147         writeBasicTopology(topologyId, nodeId, tpId);
 148         NetworkProviderExtension netExt = new NetworkProviderExtensionBuilder().setPhysicalNetwork("physicalNet")
 149             .build();
 150         List<PhysicalLocationRef> resolvedLocations = networkAware.resolveDomainLocations(netExt);
 151         PhysicalLocationRef physLocationRef = resolvedLocations.get(0);
 152         assertEquals(nodeId, physLocationRef.getNodeId());
 153         assertEquals(tpId.getValue(), physLocationRef.getInterface().get(0));
 154     }
 155
 156     private Network createTestNetwork(String name, NetworkProviderExtension ext) {
 157         return new NetworkBuilder().setUuid(new Uuid(UUID.randomUUID().toString()))
 158             .setName(name)
 159             .addAugmentation(NetworkProviderExtension.class, ext)
 160             .build();
 161     }
 162
 163     private void writeBasicTopology(TopologyId topologyId, NodeId nodeId, TpId tpId) {
 164         TerminationPoint tp = new TerminationPointBuilder().setTpId(tpId).build();
 165         Node node = new NodeBuilder().setNodeId(new NodeId(nodeId))
 166             .setTerminationPoint(ImmutableList.<TerminationPoint>of(tp))
 167             .build();
 168         Topology topology = new TopologyBuilder().setTopologyId(topologyId)
 169             .setNode(ImmutableList.<Node>of(node))
 170             .build();
 171         WriteTransaction wTx = dataBroker.newWriteOnlyTransaction();
 172         wTx.put(LogicalDatastoreType.CONFIGURATION, networkAware.getTopologyIid(new TopologyId("physicalNet")),
 173                 topology, true);
 174         DataStoreHelper.submitToDs(wTx);
 175     }
 176 }