Cleanup testutils implementation
[mdsal.git] / binding / mdsal-binding-test-utils / src / test / java / org / opendaylight / mdsal / binding / testutils / AugmentableExtensionTest.java
index cd3e3991042d61673f2c73ed46dd4ad3842332db..1678954d6699785a8b0dc7e39c1e1b8f53c0fcff 100644 (file)
@@ -57,7 +57,7 @@ public class AugmentableExtensionTest extends AbstractDataBrokerTest {
         Top actualTop = readTx.read(OPERATIONAL, id).get().get();
         AssertBeans.assertEqualByText("#{\n}", augmentableExtension.getAugmentations(actualTop));
 
-        TopLevelList topLevelList = actualTop.getTopLevelList().values().iterator().next();
+        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());
     }