elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/TestBuilders.java

   1 /*
   2  * Copyright (c) 2016 Ericsson India Global Services Pvt Ltd. 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.netvirt.elan.l2gw.nodehandlertest;
   9
  10 import com.google.common.base.Preconditions;
  11 import com.google.common.collect.ImmutableBiMap;
  12 import com.google.common.collect.ImmutableList;
  13
  14 import java.util.ArrayList;
  15 import java.util.Collections;
  16 import java.util.List;
  17 import java.util.UUID;
  18
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Uri;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.EncapsulationTypeBase;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.EncapsulationTypeVxlanOverIpv4;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorAugmentation;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorAugmentationBuilder;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorRef;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalMcastMacs;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalMcastMacsBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacs;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacsBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesBuilder;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.Managers;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.ManagersBuilder;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.ManagersKey;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsBuilder;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacs;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacsBuilder;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.managers.ManagerOtherConfigs;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.managers.ManagerOtherConfigsBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.managers.ManagerOtherConfigsKey;
  48 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSet;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSetBuilder;
  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.Node;
  52 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  53 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
  54 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
  55 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  56
  57 /**
  58  * Created by ekvsver on 8/6/2016.
  59  */
  60 public final class TestBuilders {
  61     private TestBuilders() { }
  62
  63     public static HwvtepLogicalSwitchRef buildLogicalSwitchesRef(InstanceIdentifier<Node> nodeIid,
  64                                                                  String logicalSwitchName) {
  65         InstanceIdentifier<LogicalSwitches> logicalSwitchInstanceIid =
  66                 nodeIid.augmentation(HwvtepGlobalAugmentation.class)
  67                 .child(LogicalSwitches.class, new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitchName)));
  68         return new HwvtepLogicalSwitchRef(logicalSwitchInstanceIid);
  69     }
  70
  71     public static LocalUcastMacs buildLocalUcastMacs(InstanceIdentifier<Node> nodeIid, String vmMac,
  72                                                      String vmip, String tepIp, String logicalSwitchName) {
  73         LocalUcastMacsBuilder ucmlBuilder = new LocalUcastMacsBuilder();
  74         ucmlBuilder.setIpaddr(new IpAddress(vmip.toCharArray()));
  75         ucmlBuilder.setMacEntryKey(new MacAddress(vmMac));
  76         ucmlBuilder.setMacEntryUuid(getUUid(vmMac));
  77         ucmlBuilder.setLocatorRef(buildLocatorRef(nodeIid, tepIp));
  78         ucmlBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(nodeIid, logicalSwitchName));
  79         return ucmlBuilder.build();
  80     }
  81
  82     public static RemoteUcastMacs buildRemoteUcastMacs(InstanceIdentifier<Node> nodeIid, String vmMac,
  83                                                        String vmip, String tepIp, String logicalSwitchName) {
  84         RemoteUcastMacsBuilder ucmlBuilder = new RemoteUcastMacsBuilder();
  85         ucmlBuilder.setIpaddr(new IpAddress(vmip.toCharArray()));
  86         ucmlBuilder.setMacEntryKey(new MacAddress(vmMac));
  87         ucmlBuilder.setMacEntryUuid(getUUid(vmMac));
  88         ucmlBuilder.setLocatorRef(buildLocatorRef(nodeIid, tepIp));
  89         ucmlBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(nodeIid, logicalSwitchName));
  90         return ucmlBuilder.build();
  91     }
  92
  93     public static TerminationPoint buildTerminationPoint(InstanceIdentifier<Node> nodeIid, String ip) {
  94         TerminationPointKey tpKey = new TerminationPointKey(new TpId("vxlan_over_ipv4:" + ip));
  95         TerminationPointBuilder tpBuilder = new TerminationPointBuilder();
  96         if (nodeIid != null && tpKey != null) {
  97             tpBuilder.setKey(tpKey);
  98             tpBuilder.setTpId(tpKey.getTpId());
  99             InstanceIdentifier<TerminationPoint> tpPath =
 100                     buildTpId(nodeIid, ip);
 101             HwvtepPhysicalLocatorAugmentationBuilder tpAugmentationBuilder =
 102                     new HwvtepPhysicalLocatorAugmentationBuilder();
 103             tpAugmentationBuilder.setPhysicalLocatorUuid(getUUid(ip));
 104             tpAugmentationBuilder.setEncapsulationType(createEncapsulationType("vxlan_over_ipv4"));
 105             tpAugmentationBuilder.setDstIp(new IpAddress(ip.toCharArray()));
 106             tpBuilder.addAugmentation(HwvtepPhysicalLocatorAugmentation.class, tpAugmentationBuilder.build());
 107         }
 108         return tpBuilder.build();
 109     }
 110
 111     public static LogicalSwitches buildLogicalSwitch(String logicalSwitch, String tunnelKey) {
 112         LogicalSwitchesBuilder logicalSwitchesBuilder = new LogicalSwitchesBuilder();
 113         logicalSwitchesBuilder.setKey(new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitch)));
 114         logicalSwitchesBuilder.setHwvtepNodeName(new HwvtepNodeName(logicalSwitch));
 115         logicalSwitchesBuilder.setTunnelKey(tunnelKey);
 116         Uuid lgoicalSwitchUuid = getUUid(logicalSwitch);
 117         logicalSwitchesBuilder.setLogicalSwitchUuid(lgoicalSwitchUuid);
 118         return logicalSwitchesBuilder.build();
 119     }
 120
 121     public static LocalMcastMacs buildLocalMcastMacs(InstanceIdentifier<Node> iid, String mac,
 122                                                      String logicalSwitchName, String tepIp) {
 123         LocalMcastMacsBuilder localMcastMacsBuilder = new LocalMcastMacsBuilder();
 124         if (mac.equals("unknown-dst")) {
 125             localMcastMacsBuilder.setMacEntryKey(new MacAddress("00:00:00:00:00:00"));
 126         } else {
 127             localMcastMacsBuilder.setMacEntryKey(new MacAddress(mac));
 128         }
 129         localMcastMacsBuilder.setMacEntryUuid(getUUid(mac));
 130         //mMacLocalBuilder.setIpaddr(new IpAddress(ip.toCharArray()));
 131         localMcastMacsBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(iid, logicalSwitchName));
 132         List<LocatorSet> locatorSets = new ArrayList<>();
 133         locatorSets.add(new LocatorSetBuilder().setLocatorRef(
 134                 buildLocatorRef(iid, tepIp)).build());
 135         localMcastMacsBuilder.setLocatorSet(locatorSets);
 136
 137         return localMcastMacsBuilder.build();
 138     }
 139
 140     public static RemoteMcastMacs buildRemoteMcastMacs(InstanceIdentifier<Node> iid, String mac,
 141                                                        String logicalSwitchName, String [] tepIps) {
 142
 143         RemoteMcastMacsBuilder remoteMcastMacsBuilder = new RemoteMcastMacsBuilder();
 144         if (mac.equals("unknown-dst")) {
 145             remoteMcastMacsBuilder.setMacEntryKey(new MacAddress("00:00:00:00:00:00"));
 146         } else {
 147             remoteMcastMacsBuilder.setMacEntryKey(new MacAddress(mac));
 148         }
 149         remoteMcastMacsBuilder.setMacEntryUuid(getUUid(mac));
 150         //mMacLocalBuilder.setIpaddr(new IpAddress(ip.toCharArray()));
 151         remoteMcastMacsBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(iid, logicalSwitchName));
 152         List<LocatorSet> locatorSets = new ArrayList<>();
 153         for (String tepIp : tepIps) {
 154             locatorSets.add(new LocatorSetBuilder().setLocatorRef(
 155                     buildLocatorRef(iid, tepIp)).build());
 156         }
 157         remoteMcastMacsBuilder.setLocatorSet(locatorSets);
 158         return remoteMcastMacsBuilder.build();
 159     }
 160
 161     public static List<Managers> buildManagers() {
 162         ManagersBuilder builder1 = new ManagersBuilder();
 163
 164         builder1.setKey(new ManagersKey(new Uri("test")));
 165         List<ManagerOtherConfigs> otherConfigses = new ArrayList<>();
 166
 167         otherConfigses.add(buildOtherConfig("ha_enabled", "true"));
 168         otherConfigses.add(buildOtherConfig("ha_id", "switchxyz"));
 169         builder1.setManagerOtherConfigs(otherConfigses);
 170         List<Managers> managers = new ArrayList<>();
 171         managers.add(builder1.build());
 172         return managers;
 173     }
 174
 175     public static List<Managers> buildManagers1() {
 176         ManagersBuilder builder1 = new ManagersBuilder();
 177         builder1.setKey(new ManagersKey(new Uri("test")));
 178         builder1.setManagerOtherConfigs(Collections.emptyList());
 179         return ImmutableList.of(builder1.build());
 180     }
 181
 182     public static ManagerOtherConfigs buildOtherConfig(String key, String val) {
 183         ManagerOtherConfigsBuilder otherConfigsBuilder = new ManagerOtherConfigsBuilder();
 184         ManagerOtherConfigsKey managerOtherConfigsKey = new ManagerOtherConfigsKey(key);
 185         otherConfigsBuilder.setKey(managerOtherConfigsKey);
 186         otherConfigsBuilder.setOtherConfigKey(key);
 187         otherConfigsBuilder.setOtherConfigValue(val);
 188         return otherConfigsBuilder.build();
 189     }
 190
 191     public static Uuid getUUid(String key) {
 192         return new Uuid(UUID.nameUUIDFromBytes(key.getBytes()).toString());
 193     }
 194
 195     public static HwvtepPhysicalLocatorRef buildLocatorRef(InstanceIdentifier<Node> nodeIid, String tepIp) {
 196         InstanceIdentifier<TerminationPoint> tepId = buildTpId(nodeIid, tepIp);
 197         return new HwvtepPhysicalLocatorRef(tepId);
 198     }
 199
 200     public static InstanceIdentifier<TerminationPoint> buildTpId(InstanceIdentifier<Node> nodeIid, String tepIp) {
 201         String tpKeyStr = "vxlan_over_ipv4" + ':' + tepIp;
 202         TerminationPointKey tpKey = new TerminationPointKey(new TpId(tpKeyStr));
 203         InstanceIdentifier<TerminationPoint> plIid = nodeIid.child(TerminationPoint.class, tpKey);
 204         return plIid;
 205     }
 206
 207     public static Class<? extends EncapsulationTypeBase> createEncapsulationType(String type) {
 208         Preconditions.checkNotNull(type);
 209         if (type.isEmpty()) {
 210             return EncapsulationTypeVxlanOverIpv4.class;
 211         } else {
 212             ImmutableBiMap<Class<? extends EncapsulationTypeBase>, String> encapsTypeMap
 213                     = new ImmutableBiMap.Builder<Class<? extends EncapsulationTypeBase>, String>()
 214                     .put(EncapsulationTypeVxlanOverIpv4.class, "vxlan_over_ipv4")
 215                     .build();
 216             ImmutableBiMap<String, Class<? extends EncapsulationTypeBase>> mapper =
 217                     encapsTypeMap.inverse();
 218             return mapper.get(type);
 219         }
 220     }
 221 }