elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/NodeConnectedHandlerUtils.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.netvirt.elan.l2gw.nodehandlertest;
   9
  10 import static org.opendaylight.mdsal.binding.api.WriteTransaction.CREATE_MISSING_PARENTS;
  11
  12 import java.util.ArrayList;
  13 import java.util.Arrays;
  14 import java.util.List;
  15 import java.util.UUID;
  16 import org.opendaylight.genius.infra.Datastore.Operational;
  17 import org.opendaylight.genius.infra.TypedWriteTransaction;
  18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentationBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalRef;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.PhysicalSwitchAugmentation;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.PhysicalSwitchAugmentationBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical._switch.attributes.TunnelIps;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical._switch.attributes.TunnelIpsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical._switch.attributes.TunnelIpsKey;
  30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  32 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  33
  34 /**
  35  * Created by eaksahu on 10/14/2016.
  36  */
  37 public class NodeConnectedHandlerUtils {
  38
  39     void addNode(InstanceIdentifier<Node> path,
  40             InstanceIdentifier<Node> psPath, String logicalSwitchData, String localUcasMacData, String localMcastData,
  41             String remoteMcastData, String remoteUcasteMacData, String globalTerminationPointIp,
  42             TypedWriteTransaction<Operational> tx) {
  43         NodeBuilder nodeBuilder = prepareOperationalNode(path);
  44         HwvtepGlobalAugmentationBuilder augmentationBuilder = prepareAugmentationBuilder();
  45
  46         GlobalAugmentationHelper.addLogicalSwitches(augmentationBuilder, getData(logicalSwitchData));
  47
  48         GlobalAugmentationHelper.addLocalUcastMacs(path, augmentationBuilder, getData(localUcasMacData));
  49
  50         GlobalAugmentationHelper.addLocalMcastMacs(path, augmentationBuilder, getData(localMcastData));
  51
  52         GlobalAugmentationHelper.addRemoteMcastMacs(path, augmentationBuilder, getData(remoteMcastData));
  53
  54         GlobalAugmentationHelper.addRemoteUcastMacs(path, augmentationBuilder, getData(remoteUcasteMacData));
  55
  56         GlobalAugmentationHelper.addGlobalTerminationPoints(nodeBuilder, path, getData(globalTerminationPointIp));
  57
  58         GlobalAugmentationHelper.addSwitches(augmentationBuilder, psPath);
  59
  60         nodeBuilder.addAugmentation(HwvtepGlobalAugmentation.class, augmentationBuilder.build());
  61
  62         tx.put(path, nodeBuilder.build());
  63     }
  64
  65     void addPsNode(InstanceIdentifier<Node> path, InstanceIdentifier<Node> parentPath, List<String> portNameList,
  66             TypedWriteTransaction<Operational> tx) {
  67         PhysicalSwitchAugmentationBuilder physicalSwitchAugmentationBuilder = new PhysicalSwitchAugmentationBuilder();
  68         physicalSwitchAugmentationBuilder.setManagedBy(new HwvtepGlobalRef(parentPath));
  69         physicalSwitchAugmentationBuilder.setPhysicalSwitchUuid(getUUid("d1s3"));
  70         physicalSwitchAugmentationBuilder.setHwvtepNodeName(new HwvtepNodeName("s3"));
  71         physicalSwitchAugmentationBuilder.setHwvtepNodeDescription("description");
  72
  73         List<TunnelIps> tunnelIps = new ArrayList<>();
  74         IpAddress ip = new IpAddress(new Ipv4Address("192.168.122.30"));
  75         tunnelIps.add(new TunnelIpsBuilder().withKey(new TunnelIpsKey(ip)).setTunnelIpsKey(ip).build());
  76         physicalSwitchAugmentationBuilder.setTunnelIps(tunnelIps);
  77
  78         NodeBuilder nodeBuilder = prepareOperationalNode(path);
  79         nodeBuilder.addAugmentation(PhysicalSwitchAugmentation.class, physicalSwitchAugmentationBuilder.build());
  80         PhysicalSwitchHelper.dId = parentPath;
  81         nodeBuilder.setTerminationPoint(PhysicalSwitchHelper
  82                 .addPhysicalSwitchTerminationPoints(path, portNameList));
  83
  84         tx.put(path, nodeBuilder.build(), CREATE_MISSING_PARENTS);
  85     }
  86
  87     private static NodeBuilder prepareOperationalNode(InstanceIdentifier<Node> iid) {
  88         NodeBuilder nodeBuilder = new NodeBuilder();
  89         nodeBuilder.setNodeId(iid.firstKeyOf(Node.class).getNodeId());
  90         return nodeBuilder;
  91     }
  92
  93     private static HwvtepGlobalAugmentationBuilder prepareAugmentationBuilder() {
  94         HwvtepGlobalAugmentationBuilder builder = new HwvtepGlobalAugmentationBuilder();
  95         builder.setManagers(TestBuilders.buildManagers());
  96         return builder;
  97     }
  98
  99     private static List<String> getData(String data) {
 100         return Arrays.asList(data.split(","));
 101     }
 102
 103     public static Uuid getUUid(String key) {
 104         return new Uuid(UUID.nameUUIDFromBytes(key.getBytes()).toString());
 105     }
 106 }