Make Module.getContact()/getOrganization() return Optional

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
index 31cf05c73804612d47fc323c19ff2a89636018c2..42bc8ed3be0bf0e82dbc41b21b5cbe23d0d5f743 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/YangParserTest.java
@@ -24,6 +24,7 @@ import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Iterator;
 import java.util.List;
+import java.util.Optional;
 import java.util.Set;
 import org.junit.Before;
 import org.junit.Test;
@@ -107,8 +108,8 @@ public class YangParserTest {
         assertEquals("baz", import3.getModuleName());
         assertEquals(BAZ.getRevision(), import3.getRevision());
 
-        assertEquals("opendaylight", foo.getOrganization());
-        assertEquals("http://www.opendaylight.org/", foo.getContact());
+        assertEquals(Optional.of("opendaylight"), foo.getOrganization());
+        assertEquals(Optional.of("http://www.opendaylight.org/"), foo.getContact());
         assertEquals(Revision.ofNullable("2013-02-27"), foo.getRevision());
         assertNull(foo.getReference());
     }