Bug 8307: Add the option for activating deviation statements

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/DeviationStmtTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/DeviationStmtTest.java
index 5745783a88f33161690bd09b382775db64e33422..7adc4f9595fefc7398cb56011a708ec472b3fab6 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/DeviationStmtTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/DeviationStmtTest.java
@@ -12,6 +12,7 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
+import static org.opendaylight.yangtools.yang.stmt.StmtTestUtils.sourceForResource;
 
 import java.text.ParseException;
 import java.util.Date;
@@ -29,27 +30,28 @@ import org.opendaylight.yangtools.yang.parser.spi.meta.ReactorException;
 import org.opendaylight.yangtools.yang.parser.spi.source.StatementStreamSource;
 import org.opendaylight.yangtools.yang.parser.stmt.reactor.CrossSourceStatementReactor;
 import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangInferencePipeline;
-import org.opendaylight.yangtools.yang.parser.stmt.rfc6020.YangStatementSourceImpl;
 
 public class DeviationStmtTest {
 
-    private static final StatementStreamSource FOO_MODULE =
-            new YangStatementSourceImpl("/deviation-stmt-test/foo.yang", false);
+    private static final StatementStreamSource FOO_MODULE = sourceForResource("/deviation-stmt-test/foo.yang");
+    private static final StatementStreamSource FOO_IMP_MODULE = sourceForResource("/deviation-stmt-test/foo-imp.yang");
+    private static final StatementStreamSource BAR_MODULE = sourceForResource("/deviation-stmt-test/bar.yang");
+    private static final StatementStreamSource BAR_IMP_MODULE = sourceForResource("/deviation-stmt-test/bar-imp.yang");
 
     @Test
     public void testDeviationAndDeviate() throws ReactorException, ParseException {
         final CrossSourceStatementReactor.BuildAction reactor = YangInferencePipeline.RFC6020_REACTOR.newBuild();
-        reactor.addSources(FOO_MODULE);
+        reactor.addSources(FOO_MODULE, FOO_IMP_MODULE, BAR_MODULE, BAR_IMP_MODULE);
 
         final SchemaContext schemaContext = reactor.buildEffective();
         assertNotNull(schemaContext);
 
-        final Date revision = SimpleDateFormatUtil.getRevisionFormat().parse("2016-06-23");
+        Date revision = SimpleDateFormatUtil.getRevisionFormat().parse("2016-06-23");
 
-        final Module testModule = schemaContext.findModuleByName("foo", revision);
+        Module testModule = schemaContext.findModuleByName("foo", revision);
         assertNotNull(testModule);
 
-        final Set<Deviation> deviations = testModule.getDeviations();
+        Set<Deviation> deviations = testModule.getDeviations();
         assertEquals(4, deviations.size());
 
         for (Deviation deviation : deviations) {
@@ -64,12 +66,12 @@ public class DeviationStmtTest {
                 assertEquals(DeviateKind.ADD, deviates.iterator().next().getDeviateType());
                 assertEquals("added-def-val", deviates.iterator().next().getDeviatedDefault());
                 assertFalse(deviates.iterator().next().getDeviatedConfig());
+                assertTrue(deviates.iterator().next().getDeviatedMandatory());
             } else if ("test-leaf-list".equals(targetLocalName)) {
                 assertEquals(3, deviates.size());
                 for (DeviateDefinition deviate : deviates) {
                     if (DeviateKind.ADD.equals(deviate.getDeviateType())) {
                         assertEquals(12, deviate.getDeviatedMaxElements().intValue());
-                        assertTrue(deviate.getDeviatedMandatory());
                     } else if (DeviateKind.REPLACE.equals(deviate.getDeviateType())) {
                         assertEquals(5, deviate.getDeviatedMinElements().intValue());
                         assertTrue(deviate.getDeviatedType() instanceof UnsignedIntegerTypeDefinition);
@@ -84,5 +86,69 @@ public class DeviationStmtTest {
                 assertEquals(2, deviates.iterator().next().getDeviatedUniques().size());
             }
         }
+
+        revision = SimpleDateFormatUtil.getRevisionFormat().parse("2016-09-22");
+        testModule = schemaContext.findModuleByName("bar", revision);
+        assertNotNull(testModule);
+
+        deviations = testModule.getDeviations();
+        assertEquals(7, deviations.size());
+
+        Deviation deviation1 = null;
+        Deviation deviation2 = null;
+        Deviation deviation3 = null;
+        Deviation deviation4 = null;
+        Deviation deviation5 = null;
+        Deviation deviation6 = null;
+        Deviation deviation7 = null;
+
+        for (Deviation deviation : deviations) {
+            final List<DeviateDefinition> deviates = deviation.getDeviates();
+            final String targetLocalName = deviation.getTargetPath().getLastComponent().getLocalName();
+
+            if ("bar-container-1".equals(targetLocalName)) {
+                deviation1 = deviation;
+            }
+
+            if ("bar-container-2".equals(targetLocalName)) {
+                DeviateKind deviateKind = deviates.iterator().next().getDeviateType();
+                if (deviateKind.equals(DeviateKind.DELETE)) {
+                    deviation2 = deviation;
+                } else if (deviateKind.equals(DeviateKind.ADD)) {
+                    deviation3 = deviation;
+                }
+            }
+
+            if ("bar-leaf-1".equals(targetLocalName)) {
+                if ("desc".equals(deviation.getDescription())) {
+                    deviation4 = deviation;
+                } else {
+                    deviation5 = deviation;
+                }
+            }
+
+            if ("bar-leaf-2".equals(targetLocalName)) {
+                if ("ref".equals(deviation.getReference())) {
+                    deviation6 = deviation;
+                } else {
+                    deviation7 = deviation;
+                }
+            }
+        }
+
+        assertEquals(1, deviation1.getUnknownSchemaNodes().size());
+        assertTrue(deviation1.equals(deviation1));
+        assertFalse(deviation1.equals(null));
+        assertFalse(deviation1.equals("str"));
+
+        DeviateDefinition deviate = deviation1.getDeviates().iterator().next();
+        assertTrue(deviate.equals(deviate));
+        assertFalse(deviate.equals(null));
+        assertFalse(deviate.equals("str"));
+
+        assertFalse(deviation1.equals(deviation2));
+        assertFalse(deviation2.equals(deviation3));
+        assertFalse(deviation4.equals(deviation5));
+        assertFalse(deviation6.equals(deviation7));
     }
 }