Cleanup DocumentedNode

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveModuleTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveModuleTest.java
index c96fccc457b2161def2442c65ec9c3ff41537146..7a5a4482b626e5a343ff1a654d52a137d315266e 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveModuleTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveModuleTest.java
@@ -112,7 +112,7 @@ public class EffectiveModuleTest {
         final SchemaPath importedContSchemaPath = SchemaPath.create(true, importedContQName);
         assertEquals(importedContSchemaPath, deviationStmt.getTargetPath());
         assertEquals(DeviateKind.ADD, deviationStmt.getDeviates().iterator().next().getDeviateType());
-        assertEquals("deviate reference", deviationStmt.getReference());
+        assertEquals(Optional.of("deviate reference"), deviationStmt.getReference());
 
         final Set<IdentitySchemaNode> identities = rootModule.getIdentities();
         assertEquals(1, identities.size());
@@ -124,8 +124,8 @@ public class EffectiveModuleTest {
         assertNotNull(featureStmt);
         assertEquals(FEATURE1, featureStmt.getQName());
         assertEquals(FEATURE1_SCHEMA_PATH, featureStmt.getPath());
-        assertEquals("feature1 description", featureStmt.getDescription());
-        assertEquals("feature1 reference", featureStmt.getReference());
+        assertEquals(Optional.of("feature1 description"), featureStmt.getDescription());
+        assertEquals(Optional.of("feature1 reference"), featureStmt.getReference());
         assertEquals(Status.CURRENT, featureStmt.getStatus());
 
         final List<ExtensionDefinition> extensionSchemaNodes = rootModule.getExtensionSchemaNodes();