Update TypeDefinition design

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TypedefSubStmtsTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TypedefSubStmtsTest.java
index 8cd5caebf439eb2e2eedeb14551008baa034cf5c..b5959a2ba0d363f99926a40ad1085b791fc57bc6 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TypedefSubStmtsTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/TypedefSubStmtsTest.java
@@ -12,6 +12,7 @@ import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
 import static org.opendaylight.yangtools.yang.stmt.StmtTestUtils.sourceForResource;
 
+import java.util.Optional;
 import java.util.Set;
 import org.junit.Test;
 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
@@ -40,7 +41,7 @@ public class TypedefSubStmtsTest {
         TypeDefinition<?> typedef = typedefs.iterator().next();
         assertEquals("time-of-the-day", typedef.getQName().getLocalName());
         assertEquals("string", typedef.getBaseType().getQName().getLocalName());
-        assertEquals("24-hour-clock", typedef.getUnits());
-        assertEquals("1am", typedef.getDefaultValue().toString());
+        assertEquals(Optional.of("24-hour-clock"), typedef.getUnits());
+        assertEquals("1am", typedef.getDefaultValue().map(Object::toString).orElse(null));
     }
 }