schema: fix type module method

The module function of the Type class was not working. Fix it, and
restore the unit test. Note that it returns the module where the type is
defined, not the typedef.

Signed-off-by: Samuel Gauthier <samuel.gauthier@6wind.com>
Acked-by: Robin Jarry <robin@jarry.cc>

Author: samuel-gauthier

libyang/schema.py

@@ -626,11 +626,9 @@ def all_patterns(self) -> Iterator[Tuple[str, bool]]:
             yield from self.patterns()
 
     def module(self) -> Module:
-        # TODO: pointer to the parsed module wehere is the type defined is in self.cdata_parsed.pmod
-        # however there is no way how to get name of the module from lysp_module
-        if not self.cdata.der.module:
+        if not self.cdata_parsed:
             return None
-        return Module(self.context, self.cdata.der.module)
+        return Module(self.context, self.cdata_parsed.pmod.mod)
 
     def extensions(self) -> Iterator[ExtensionCompiled]:
         for extension in ly_array_iter(self.cdata.exts):

tests/test_schema.py

@@ -376,9 +376,9 @@ def test_leaf_type_derived(self):
         self.assertIsInstance(t, Type)
         self.assertEqual(t.name(), "types:host")
         self.assertEqual(t.base(), Type.STRING)
-        # mod = t.module()
-        # self.assertIsNot(mod, None)
-        # self.assertEqual(mod.name(), "wtf-types")
+        mod = t.module()
+        self.assertIsNot(mod, None)
+        self.assertEqual(mod.name(), "yolo-system")
 
     def test_leaf_type_status(self):
         leaf = next(self.ctx.find_path("/yolo-system:conf/yolo-system:hostname"))