BUG-4688: Rework SchemaContext module lookups

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

diff --git a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveIdentityTest.java b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveIdentityTest.java
index d309b864ff2609b3018ac064eba8ba2ea172155c..098022c1d44b25ac8220427946e709be662b87f7 100644 (file)
--- a/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveIdentityTest.java
+++ b/yang/yang-parser-impl/src/test/java/org/opendaylight/yangtools/yang/stmt/EffectiveIdentityTest.java
@@ -63,11 +63,7 @@ public class EffectiveIdentityTest {
 
         assertNotNull(result);
 
-        Module module = result.findModuleByName("identity-test",
-                SimpleDateFormatUtil.DEFAULT_DATE_REV);
-
-        assertNotNull(module);
-
+        Module module = result.findModule("identity-test", SimpleDateFormatUtil.DEFAULT_DATE_REV).get();
         Set<IdentitySchemaNode> identities = module.getIdentities();
 
         assertNotNull(identities);