Clean up TreeNode API
[yangtools.git] / parser / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / YT1383Test.java
1 /*
2  * Copyright (c) 2021 PANTHEON.tech, s.r.o. 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.assertNotNull;
13 import static org.junit.jupiter.api.Assertions.assertNull;
14
15 import org.junit.jupiter.api.Test;
16 import org.opendaylight.yangtools.yang.model.api.stmt.CaseEffectiveStatement;
17 import org.opendaylight.yangtools.yang.model.api.stmt.CaseStatement;
18 import org.opendaylight.yangtools.yang.model.api.stmt.ChoiceEffectiveStatement;
19 import org.opendaylight.yangtools.yang.model.api.stmt.ContainerStatement;
20
21 class YT1383Test extends AbstractYangTest {
22     @Test
23     void testDeclaredImplicitCase() {
24         final var foo = assertEffectiveModel("/bugs/YT1383/foo.yang").getModuleStatements().values().iterator()
25             .next().findFirstEffectiveSubstatement(ChoiceEffectiveStatement.class).orElseThrow();
26
27         // Effective view of things
28         final var effStatements = foo.effectiveSubstatements();
29         assertEquals(2, effStatements.size());
30         final var bar = effStatements.get(0);
31         assertInstanceOf(CaseEffectiveStatement.class, bar);
32         assertNotNull(bar.getDeclared());
33         final var baz = effStatements.get(1);
34         assertInstanceOf(CaseEffectiveStatement.class, baz);
35         assertNull(baz.getDeclared());
36
37         // Declared view of things
38         final var fooDecl = foo.getDeclared();
39         assertNotNull(fooDecl);
40         final var declStatements = fooDecl.declaredSubstatements();
41         assertEquals(2, declStatements.size());
42         final var barDecl = declStatements.get(0);
43         assertInstanceOf(CaseStatement.class, barDecl);
44         final var bazDecl = declStatements.get(1);
45         assertInstanceOf(ContainerStatement.class, bazDecl);
46     }
47 }