Peel MustConstraintAware from ConstraintDefinition

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserSimpleTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserSimpleTest.java
index bff629f781dfb063a8e83a27f2c881ff128b272b..a827ebf5ddf120df4e02540ae263400e34b3345f 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserSimpleTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserSimpleTest.java
@@ -76,8 +76,8 @@ public class YangParserSimpleTest {
         assertTrue(data.isMandatory());
         final ConstraintDefinition constraints = data.getConstraints();
         assertEquals("class != 'wheel'", constraints.getWhenCondition().get().toString());
-        final Collection<MustDefinition> mustConstraints = constraints.getMustConstraints();
-        assertEquals(2, constraints.getMustConstraints().size());
+        final Collection<MustDefinition> mustConstraints = data.getMustConstraints();
+        assertEquals(2, mustConstraints.size());
 
         final String must1 = "ifType != 'ethernet' or (ifType = 'ethernet' and ifMTU = 1500)";
         final String must2 = "ifType != 'atm' or (ifType = 'atm' and ifMTU <= 17966 and ifMTU >= 64)";
@@ -121,8 +121,8 @@ public class YangParserSimpleTest {
         // constraints
         final ConstraintDefinition constraints = nodes.getConstraints();
         assertEquals("class != 'wheel'", constraints.getWhenCondition().get().toString());
-        final Collection<MustDefinition> mustConstraints = constraints.getMustConstraints();
-        assertEquals(2, constraints.getMustConstraints().size());
+        final Collection<MustDefinition> mustConstraints = nodes.getMustConstraints();
+        assertEquals(2, mustConstraints.size());
 
         final String must1 = "ifType != 'atm' or (ifType = 'atm' and ifMTU <= 17966 and ifMTU >= 64)";
         final String errMsg1 = "An atm MTU must be  64 .. 17966";