def allTest():
''''' 创建测试集'''
suite1 = TestLoader().loadTestsFromTestCase(login_case_ddt.LoginCase)
suite2 = TestLoader().loadTestsFromTestCase(
release_topic_case_ddt.ReleasdTopic)
alltests = TestSuite([suite1, suite2])
return alltests
def getTestsFromModule(module_path):
klass = getTestClassFromModule(module_path)
if klass is None:
raise UserWarning("No tests classes found in: '%s'" % module_path)
loader = TestLoader()
return loader.loadTestsFromTestCase(klass)
def run_unit_tests():
"""
Discover all unit test cases located in the test directory and run
"""
test_dir = os.path.dirname(__file__)
top_dir = os.path.dirname(os.path.dirname(test_dir))
test_loader = TestLoader()
run_unittest(test_loader.discover(test_dir, pattern="unit_test.py"))
def findTestsInDirectory(path, recursive=True , pattern = "test*.py"):
loader = TestLoader()
test_suite = loader.discover(path , pattern = pattern)
for (root, dirnames, filenames) in os.walk( path ):
for directory in dirnames:
test_suite.addTests(ErtTestRunner.findTestsInDirectory(os.path.join(root, directory), recursive , pattern))
return test_suite
def test_given(tests):
module = sys.modules[__name__]
if tests == None:
defaultTest = None
else:
loader = TestLoader()
defaultTest = TestSuite()
tests = loader.loadTestsFromNames(tests, module)
defaultTest.addTests(tests)
main(defaultTest=defaultTest)
def run(self, *args, **kwargs):
testloader = TestLoader()
testSuite = testloader.discover(self.test_suite,
top_level_dir='src/main/python')
test_result = ExtendedTeamcityTestRunner().run(testSuite)
test_result.printResults()
if test_result.wasSuccessful():
exit(0)
else:
exit(1)
def get_tests(d = '.'):
T = TestLoader()
t = T.discover(d)
stack = [t]
tests = []
while stack:
s = stack.pop()
for t in s:
if isinstance(t, TestSuite):
stack.append(t)
else:
tests.append(t)
return [test_name(v) for v in tests] # all test qualified names from this dir
def main():
# flake8
flake8 = get_style_guide(exclude=['.tox', 'build'])
report = flake8.check_files([BASE_DIR])
exit_code = print_report(report, flake8)
if exit_code > 0:
return exit_code
# unit tests
suite = TestLoader().discover(CODE_DIR, top_level_dir=BASE_DIR)
result = TextTestRunner().run(suite)
if result.wasSuccessful():
return 0
else:
return 1
def run(self):
# Force an in-place build for testing speedups
cmd = self.reinitialize_command('build_ext')
setattr(cmd, 'inplace', 1)
self.run_command('build_ext')
if sys.version_info[0:2] <= (2, 6):
try:
from unittest2 import TextTestRunner, TestLoader
except ImportError:
raise ImportError(
"unittest2 is required to run tests with python-%d.%d" %
sys.version_info[0:2])
else:
from unittest import TextTestRunner, TestLoader
import shapely.tests
tests = TestLoader().loadTestsFromName('test_suite', shapely.tests)
runner = TextTestRunner(verbosity=2)
result = runner.run(tests)
if not result.wasSuccessful():
sys.exit(1)
def get_all_tests(self):
tests = TestLoader().discover(self.testroot)
return [test for test in iterate_tests(tests)]
def runTestsInClass(classpath, test_verbosity=3):
klass = ErtTestRunner.importClass(classpath)
loader = TestLoader()
tests = loader.loadTestsFromTestCase(klass)
testRunner = TextTestRunner(verbosity=test_verbosity)
testRunner.run(tests)
try:
import mock
except ImportError:
raise ImportError("The BatchApps Python Client test suite requires "
"the mock package to run on Python 3.2 and below.\n"
"Please install this package to continue.")
try:
from teamcity import is_running_under_teamcity
from teamcity.unittestpy import TeamcityTestRunner
TC_BUILD = is_running_under_teamcity()
except ImportError:
TC_BUILD = False
if __name__ == '__main__':
if TC_BUILD:
runner = TeamcityTestRunner()
else:
runner = TextTestRunner(verbosity=2)
test_dir = os.path.dirname(__file__)
top_dir = os.path.dirname(os.path.dirname(test_dir))
test_loader = TestLoader()
suite = test_loader.discover(test_dir,
pattern="unittest_*.py",
top_level_dir=top_dir)
runner.run(suite)
def suite():
start_dir = os.path.dirname(__file__)
return TestLoader().discover(start_dir)
def test_suite():
from unittest2 import TestLoader, TestSuite
return TestLoader().loadTestsFromName(__name__)
def getTestsFromTestClass(test_class_path, argv=None):
klass = ErtTestRunner.importClass(test_class_path)
klass.argv = argv
loader = TestLoader()
return loader.loadTestsFromTestCase(klass)
#!/usr/bin/env python
import os
import sys
from unittest2 import TestLoader, TextTestRunner
if __name__ == '__main__':
tests_dir = os.path.dirname(os.path.abspath(__file__))
authorize_dir = os.path.join(tests_dir, os.path.pardir)
sys.path.append(authorize_dir)
suite = TestLoader().discover(tests_dir)
runner = TextTestRunner(verbosity=1)
result = runner.run(suite)
sys.exit(not result.wasSuccessful())
import os
import sys
from xmlrunner import XMLTestRunner
try:
from unittest2 import TestLoader, TextTestRunner
except ImportError:
from unittest import TestLoader, TextTestRunner
base_path = os.path.dirname(__file__)
parser = optparse.OptionParser()
parser.add_option("-x",
"--xml-prefix",
dest="xml_prefix",
help="generate XML test report in given directory",
default=None)
(options, args) = parser.parse_args()
loader = TestLoader()
tests = loader.discover(os.path.join(base_path, 'cvmfs/test'),
pattern='*_test.py')
runner = None
if options.xml_prefix:
runner = XMLTestRunner(output=options.xml_prefix, verbosity=2)
else:
runner = TextTestRunner(verbosity=2)
runner.run(tests)
