Use correct key-arg splitting

[yangtools.git] / yang / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug5884Test.java

diff --git a/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5884Test.java b/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5884Test.java
index f533b61f341d86ac5dbb280f9d06c4d1a2f293df..74d05dac60680bea179354dee8a0c934bf2c1660 100644 (file)
--- a/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5884Test.java
+++ b/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5884Test.java
@@ -7,12 +7,11 @@
  */
 package org.opendaylight.yangtools.yang.stmt;
 
+import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.isA;
 import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertThat;
 
 import java.util.Iterator;
-import java.util.Set;
 import org.junit.Test;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.common.Revision;
@@ -41,16 +40,12 @@ public class Bug5884Test {
         final ContainerSchemaNode testContainer = (ContainerSchemaNode) rootContainer.getDataChildByName(
             testContainerQname);
         final ChoiceSchemaNode dataChildByName = (ChoiceSchemaNode) testContainer.getDataChildByName(choice);
-        final Set<AugmentationSchemaNode> augmentations = foo.getAugmentations();
-        final Set<AugmentationSchemaNode> availableAugmentations = dataChildByName.getAvailableAugmentations();
-        final Iterator<AugmentationSchemaNode> iterator = augmentations.iterator();
-        final Iterator<AugmentationSchemaNode> availableIterator = availableAugmentations.iterator();
 
-        testIterator(iterator);
-        testIterator(availableIterator);
+        testIterator(foo.getAugmentations().iterator());
+        testIterator(dataChildByName.getAvailableAugmentations().iterator());
     }
 
-    private static void testIterator(final Iterator<AugmentationSchemaNode> iterator) {
+    private static void testIterator(final Iterator<? extends AugmentationSchemaNode> iterator) {
         while (iterator.hasNext()) {
             AugmentationSchemaNode allAugments = iterator.next();
             final DataSchemaNode currentChoice = allAugments.getChildNodes().iterator().next();