Clean up TreeNode API
[yangtools.git] / parser / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug6972Test.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.assertSame;
13
14 import org.junit.jupiter.api.Test;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.common.Revision;
17 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
18 import org.opendaylight.yangtools.yang.model.api.LeafSchemaNode;
19 import org.opendaylight.yangtools.yang.model.api.Module;
20 import org.opendaylight.yangtools.yang.model.api.stmt.UnitsEffectiveStatement;
21
22 class Bug6972Test extends AbstractYangTest {
23     @Test
24     void allUnitsShouldBeTheSameInstance() {
25         final var schemaContext = assertEffectiveModelDir("/bugs/bug6972");
26         assertEquals(3, schemaContext.getModules().size());
27
28         final Revision revision = Revision.of("2016-10-20");
29         final var foo = schemaContext.findModule("foo", revision).orElseThrow();
30         final var bar = schemaContext.findModule("bar", revision).orElseThrow();
31         final var baz = schemaContext.findModule("baz", revision).orElseThrow();
32
33         final QName barExportCont = QName.create("bar-ns", "bar-export", revision);
34         final QName barFooCont = QName.create("bar-ns", "bar-foo", revision);
35         final QName barFooLeaf = QName.create("bar-ns", "foo", revision);
36
37         final var unitsBar1 = getEffectiveUnits(bar, barExportCont, barFooLeaf);
38         assertSame(unitsBar1, getEffectiveUnits(bar, barFooCont, barFooLeaf));
39
40         final QName bazExportCont = QName.create("baz-ns", "baz-export", revision);
41         final QName bazFooCont = QName.create("baz-ns", "baz-foo", revision);
42         final QName bazFooLeaf = QName.create("baz-ns", "foo", revision);
43
44         assertSame(unitsBar1, getEffectiveUnits(baz, bazExportCont, bazFooLeaf));
45         assertSame(unitsBar1, getEffectiveUnits(baz, bazFooCont, bazFooLeaf));
46     }
47
48     private static UnitsEffectiveStatement getEffectiveUnits(final Module module, final QName containerQName,
49             final QName leafQName) {
50         final var cont = assertInstanceOf(ContainerSchemaNode.class, module.getDataChildByName(containerQName));
51         return assertInstanceOf(LeafSchemaNode.class, cont.getDataChildByName(leafQName)).asEffectiveStatement()
52             .streamEffectiveSubstatements(UnitsEffectiveStatement.class)
53             .findFirst()
54             .orElse(null);
55     }
56 }