add unit test for klong_exports

klongpy/sys_fn.py

@@ -343,7 +343,11 @@ def hello():
                 print(f".py: {e}")
                 raise RuntimeError("module could not be imported: {x}")
     try:
-        import_items = filter(lambda p: not (p[0].startswith("__")), module.__dict__.items())
+        klong_exports = module.__dict__.get("klong_exports")
+        if klong_exports is None:
+            import_items = filter(lambda p: not (p[0].startswith("__")), module.__dict__.items())
+        else:
+            import_items = klong_exports.items()
         if import_items is not None:
             ctx = klong._context.pop()
             try:

tests/plugins/custom_export/__init__.py

@@ -0,0 +1,4 @@
+from .exports import command_map
+
+klong_exports = command_map()
+

tests/plugins/custom_export/exports.py

@@ -0,0 +1,6 @@
+
+def hello():
+    return "hello, world!"
+
+def command_map():
+    return {".hello": hello}

tests/test_sys_fn.py

@@ -252,6 +252,14 @@ def test_sys_python_load_custom_module(self):
         finally:
             sys.path.pop()
 
+    def test_sys_python_load_custom_export(self):
+        tests_dir = os.path.dirname(os.path.abspath(__file__))
+        plugins_dir = os.path.join(tests_dir, "plugins")
+        fpath = os.path.join(plugins_dir, "custom_export")
+        klong = KlongInterpreter()
+        self.assertEqual(klong(f'.py("{fpath}")'),1)
+        self.assertEqual(klong('.hello()'),"hello, world!")
+
     def test_eval_sys_random_number(self):
         r = eval_sys_random_number()
         r2 = eval_sys_random_number()