elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/TestUtil.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 static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertFalse;
  12 import static org.junit.Assert.assertTrue;
  13 import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.OPERATIONAL;
  14
  15 import com.google.common.base.Optional;
  16 import com.google.common.util.concurrent.CheckedFuture;
  17 import java.util.Arrays;
  18 import java.util.List;
  19 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  21 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  22 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
  23 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  24 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  25 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  27 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33 /**
  34  * Created by eaksahu on 8/12/2016.
  35  */
  36 public final class TestUtil {
  37     static Logger LOG = LoggerFactory.getLogger(TestUtil.class);
  38
  39     private TestUtil() { }
  40
  41     public static void deleteNode(ReadWriteTransaction tx, InstanceIdentifier<Node> id) throws Exception {
  42         tx.delete(OPERATIONAL, id);
  43         tx.submit().checkedGet();
  44     }
  45
  46     public static void verifyHAOpNode(Node d1GlobalOpNode, Node haGlobalOpNode, Node d1PsOpNode, Node haPsOpNode,
  47                                       InstanceIdentifier<Node> haId, InstanceIdentifier<Node> d1PsId,
  48                                       InstanceIdentifier<Node> haPsId, NodeId haNodeId, DataBroker dataBroker)
  49             throws ReadFailedException {
  50         ReadWriteTransaction transaction = dataBroker.newReadWriteTransaction();
  51         TestComparators.compareLogicalSwitches(d1GlobalOpNode, haGlobalOpNode, haId);
  52         TestComparators.compareRemoteUcastMacs(d1GlobalOpNode, haGlobalOpNode, haId);
  53         TestComparators.compareRemoteMcastMacs(d1GlobalOpNode, haGlobalOpNode, haId);
  54         TestComparators.compareLocalUcastMacs(d1GlobalOpNode, haGlobalOpNode, haId);
  55         TestComparators.compareLocalMcastMacs(d1GlobalOpNode, haGlobalOpNode, haId);
  56         TestComparators.verifySwitches(haGlobalOpNode, haPsOpNode);
  57         TestComparators.verifySwitches(d1GlobalOpNode, d1PsOpNode);
  58         TestComparators.comparePhysicalSwitches(d1PsOpNode, haPsOpNode, d1PsId, haPsId, transaction, "s3",
  59                 d1GlobalOpNode, haGlobalOpNode);
  60     }
  61
  62     public static void verifyHAOpNode(Node d1GlobalOpNode, Node d2GlobalOpNode, Node haGlobalOpNode,
  63                                       Node d1PsOpNode, Node d2PsOpNode, Node haPsOpNode,
  64                                       InstanceIdentifier<Node> haId, InstanceIdentifier<Node> d1PsId,
  65                                       InstanceIdentifier<Node> d2PsId, InstanceIdentifier<Node> haPsId,
  66                                       NodeId haNodeId, DataBroker dataBroker) throws ReadFailedException {
  67         ReadWriteTransaction transaction = dataBroker.newReadWriteTransaction();
  68         TestComparators.compareLogicalSwitches(d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode, haId);
  69         TestComparators.compareRemoteUcastMacs(d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode, haId);
  70         TestComparators.compareRemoteMcastMacs(d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode, haId);
  71         TestComparators.compareLocalUcastMacs(d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode, haId);
  72         TestComparators.compareLocalMcastMacs(d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode, haId);
  73         TestComparators.verifySwitches(haGlobalOpNode, haPsOpNode);
  74         TestComparators.verifySwitches(d1GlobalOpNode, d1PsOpNode);
  75         TestComparators.verifySwitches(d2GlobalOpNode, d2PsOpNode);
  76         TestComparators.comparePhysicalSwitches(d1PsOpNode, d2PsOpNode, haPsOpNode, d1PsId, d2PsId, haPsId,
  77                 transaction, "s3", d1GlobalOpNode, d2GlobalOpNode, haGlobalOpNode);
  78     }
  79
  80     public static Node readNode(LogicalDatastoreType datastoreType, InstanceIdentifier<Node> id, DataBroker dataBroker)
  81             throws Exception {
  82         if (dataBroker.newReadWriteTransaction().read(datastoreType, id).checkedGet().isPresent()) {
  83             return dataBroker.newReadWriteTransaction().read(datastoreType, id).checkedGet().get();
  84         }
  85         return null;
  86     }
  87
  88     public static Optional<Node> readNode(LogicalDatastoreType datastoreType, InstanceIdentifier<Node> id,
  89                                           ReadOnlyTransaction tx) throws Exception {
  90         return tx.read(datastoreType, id).checkedGet();
  91     }
  92
  93     static void updateNode(LogicalDatastoreType datastoreType, InstanceIdentifier<Node> id, Node node,
  94                            DataBroker dataBroker) throws Exception {
  95         WriteTransaction transaction = dataBroker.newWriteOnlyTransaction();
  96         transaction.merge(datastoreType, id, node, WriteTransaction.CREATE_MISSING_PARENTS);
  97         transaction.submit();
  98     }
  99
 100
 101     static CheckedFuture<Void, TransactionCommitFailedException> submitNode(LogicalDatastoreType datastoreType,
 102             InstanceIdentifier<Node> id, Node node, WriteTransaction transaction) throws Exception {
 103         transaction.put(datastoreType, id, node, WriteTransaction.CREATE_MISSING_PARENTS);
 104         return transaction.submit();
 105     }
 106
 107     static Optional<Node> readNodeOptional(LogicalDatastoreType datastoreType, InstanceIdentifier<Node> id,
 108                                            DataBroker dataBroker) throws Exception {
 109         return dataBroker.newReadWriteTransaction().read(datastoreType, id).checkedGet();
 110     }
 111
 112     public static void verifyHaAfterDelete(InstanceIdentifier<Node> nodeId, LogicalDatastoreType datastoreType,
 113                                            String message, DataBroker dataBroker) throws Exception {
 114         Optional<Node> nodeOptional = TestUtil.readNodeOptional(datastoreType, nodeId, dataBroker);
 115         if (!message.contains("D2")) {
 116             assertTrue(message, nodeOptional.isPresent());
 117         } else {
 118             assertFalse(message, nodeOptional.isPresent());
 119         }
 120         if (nodeOptional.isPresent()) {
 121             LOG.info("Node data{}", nodeOptional.get());
 122         }
 123     }
 124
 125     public static void verifyHAconfigNode(InstanceIdentifier<Node> nodeId, DataBroker dataBroker, String message)
 126             throws Exception {
 127         Optional<Node> nodeOptional = TestUtil.readNodeOptional(LogicalDatastoreType.CONFIGURATION, nodeId, dataBroker);
 128         assertTrue(message, nodeOptional.isPresent());
 129         if (nodeOptional.isPresent()) {
 130             nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class).getManagers();
 131         }
 132     }
 133
 134     public static void verifyHAconfigNode(Node haConfig, Node d1Node) throws Exception {
 135         String haid = haConfig.getAugmentation(HwvtepGlobalAugmentation.class).getManagers()
 136                 .get(0).getManagerOtherConfigs().get(0).getOtherConfigValue();
 137         String d1id = d1Node.getNodeId().getValue();
 138         assertEquals("Other config should contain D1 as child manager", haid, d1id);
 139     }
 140
 141     public static void verifyHAconfigNode(Node haConfig, Node d1Node, Node d2Node) throws Exception {
 142         String haid = haConfig.getAugmentation(HwvtepGlobalAugmentation.class).getManagers()
 143                 .get(0).getManagerOtherConfigs().get(0).getOtherConfigValue();
 144         String[] haids = haid.split(",");
 145         List<String> haidSlist = Arrays.asList(haids);
 146         assertEquals("Ha Other config size should be 2 after creationg of D2 Node", 2, haids.length);
 147         String d1id = d1Node.getNodeId().getValue();
 148         String d2id = d2Node.getNodeId().getValue();
 149         assertTrue("ha should contain d1/d2 id as other config", haidSlist.contains(d1id));
 150         assertTrue("ha should contain d1/d2 id as other config", haidSlist.contains(d2id));
 151     }
 152
 153 }