test.py

#!/usr/bin/env python3
import pathlib
import tempfile
from argparse import ArgumentParser

from test.config import Config
from test.testunit import TestResult
from test.testlist import tests as all_tests
from test.utils import Colour


def main():
    parser = ArgumentParser()

    parser.add_argument("tests", metavar="test", type=str, nargs="+", help="Names of tests to be run, or 'all'")
    parser.add_argument("-v", "--verbose", action="store_true", help="Print per-test information")
    parser.add_argument("--wait", action="store_true", help="Once the tests are complete, wait for user input before "
                                                            "deleting the artifacts")
    parser.add_argument("--python", type=str, default="python3", help="Name of Python executable")

    args = parser.parse_args()

    config = Config()
    config.verbose = args.verbose
    config.python = args.python

    by_name = dict()
    for test in all_tests:
        by_name[test.name] = test

    if args.tests == ["all"]:
        args.tests = sorted(by_name)

    tempdir = tempfile.TemporaryDirectory(prefix="ljd-test-")

    for name in args.tests:
        test = by_name[name]
        result = test.test(config, pathlib.Path(tempdir.name))

        if result == TestResult.PASS:
            Colour.GREEN.write("[*] PASS ")
        elif result == TestResult.FAIL:
            Colour.RED.write("[x] FAIL ")
        else:
            assert result == TestResult.ERROR
            Colour.RED.set_bg()
            Colour.WHITE.write("[!]")
            Colour.RED.write(" ERR  ")

        print(name)

    if args.wait:
        input("Press enter to continue, files at %s " % tempdir.name)


if __name__ == "__main__":
    main()