seitime-frappe/py/webnotes/tests.py
2011-09-14 18:55:15 +05:30

63 lines
1.4 KiB
Python

"""
Run tests from modules. Sets up database connection, modules path and session before running test
Usage: from shell, run
python tests.py [test modules]
Options:
test modules: list of modules separated by space
if no modules are specified, it will run all "tests.py" files from all modules
"""
import sys, os
import unittest
# webnotes path
sys.path.append('lib/py')
# modules path
import webnotes
import webnotes.defs
if webnotes.defs.__dict__.get('modules_path'):
sys.path.append(webnotes.defs.modules_path)
def get_tests():
"""
Returns list of test modules identified by "test*.py"
"""
ret = []
for walk_tuple in os.walk(webnotes.defs.modules_path):
for test_file in filter(lambda x: x.startswith('test') and x.endswith('.py'), walk_tuple[2]):
dir_path = os.path.relpath(walk_tuple[0], webnotes.defs.modules_path)
if dir_path=='.':
ret.append(test_file[:-3])
else:
ret.append(dir_path.replace('/', '.') + '.' + test_file[:-3])
return ret
def setup():
"""
Sets up connection and session
"""
from webnotes.db import Database
webnotes.conn = Database()
webnotes.session = {'user':'Administrator'}
if __name__=='__main__':
setup()
if len(sys.argv) > 1:
tests_list = sys.argv[1:]
# for unittest.main
sys.argv = sys.argv[:1]
else:
tests_list = get_tests()
for tests in tests_list:
exec 'from %s import *' % str(tests)
unittest.main()