Implement SourceInfo extraction
[yangtools.git] / parser / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug4610Test.java
1 /*
2  * Copyright (c) 2016 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.yang.stmt;
9
10 import static org.junit.jupiter.api.Assertions.assertEquals;
11 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
12 import static org.junit.jupiter.api.Assertions.assertNotEquals;
13 import static org.junit.jupiter.api.Assertions.assertSame;
14
15 import org.junit.jupiter.api.Test;
16 import org.opendaylight.yangtools.yang.common.QName;
17 import org.opendaylight.yangtools.yang.common.QNameModule;
18 import org.opendaylight.yangtools.yang.common.Revision;
19 import org.opendaylight.yangtools.yang.common.XMLNamespace;
20 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
21 import org.opendaylight.yangtools.yang.model.api.stmt.ContainerEffectiveStatement;
22 import org.opendaylight.yangtools.yang.model.api.stmt.ContainerStatement;
23 import org.opendaylight.yangtools.yang.model.api.stmt.GroupingEffectiveStatement;
24 import org.opendaylight.yangtools.yang.model.api.stmt.ModuleEffectiveStatement;
25 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
26
27 class Bug4610Test extends AbstractYangTest {
28     @Test
29     void test() {
30         final var context = assertEffectiveModelDir("/bugs/bug4610");
31         final Revision revision = Revision.of("2015-12-12");
32
33         // Original
34         final QName c1Bar = QName.create(QNameModule.create(XMLNamespace.of("bar"), revision), "c1");
35         final ContainerEffectiveStatement g1container = findContainer(context, QName.create(c1Bar, "g1"), c1Bar);
36         final QName g1argument = g1container.argument();
37         final ContainerStatement g1original = g1container.getDeclared();
38
39         final ContainerEffectiveStatement g2container = findContainer(context, QName.create(c1Bar, "g2"), c1Bar);
40         assertEquals(g1argument, g2container.argument());
41         assertSame(g1original, g2container.getDeclared());
42
43         final QName c1Foo = QName.create(QNameModule.create(XMLNamespace.of("foo"), revision), "c1");
44         final ContainerEffectiveStatement g3container = findContainer(context, QName.create(c1Foo, "g3"), c1Foo);
45         assertNotEquals(g1argument, g3container.argument());
46         assertSame(g1original, g3container.getDeclared());
47
48         final SchemaTreeEffectiveStatement<?> rootContainer = context.getModuleStatement(c1Foo.getModule())
49             .findSchemaTreeNode(QName.create(c1Foo, "root"), c1Foo).orElseThrow();
50         assertInstanceOf(ContainerEffectiveStatement.class, rootContainer);
51         assertNotEquals(g1argument, rootContainer.argument());
52         assertSame(g1original, rootContainer.getDeclared());
53     }
54
55     private static ContainerEffectiveStatement findContainer(final EffectiveModelContext context, final QName grouping,
56         final QName container) {
57         final ModuleEffectiveStatement module = context.getModuleStatement(grouping.getModule());
58         final GroupingEffectiveStatement grp = module.streamEffectiveSubstatements(GroupingEffectiveStatement.class)
59             .filter(stmt -> grouping.equals(stmt.argument()))
60             .findAny().orElseThrow();
61
62         final SchemaTreeEffectiveStatement<?> node = grp.findSchemaTreeNode(container).orElse(null);
63         assertInstanceOf(ContainerEffectiveStatement.class, node);
64         return (ContainerEffectiveStatement) node;
65     }
66 }