Use SchemaNodeIdentifier in AugmentationSchemaNode

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

diff --git a/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5942Test.java b/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5942Test.java
index 6e219a8c3ee4c572a54805d991d7ce0b39ecb46b..93125438deb1680a53bb12268024922344f4883b 100644 (file)
--- a/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5942Test.java
+++ b/yang/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug5942Test.java
@@ -7,16 +7,15 @@
  */
 package org.opendaylight.yangtools.yang.stmt;
 
+import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.instanceOf;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertThat;
 import static org.junit.Assert.assertTrue;
 
-import java.util.List;
+import java.util.Collection;
 import java.util.Optional;
-import java.util.Set;
 import org.junit.Test;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
@@ -37,7 +36,7 @@ public class Bug5942Test {
         final DataSchemaNode root = schemaContext.getDataChildByName(QName.create("foo", "2016-06-02", "root"));
         assertTrue(root instanceof ContainerSchemaNode);
 
-        final Set<UsesNode> uses = ((ContainerSchemaNode) root).getUses();
+        final Collection<? extends UsesNode> uses = ((ContainerSchemaNode) root).getUses();
         assertEquals(1, uses.size());
         final UsesNode usesNode = uses.iterator().next();
 
@@ -50,7 +49,7 @@ public class Bug5942Test {
         assertThat(when, instanceOf(WithExpression.class));
         assertEquals("0!=1", when.getOriginalString());
 
-        final List<UnknownSchemaNode> unknownSchemaNodes = usesNode.getUnknownSchemaNodes();
+        final Collection<? extends UnknownSchemaNode> unknownSchemaNodes = usesNode.getUnknownSchemaNodes();
         assertEquals(1, unknownSchemaNodes.size());
         final UnknownSchemaNode unknownSchemaNode = unknownSchemaNodes.iterator().next();
         assertEquals("argument", unknownSchemaNode.getNodeParameter());