bumblebee-status/generate-base-tests.py

#!/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")
                if ".datetimetz" in line:
                    deps.extend(dependencies("bumblebee_status/modules/contrib/datetimetz.py"))
                elif not is_internal(dep):
                    deps.append(dep)
    return deps

def write_test(testname, modname, deps):
    fqmn = ".".join(["modules", testname.split(os.sep)[2], modname])
    if not os.path.exists(testname):
        with open(testname, "w") as f:
            f.writelines([
                "import pytest\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",
            "    __import__(\"{}\")\n\n".format(fqmn),
        ])

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