Remove DOMMountPoint's EffectiveModelContext
[mdsal.git] / binding / mdsal-binding-test-utils / src / test / java / org / opendaylight / mdsal / binding / testutils / AugmentableExtensionTest.java
index efebed4232d89e0ac8e00dbf0060b9778e0b4501..1678954d6699785a8b0dc7e39c1e1b8f53c0fcff 100644 (file)
@@ -54,10 +54,10 @@ public class AugmentableExtensionTest extends AbstractDataBrokerTest {
 
         ReadTransaction readTx = getDataBroker().newReadOnlyTransaction();
         InstanceIdentifier<Top> id = InstanceIdentifier.create(Top.class);
-        Top actualTop = readTx.read(OPERATIONAL, id).checkedGet().get();
+        Top actualTop = readTx.read(OPERATIONAL, id).get().get();
         AssertBeans.assertEqualByText("#{\n}", augmentableExtension.getAugmentations(actualTop));
 
-        TopLevelList topLevelList = actualTop.getTopLevelList().get(0);
+        TopLevelList topLevelList = actualTop.nonnullTopLevelList().values().iterator().next();
         AssertDataObjects.assertEqualByText("#{\n"
                 + "    TreeComplexUsesAugment -> new TreeComplexUsesAugmentBuilder >> [\n"
                 + "        containerWithUses = new ContainerWithUsesBuilder >> [\n"
@@ -67,8 +67,8 @@ public class AugmentableExtensionTest extends AbstractDataBrokerTest {
                 + "}", augmentableExtension.getAugmentations(topLevelList));
     }
 
-    <T extends DataObject> void put(WriteTransaction tx, LogicalDatastoreType store,
-            Map.Entry<InstanceIdentifier<T>, T> obj) {
+    <T extends DataObject> void put(final WriteTransaction tx, final LogicalDatastoreType store,
+            final Map.Entry<InstanceIdentifier<T>, T> obj) {
         tx.put(OPERATIONAL, obj.getKey(), obj.getValue());
     }