Remove EffectiveStatement namespaces
[yangtools.git] / parser / yang-parser-rfc7950 / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug6183Test.java
index 8414172d0b5c5e003106078cc7b6760c2fbdef2b..aadcdf7867d742c81fc0bfc5242af08a38697dbf 100644 (file)
@@ -13,7 +13,6 @@ import static org.junit.Assert.assertTrue;
 
 import org.junit.Test;
 import org.opendaylight.yangtools.yang.common.QName;
-import org.opendaylight.yangtools.yang.model.api.CaseSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.ChoiceSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.DataNodeContainer;
@@ -48,10 +47,10 @@ public class Bug6183Test extends AbstractYangTest {
 
         assertEquals(4, myChoice.getCases().size());
 
-        final CaseSchemaNode implCase = myChoice.findCase(foo("implicit-case-container")).get();
-        final CaseSchemaNode declCaseOne = myChoice.findCase(foo("declared-case-one")).get();
-        final CaseSchemaNode secondImplCase = myChoice.findCase(foo("second-implicit-case-container")).get();
-        final CaseSchemaNode declCaseTwo = myChoice.findCase(foo("declared-case-two")).get();
+        final var implCase = myChoice.findCaseNode(foo("implicit-case-container")).orElseThrow();
+        final var declCaseOne = myChoice.findCaseNode(foo("declared-case-one")).orElseThrow();
+        final var secondImplCase = myChoice.findCaseNode(foo("second-implicit-case-container")).orElseThrow();
+        final var declCaseTwo = myChoice.findCaseNode(foo("declared-case-two")).orElseThrow();
 
         assertEquals(1, declCaseOne.getChildNodes().size());
         assertFalse(getLeafSchemaNode(declCaseOne, "leaf-in-declare-case-one").isAugmenting());