X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=elanmanager%2Fimpl%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fnetvirt%2Felan%2Fl2gw%2Fnodehandlertest%2FNodeConnectedHandlerUtils.java;h=ff1133673b88cdd419bd8053230a3faa10a0ec85;hb=cf1ea9172b94fc17e45391b62bf4ca24ac5c5fe0;hp=440e16482c4cdd6b0d690535606d9450a72780b6;hpb=7e303c32a5395cc3ee718cbc324bd047f872dd57;p=netvirt.git

diff --git a/elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/NodeConnectedHandlerUtils.java b/elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/NodeConnectedHandlerUtils.java
index 440e16482c..ff1133673b 100644
--- a/elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/NodeConnectedHandlerUtils.java
+++ b/elanmanager/impl/src/test/java/org/opendaylight/netvirt/elan/l2gw/nodehandlertest/NodeConnectedHandlerUtils.java
@@ -7,15 +7,13 @@
  */
 package org.opendaylight.netvirt.elan.l2gw.nodehandlertest;
 
-import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.OPERATIONAL;
 
-import com.google.common.util.concurrent.CheckedFuture;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
 import java.util.UUID;
-import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
-import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
+import org.opendaylight.genius.infra.Datastore.Operational;
+import org.opendaylight.genius.infra.TypedWriteTransaction;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
@@ -37,14 +35,12 @@ import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  */
 public class NodeConnectedHandlerUtils {
 
-    CheckedFuture<Void, TransactionCommitFailedException> addNode(InstanceIdentifier<Node> path,
+    void addNode(InstanceIdentifier<Node> path,
             InstanceIdentifier<Node> psPath, String logicalSwitchData, String localUcasMacData, String localMcastData,
             String remoteMcastData, String remoteUcasteMacData, String globalTerminationPointIp,
-            WriteTransaction transaction) throws Exception {
-        NodeBuilder nodeBuilder = null;
-        HwvtepGlobalAugmentationBuilder augmentationBuilder = null;
-        nodeBuilder = prepareOperationalNode(path);
-        augmentationBuilder = prepareAugmentationBuilder();
+            TypedWriteTransaction<Operational> tx) {
+        NodeBuilder nodeBuilder = prepareOperationalNode(path);
+        HwvtepGlobalAugmentationBuilder augmentationBuilder = prepareAugmentationBuilder();
 
         GlobalAugmentationHelper.addLogicalSwitches(augmentationBuilder, getData(logicalSwitchData));
 
@@ -62,15 +58,11 @@ public class NodeConnectedHandlerUtils {
 
         nodeBuilder.addAugmentation(HwvtepGlobalAugmentation.class, augmentationBuilder.build());
 
-        return TestUtil.submitNode(OPERATIONAL, path, nodeBuilder.build(), transaction);
+        tx.put(path, nodeBuilder.build());
     }
 
-    CheckedFuture<Void, TransactionCommitFailedException> addPsNode(InstanceIdentifier<Node> path,
-            InstanceIdentifier<Node> parentPath, List<String> portNameList, WriteTransaction transaction)
-            throws Exception {
-        NodeBuilder nodeBuilder = null;
-
-        nodeBuilder = prepareOperationalNode(path);
+    void addPsNode(InstanceIdentifier<Node> path, InstanceIdentifier<Node> parentPath, List<String> portNameList,
+            TypedWriteTransaction<Operational> tx) {
         PhysicalSwitchAugmentationBuilder physicalSwitchAugmentationBuilder = new PhysicalSwitchAugmentationBuilder();
         physicalSwitchAugmentationBuilder.setManagedBy(new HwvtepGlobalRef(parentPath));
         physicalSwitchAugmentationBuilder.setPhysicalSwitchUuid(getUUid("d1s3"));
@@ -82,27 +74,28 @@ public class NodeConnectedHandlerUtils {
         tunnelIps.add(new TunnelIpsBuilder().withKey(new TunnelIpsKey(ip)).setTunnelIpsKey(ip).build());
         physicalSwitchAugmentationBuilder.setTunnelIps(tunnelIps);
 
+        NodeBuilder nodeBuilder = prepareOperationalNode(path);
         nodeBuilder.addAugmentation(PhysicalSwitchAugmentation.class, physicalSwitchAugmentationBuilder.build());
         PhysicalSwitchHelper.dId = parentPath;
         nodeBuilder.setTerminationPoint(PhysicalSwitchHelper
-                .addPhysicalSwitchTerminationPoints(path, transaction, portNameList));
+                .addPhysicalSwitchTerminationPoints(path, portNameList));
 
-        return TestUtil.submitNode(OPERATIONAL, path, nodeBuilder.build(), transaction);
+        tx.mergeParentStructurePut(path, nodeBuilder.build());
     }
 
-    NodeBuilder prepareOperationalNode(InstanceIdentifier<Node> iid) {
+    private static NodeBuilder prepareOperationalNode(InstanceIdentifier<Node> iid) {
         NodeBuilder nodeBuilder = new NodeBuilder();
         nodeBuilder.setNodeId(iid.firstKeyOf(Node.class).getNodeId());
         return nodeBuilder;
     }
 
-    HwvtepGlobalAugmentationBuilder prepareAugmentationBuilder() {
+    private static HwvtepGlobalAugmentationBuilder prepareAugmentationBuilder() {
         HwvtepGlobalAugmentationBuilder builder = new HwvtepGlobalAugmentationBuilder();
         builder.setManagers(TestBuilders.buildManagers());
         return builder;
     }
 
-    public List<String> getData(String data) {
+    private static List<String> getData(String data) {
         return Arrays.asList(data.split(","));
     }