BUG-4688: Rework SchemaContext module lookups

[yangtools.git] / yang / yang-model-export / src / test / java / org / opendaylight / yangtools / yang / model / export / test / Bug6856Test.java

diff --git a/yang/yang-model-export/src/test/java/org/opendaylight/yangtools/yang/model/export/test/Bug6856Test.java b/yang/yang-model-export/src/test/java/org/opendaylight/yangtools/yang/model/export/test/Bug6856Test.java
index 028809a10dc6483f727482aa5ba39cef8ae577b4..9df4db341a29672ddbfe0b2c0e8f9b44f32a9088 100644 (file)
--- a/yang/yang-model-export/src/test/java/org/opendaylight/yangtools/yang/model/export/test/Bug6856Test.java
+++ b/yang/yang-model-export/src/test/java/org/opendaylight/yangtools/yang/model/export/test/Bug6856Test.java
@@ -15,7 +15,6 @@ import static org.junit.Assert.assertTrue;
 import java.io.BufferedOutputStream;
 import java.io.ByteArrayOutputStream;
 import java.io.OutputStream;
-import java.util.Date;
 import org.junit.Test;
 import org.opendaylight.yangtools.yang.common.QName;
 import org.opendaylight.yangtools.yang.model.api.Module;
@@ -34,9 +33,7 @@ public class Bug6856Test {
         final OutputStream byteArrayOutputStream = new ByteArrayOutputStream();
         final BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(byteArrayOutputStream);
 
-        final Date revision = QName.parseRevision("2017-02-28");
-
-        final Module fooModule = schemaContext.findModuleByName("foo", revision);
+        final Module fooModule = schemaContext.findModule("foo", QName.parseRevision("2017-02-28")).get();
         YinExportUtils.writeModuleToOutputStream(schemaContext, fooModule, bufferedOutputStream);
 
         final String output = byteArrayOutputStream.toString();
@@ -56,9 +53,7 @@ public class Bug6856Test {
         final OutputStream byteArrayOutputStream = new ByteArrayOutputStream();
         final BufferedOutputStream bufferedOutputStream = new BufferedOutputStream(byteArrayOutputStream);
 
-        final Date revision = QName.parseRevision("2017-02-28");
-
-        final Module barModule = schemaContext.findModuleByName("bar", revision);
+        final Module barModule = schemaContext.findModule("bar", QName.parseRevision("2017-02-28")).get();
         YinExportUtils.writeModuleToOutputStream(schemaContext, barModule, bufferedOutputStream);
 
         final String output = byteArrayOutputStream.toString();