Update ChoiceSchemaNode design

[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / yin / YinFileChoiceStmtTest.java

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileChoiceStmtTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileChoiceStmtTest.java
index 61351d5b8e92e27598abaf88151c045590d9c67c..5be7b4dca9c9fbc4f64086780b4a7a0d85449f6b 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileChoiceStmtTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/yin/YinFileChoiceStmtTest.java
@@ -61,7 +61,7 @@ public class YinFileChoiceStmtTest {
         assertEquals(1, choice.getCases().size());
 
         // this choice is augmented (see main-impl.yang.xml)
-        final Iterator<ChoiceCaseNode> casesIterator = choice.getCases().iterator();
+        final Iterator<ChoiceCaseNode> casesIterator = choice.getCases().values().iterator();
         final ChoiceCaseNode caseNode = casesIterator.next();
         assertEquals("main-impl", caseNode.getQName().getLocalName());
         assertEquals(13, caseNode.getChildNodes().size());