unittest_suite.py

#!/usr/bin/python

import os
import unittest

# ensure the root is where it should be
root = os.path.abspath(os.path.dirname(__file__))
from client import setup_modules
setup_modules.setup(base_path=root, root_module_name="autotest_lib")


suites = []
def lister(dummy, dirname, files):
    loader = unittest.TestLoader()
    for f in files:
        if f.endswith('_unittest.py'):
            temp = os.path.join(dirname, f).strip('.py')
            mod = 'autotest_lib' + temp[len(root):].replace('/', '.')
            try:
                suite = loader.loadTestsFromName(mod)
                suites.append(suite)
            except:
                print "module failed to load: %s" % mod


if __name__ == "__main__":
    os.path.walk(root, lister, None)
    alltests = unittest.TestSuite(suites)
    runner = unittest.TextTestRunner(verbosity=2)
    runner.run(alltests)