unittest_suite.py

#!/usr/bin/python

import os, sys
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('/', '.'))
			suite = loader.loadTestsFromName(mod)
			suites.append(suite)


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