From dc70527797a84c568559c5ef4759a19f975fe9bc Mon Sep 17 00:00:00 2001
From: tobi-wan-kenobi <tobi-wan-kenobi@users.noreply.github.com>
Date: Mon, 20 Jul 2020 13:16:37 +0200
Subject: [PATCH] [util] add a script to generate basic autotests

Add a small utility that can generate a basic "import this unless you
are missing dependencies" test for all modules.
---
 generate-base-tests.py | 82 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100755 generate-base-tests.py

diff --git a/generate-base-tests.py b/generate-base-tests.py
new file mode 100755
index 0000000..2c08165
--- /dev/null
+++ b/generate-base-tests.py
@@ -0,0 +1,82 @@
+#!/usr/bin/env python
+
+import os
+import re
+import sys
+import glob
+
+def is_psl(module):
+    lib_path = os.path.dirname(os.__file__)
+    old_sys = sys.path
+    sys.path = [lib_path]
+    is_psl = True
+    try:
+        __import__(module)
+    except Exception as e:
+        is_psl = False
+    sys.path = old_sys
+
+    return is_psl
+
+
+def is_internal(module):
+    if module.startswith("core.") or module == "core": return True
+    if module.startswith("util.") or module == "util": return True
+    if module.startswith("."): return True
+
+    return is_psl(module)
+
+def dependencies(filename):
+    deps = []
+    with open(filename) as f:
+        for line in f:
+            if "import" in line:
+                match = re.match("\s*(from (\S+) )?import (\S+)", line)
+                if not match: continue
+                dep = match.group(2) or match.group(3)
+                if "util.popup" in dep or ("util" in line and "popup" in line):
+                    deps.append("tkinter")
+                elif not is_internal(dep):
+                    deps.append(dep)
+    return deps
+
+def write_test(testname, modname, deps):
+    if not os.path.exists(testname):
+        with open(testname, "w") as f:
+            f.writelines([
+                "import pytest\n\n",
+                "import core.module\n\n",
+            ])
+            for dep in deps:
+                f.write("pytest.importorskip(\"{}\")\n\n".format(dep))
+
+    with open(testname) as f:
+        for line in f:
+            if "def test_load_module(" in line:
+                print("skipping {}, already contains test".format(modname))
+                return
+
+    print("writing base test for {}".format(modname))
+    with open(testname, "a+") as f:
+        f.writelines([
+            "def test_load_module():\n",
+            "    core.module.load(\"{}\")\n\n".format(modname),
+        ])
+
+def main():
+    for f in glob.glob("bumblebee_status/modules/*/*.py"):
+        if os.path.basename(f) == "__init__.py": continue
+
+        modname = os.path.splitext(os.path.basename(f))[0]
+
+        modpath = os.path.dirname(f)
+        deps = dependencies(f)
+        testname = os.path.join("tests", "modules", modpath.split(os.sep)[2], "test_{}.py".format(modname))
+
+        write_test(testname, modname, deps)
+        
+
+if __name__ == "__main__":
+    main()
+
+# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4