Further cleanup of tests

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

diff --git a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YT956Test.java b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YT956Test.java
index 7f2693c838f216738150948159909778c22d5425..19cf648685c3fe6fe57b9c6450b26af0c793c176 100644 (file)
--- a/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YT956Test.java
+++ b/parser/yang-parser-rfc7950/src/test/java/org/opendaylight/yangtools/yang/stmt/YT956Test.java
@@ -16,15 +16,15 @@ import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
 
-public class YT956Test {
+public class YT956Test extends AbstractYangTest {
     private static final QName ANOTHER_CONTAINER = QName.create("http://www.example.com/anothermodule",
         "another-container");
     private static final QName FIRST_AUGMENT = QName.create("http://www.example.com/mainmodule", "first-augment");
 
     @Test
-    public void testAugmentationConditional() throws Exception {
-        final DataSchemaNode another = StmtTestUtils.parseYangSources("/bugs/YT956/")
-                .findDataChildByName(ANOTHER_CONTAINER).get();
+    public void testAugmentationConditional() {
+        final DataSchemaNode another = assertEffectiveModelDir("/bugs/YT956/")
+                .getDataChildByName(ANOTHER_CONTAINER);
         assertThat(another, isA(ContainerSchemaNode.class));
         final ContainerSchemaNode anotherContainer = (ContainerSchemaNode) another;