BUG-4688: switch revisions from Date to Revision

[yangtools.git] / yang / yang-parser-impl / src / test / java / org / opendaylight / yangtools / yang / stmt / Bug8597Test.java

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug8597Test.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug8597Test.java
index 46d3ffc67a22b3a9aa09e3ad3eea17923e1568ed..f4a7f6d60e1f27cc0c8f75bfecce78f65a1fd82a 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug8597Test.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/Bug8597Test.java
@@ -11,9 +11,10 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.fail;
 
+import java.util.Optional;
 import java.util.Set;
 import org.junit.Test;
-import org.opendaylight.yangtools.yang.common.QName;
+import org.opendaylight.yangtools.yang.common.Revision;
 import org.opendaylight.yangtools.yang.model.api.Module;
 import org.opendaylight.yangtools.yang.model.api.ModuleImport;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
@@ -24,18 +25,18 @@ public class Bug8597Test {
         final SchemaContext context = StmtTestUtils.parseYangSources("/bugs/bug8597");
         assertNotNull(context);
 
-        final Module foo = context.findModule("foo", null).get();
+        final Module foo = context.findModule("foo").get();
         final Set<ModuleImport> imports = foo.getImports();
 
         for (final ModuleImport moduleImport : imports) {
             switch (moduleImport.getModuleName()) {
                 case "bar":
-                    assertEquals(QName.parseRevision("1970-01-01"), moduleImport.getRevision());
+                    assertEquals(Optional.of(Revision.valueOf("1970-01-01")), moduleImport.getRevision());
                     assertEquals("bar-ref", moduleImport.getReference());
                     assertEquals("bar-desc", moduleImport.getDescription());
                     break;
                 case "baz":
-                    assertEquals(QName.parseRevision("2010-10-10"), moduleImport.getRevision());
+                    assertEquals(Optional.of(Revision.valueOf("2010-10-10")), moduleImport.getRevision());
                     assertEquals("baz-ref", moduleImport.getReference());
                     assertEquals("baz-desc", moduleImport.getDescription());
                     break;