X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=data%2Fyang-data-api%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Fyang%2Fdata%2Fapi%2Fschema%2Fstream%2FNormalizedNodeWriterTest.java;fp=data%2Fyang-data-api%2Fsrc%2Ftest%2Fjava%2Forg%2Fopendaylight%2Fyangtools%2Fyang%2Fdata%2Fapi%2Fschema%2Fstream%2FNormalizedNodeWriterTest.java;h=582c71e47ff9dc3305536927b647d613e0a90eac;hb=3d2579db1ac441dadf84e0ff7dab904832a87bf2;hp=e5c5b20070cef56a0367e58bc0fceb52c6c61b43;hpb=9a51cd8e2646e02c1c44c7de7c375e29dfe842f9;p=yangtools.git

diff --git a/data/yang-data-api/src/test/java/org/opendaylight/yangtools/yang/data/api/schema/stream/NormalizedNodeWriterTest.java b/data/yang-data-api/src/test/java/org/opendaylight/yangtools/yang/data/api/schema/stream/NormalizedNodeWriterTest.java
index e5c5b20070..582c71e47f 100644
--- a/data/yang-data-api/src/test/java/org/opendaylight/yangtools/yang/data/api/schema/stream/NormalizedNodeWriterTest.java
+++ b/data/yang-data-api/src/test/java/org/opendaylight/yangtools/yang/data/api/schema/stream/NormalizedNodeWriterTest.java
@@ -28,7 +28,6 @@ import org.opendaylight.yangtools.yang.common.XMLNamespace;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
-import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
 import org.opendaylight.yangtools.yang.data.api.schema.ChoiceNode;
 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
 import org.opendaylight.yangtools.yang.data.api.schema.DOMSourceAnyxmlNode;
@@ -97,8 +96,6 @@ public class NormalizedNodeWriterTest {
 
         assertNotNull(orderedNormalizedNodeWriter.write(mock(ChoiceNode.class)));
 
-        assertNotNull(orderedNormalizedNodeWriter.write(mock(AugmentationNode.class)));
-
         final UnkeyedListNode mockedUnkeyedListNode = mock(UnkeyedListNode.class);
         final Set<?> value = Set.of(mockedUnkeyedListEntryNode);
         doReturn(value).when(mockedUnkeyedListNode).body();
@@ -115,12 +112,8 @@ public class NormalizedNodeWriterTest {
         orderedNormalizedNodeWriter.flush();
         orderedNormalizedNodeWriter.close();
 
-        final NormalizedNodeWriter normalizedNodeWriter = NormalizedNodeWriter.forStreamWriter(
-                loggingNormalizedNodeStreamWriter, false);
-
-        assertNotNull(normalizedNodeWriter.write(mockedMapEntryNode));
-
-        normalizedNodeWriter.flush();
-        normalizedNodeWriter.close();
+        try (var nnWriter = NormalizedNodeWriter.forStreamWriter(loggingNormalizedNodeStreamWriter, false)) {
+            assertNotNull(nnWriter.write(mockedMapEntryNode));
+        }
     }
 }