hwvtepsouthbound/hwvtepsouthbound-impl/src/test/java/org/opendaylight/ovsdb/hwvtepsouthbound/TestBuilders.java

   1 /*
   2  * Copyright © 2016, 2017 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.ovsdb.hwvtepsouthbound;
   9
  10 import java.util.ArrayList;
  11 import java.util.Arrays;
  12 import java.util.List;
  13 import java.util.Map;
  14 import java.util.UUID;
  15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressBuilder;
  16 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
  17 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorAugmentationBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorRef;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsKey;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacs;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacsBuilder;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacsKey;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSet;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSetBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TpId;
  35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointBuilder;
  38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
  39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  40 import org.opendaylight.yangtools.yang.binding.util.BindingMap;
  41
  42 public final class TestBuilders {
  43
  44     public static final String VXLAN_OVER_IPV4 = "vxlan_over_ipv4";
  45
  46     private TestBuilders() {
  47     }
  48
  49     public static Map<LogicalSwitchesKey, LogicalSwitches> logicalSwitches(String[]... data) {
  50         return Arrays.stream(data).map(TestBuilders::buildLogicalSwitch).collect(BindingMap.toOrderedMap());
  51     }
  52
  53     public static Map<RemoteMcastMacsKey, RemoteMcastMacs> remoteMcastMacs(InstanceIdentifier<Node> iid,
  54                                                                            String[]... data) {
  55         return Arrays.stream(data)
  56             .map(row -> TestBuilders.buildRemoteMcastMacs(iid, row[0], row[1], Arrays.copyOfRange(row, 2, row.length)))
  57             .collect(BindingMap.toOrderedMap());
  58     }
  59
  60     public static Map<RemoteUcastMacsKey, RemoteUcastMacs> remoteUcastMacs(InstanceIdentifier<Node> iid,
  61                                                                            String[]... data) {
  62         return Arrays.stream(data)
  63             .map(row -> TestBuilders.buildRemoteUcastMacs(iid, row[0], row[1], row[2], row[3]))
  64             .collect(BindingMap.toOrderedMap());
  65     }
  66
  67     public static Map<TerminationPointKey, TerminationPoint> globalTerminationPoints(InstanceIdentifier<Node> nodeIid,
  68                                                                                      String[]... data) {
  69         return Arrays.stream(data)
  70             .map(row -> TestBuilders.buildTerminationPoint(nodeIid, row[0]))
  71             .collect(BindingMap.toOrderedMap());
  72     }
  73
  74     public static HwvtepLogicalSwitchRef buildLogicalSwitchesRef(InstanceIdentifier<Node> nodeIid,
  75                                                                  String logicalSwitchName) {
  76         InstanceIdentifier<LogicalSwitches> switchIid = nodeIid.augmentation(HwvtepGlobalAugmentation.class)
  77                 .child(LogicalSwitches.class, new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitchName)));
  78         return new HwvtepLogicalSwitchRef(switchIid);
  79     }
  80
  81     public static RemoteUcastMacs buildRemoteUcastMacs(InstanceIdentifier<Node> nodeIid, String vmMac,
  82                                                        String vmip, String tepIp, String logicalSwitchName) {
  83         RemoteUcastMacsBuilder ucmlBuilder = new RemoteUcastMacsBuilder();
  84         ucmlBuilder.setIpaddr(IpAddressBuilder.getDefaultInstance(vmip));
  85         ucmlBuilder.setMacEntryKey(new MacAddress(vmMac));
  86         ucmlBuilder.setMacEntryUuid(getUUid(vmMac));
  87         ucmlBuilder.setLocatorRef(buildLocatorRef(nodeIid, tepIp));
  88         ucmlBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(nodeIid, logicalSwitchName));
  89         return ucmlBuilder.build();
  90     }
  91
  92     public static TerminationPoint buildTerminationPoint(InstanceIdentifier<Node> nodeIid, String ip) {
  93         TerminationPointKey tpKey = new TerminationPointKey(new TpId("vxlan_over_ipv4:" + ip));
  94         TerminationPointBuilder tpBuilder = new TerminationPointBuilder();
  95         if (nodeIid != null) {
  96             tpBuilder.withKey(tpKey);
  97             tpBuilder.setTpId(tpKey.getTpId());
  98             tpBuilder.addAugmentation(new HwvtepPhysicalLocatorAugmentationBuilder()
  99                 .setPhysicalLocatorUuid(getUUid(ip))
 100                 .setEncapsulationType(HwvtepSouthboundMapper.createEncapsulationType(VXLAN_OVER_IPV4))
 101                 .setDstIp(IpAddressBuilder.getDefaultInstance(ip))
 102                 .build());
 103         }
 104         return tpBuilder.build();
 105     }
 106
 107     public static LogicalSwitches buildLogicalSwitch(String... keys) {
 108         String logicalSwitch = keys[0];
 109         String tunnelKey = keys[1];
 110         LogicalSwitchesBuilder logicalSwitchesBuilder = new LogicalSwitchesBuilder();
 111         logicalSwitchesBuilder.withKey(new LogicalSwitchesKey(new HwvtepNodeName(logicalSwitch)));
 112         logicalSwitchesBuilder.setHwvtepNodeName(new HwvtepNodeName(logicalSwitch));
 113         logicalSwitchesBuilder.setTunnelKey(tunnelKey);
 114         Uuid uuid = getUUid(logicalSwitch);
 115         logicalSwitchesBuilder.setLogicalSwitchUuid(uuid);
 116         return logicalSwitchesBuilder.build();
 117     }
 118
 119     public static RemoteMcastMacs buildRemoteMcastMacs(InstanceIdentifier<Node> iid, String mac,
 120                                                        String logicalSwitchName, String[] tepIps) {
 121
 122         RemoteMcastMacsBuilder macLocalBuilder = new RemoteMcastMacsBuilder();
 123         if (mac.equals(HwvtepSouthboundConstants.UNKNOWN_DST_STRING)) {
 124             macLocalBuilder.setMacEntryKey(HwvtepSouthboundConstants.UNKNOWN_DST_MAC);
 125         } else {
 126             macLocalBuilder.setMacEntryKey(new MacAddress(mac));
 127         }
 128         macLocalBuilder.setMacEntryUuid(getUUid(mac));
 129         macLocalBuilder.setLogicalSwitchRef(buildLogicalSwitchesRef(iid, logicalSwitchName));
 130         List<LocatorSet> locatorSets = new ArrayList<>();
 131         for (String tepIp : tepIps) {
 132             locatorSets.add(new LocatorSetBuilder().setLocatorRef(
 133                     buildLocatorRef(iid, tepIp)).build());
 134         }
 135         macLocalBuilder.setLocatorSet(locatorSets);
 136         return macLocalBuilder.build();
 137     }
 138
 139     public static HwvtepPhysicalLocatorRef buildLocatorRef(InstanceIdentifier<Node> nodeIid,String tepIp) {
 140         InstanceIdentifier<TerminationPoint> tepId = buildTpId(nodeIid, tepIp);
 141         return new HwvtepPhysicalLocatorRef(tepId);
 142     }
 143
 144     public static Uuid getUUid(String key) {
 145         return new Uuid(UUID.nameUUIDFromBytes(key.getBytes()).toString());
 146     }
 147
 148     public static InstanceIdentifier<TerminationPoint> buildTpId(InstanceIdentifier<Node> nodeIid,String tepIp) {
 149         String tpKeyStr = VXLAN_OVER_IPV4 + ':' + tepIp;
 150         TerminationPointKey tpKey = new TerminationPointKey(new TpId(tpKeyStr));
 151         return nodeIid.child(TerminationPoint.class, tpKey);
 152     }
 153 }