Migrate to simplified addAugmentation() method

[genius.git] / itm / itm-impl / src / test / java / org / opendaylight / genius / itm / tests / OvsdbTestUtil.java

diff --git a/itm/itm-impl/src/test/java/org/opendaylight/genius/itm/tests/OvsdbTestUtil.java b/itm/itm-impl/src/test/java/org/opendaylight/genius/itm/tests/OvsdbTestUtil.java
index 7c5a69290719c47d059b308029f67c97d7f53421..6005d0ab8d197b197011931f73d79c6b800de340 100644 (file)
--- a/itm/itm-impl/src/test/java/org/opendaylight/genius/itm/tests/OvsdbTestUtil.java
+++ b/itm/itm-impl/src/test/java/org/opendaylight/genius/itm/tests/OvsdbTestUtil.java
@@ -23,10 +23,8 @@ import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddressBuilder;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.DatapathId;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentationBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeName;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentationBuilder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ConnectionInfo;
@@ -115,7 +113,7 @@ public final class OvsdbTestUtil {
         ovsdbNodeAugBuilder.setOpenvswitchOtherConfigs(otherConfigsList);
 
         // add OvsdbNodeAugmentation into Node
-        nodeBuilder.addAugmentation(OvsdbNodeAugmentation.class, ovsdbNodeAugBuilder.build());
+        nodeBuilder.addAugmentation(ovsdbNodeAugBuilder.build());
         Node ovsdbNode = nodeBuilder.build();
 
         WriteTransaction transaction = dataBroker.newWriteOnlyTransaction();
@@ -185,7 +183,7 @@ public final class OvsdbTestUtil {
         ovsdbNodeAugBuilder.setOpenvswitchOtherConfigs(otherConfigsList);
 
         // add OvsdbNodeAugmentation into Node
-        nodeBuilder.addAugmentation(OvsdbNodeAugmentation.class, ovsdbNodeAugBuilder.build());
+        nodeBuilder.addAugmentation(ovsdbNodeAugBuilder.build());
         Node ovsdbNode = nodeBuilder.build();
 
         //ReadWriteTransaction transaction = dataBroker.newReadWriteTransaction();
@@ -212,7 +210,7 @@ public final class OvsdbTestUtil {
         ovsdbBridgeAugmentationBuilder.setManagedBy(new OvsdbNodeRef(
             SouthboundUtils.createInstanceIdentifier(nodeKey.getNodeId())));
 
-        bridgeNodeBuilder.addAugmentation(OvsdbBridgeAugmentation.class, ovsdbBridgeAugmentationBuilder.build());
+        bridgeNodeBuilder.addAugmentation(ovsdbBridgeAugmentationBuilder.build());
 
         Node bridgeNode = bridgeNodeBuilder.build();