testapps.py

import unittest
import os
import sys
import argparse
import importlib
from core.helpers import list_apps
from core.config.paths import apps_path


def cmd_line():
    parser = argparse.ArgumentParser("Test Apps")
    parser.add_argument('-a', '--apps', nargs='*', type=str, required=False,
                        help='List of apps for which you would like to test')
    parser.add_argument('-A', '--all', action='store_true', help='Test all apps')
    args = parser.parse_args()
    if (not args.all) and (not args.apps):
        parser.print_help()
    return args


def get_tests(app_name):
    tests_path = os.path.join(apps_path, app_name, 'tests')
    if os.path.isdir(tests_path):
        test_files = [os.path.splitext(f)[0]
                      for f in os.listdir(tests_path) if (os.path.isfile(os.path.join(tests_path, f))
                                                          and f.endswith('.py')
                                                          and f != '__init__.py')]
        test_modules = [importlib.import_module('apps.{0}.tests.{1}'.format(app_name, test_module))
                        for test_module in test_files]
        return test_modules
    else:
        print('App {0} has no test directory!'.format(app_name))


def test_app(app_name):
    print('Testing app: {0}'.format(app))
    test_modules = get_tests(app_name)
    if test_modules:
        suite = unittest.TestSuite()
        suite.addTests([unittest.TestLoader().loadTestsFromModule(test_module)
                        for test_module in test_modules])
        return unittest.TextTestRunner(verbosity=1).run(suite).wasSuccessful()
    elif test_modules is None or len(test_modules) == 0:
        print("App {0} has no tests. Don't be that person. Write your tests.".format(app_name))
        return True

if __name__ == '__main__':
    cmd_args = cmd_line()
    all_apps = list_apps()
    ret = True
    if cmd_args.all:
        for app in all_apps:
            ret &= test_app(app)
    elif cmd_args.apps:
        for app in cmd_args.apps:
            if app in all_apps:
                ret &= test_app(app)
            else:
                print('App {0} not found!'.format(app))
    sys.exit(not ret)