Populate parser/ hierarchy

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

diff --git a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4610Test.java b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4610Test.java
new file mode 100644 (file)
index 0000000..e0c8922
--- /dev/null
+++ b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug4610Test.java
@@ -0,0 +1,67 @@
+/*
+ * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.yangtools.yang.stmt;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotEquals;
+import static org.junit.Assert.assertSame;
+
+import org.junit.Test;
+import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.QNameModule;
+import org.opendaylight.yangtools.yang.common.Revision;
+import org.opendaylight.yangtools.yang.common.XMLNamespace;
+import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
+import org.opendaylight.yangtools.yang.model.api.stmt.ContainerEffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.ContainerStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.GroupingEffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
+import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
+
+public class Bug4610Test {
+    @Test
+    public void test() throws Exception {
+        final EffectiveModelContext context = StmtTestUtils.parseYangSources("/bugs/bug4610");
+        final Revision revision = Revision.of("2015-12-12");
+
+        // Original
+        final QName c1Bar = QName.create(QNameModule.create(XMLNamespace.of("bar"), revision), "c1");
+        final ContainerEffectiveStatement g1container = findContainer(context, QName.create(c1Bar, "g1"), c1Bar);
+        final QName g1argument = g1container.argument();
+        final ContainerStatement g1original = g1container.getDeclared();
+
+        final ContainerEffectiveStatement g2container = findContainer(context, QName.create(c1Bar, "g2"), c1Bar);
+        assertEquals(g1argument, g2container.argument());
+        assertSame(g1original, g2container.getDeclared());
+
+        final QName c1Foo = QName.create(QNameModule.create(XMLNamespace.of("foo"), revision), "c1");
+        final ContainerEffectiveStatement g3container = findContainer(context, QName.create(c1Foo, "g3"), c1Foo);
+        assertNotEquals(g1argument, g3container.argument());
+        assertSame(g1original, g3container.getDeclared());
+
+        final SchemaTreeEffectiveStatement<?> rootContainer = context.getModuleStatement(c1Foo.getModule())
+            .findSchemaTreeNode(QName.create(c1Foo, "root"), c1Foo).orElseThrow();
+        assertThat(rootContainer, instanceOf(ContainerEffectiveStatement.class));
+        assertNotEquals(g1argument, rootContainer.argument());
+        assertSame(g1original, rootContainer.getDeclared());
+    }
+
+    private static ContainerEffectiveStatement findContainer(final EffectiveModelContext context, final QName grouping,
+            final QName container) {
+        final ModuleEffectiveStatement module = context.getModuleStatement(grouping.getModule());
+        final GroupingEffectiveStatement grp = module.streamEffectiveSubstatements(GroupingEffectiveStatement.class)
+            .filter(stmt -> grouping.equals(stmt.argument()))
+            .findAny().orElseThrow();
+
+        final SchemaTreeEffectiveStatement<?> node = grp.findSchemaTreeNode(container).orElse(null);
+        assertThat(node, instanceOf(ContainerEffectiveStatement.class));
+        return (ContainerEffectiveStatement) node;
+    }
+}