Remove Augmentation{Identifier,Node}

[yangtools.git] / data / yang-data-api / src / test / java / org / opendaylight / yangtools / yang / data / api / schema / stream / NormalizedNodeWriterTest.java

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 e5c5b20070cef56a0367e58bc0fceb52c6c61b43..582c71e47ff9dc3305536927b647d613e0a90eac 100644 (file)
--- 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));
+        }
     }
 }