def load_tests(loader, tests, pattern): for entry in pkg_resources.iter_entry_points('vanir.tests.extra'): try: for test_case in entry.load()(): tests.addTests( loader.loadTestsFromNames([ '{}.{}'.format(test_case.__module__, test_case.__name__) ])) except Exception as err: # pylint: disable=broad-except def runTest(self): raise err ExtraLoadFailure = type('ExtraLoadFailure', (vanir.tests.VanirTestCase, ), {entry.name: runTest}) tests.addTest(ExtraLoadFailure(entry.name)) for entry in pkg_resources.iter_entry_points( 'vanir.tests.extra.for_template'): try: for test_case in entry.load()(): tests.addTests( loader.loadTestsFromNames( vanir.tests.create_testcases_for_templates( test_case.__name__, test_case, module=sys.modules[test_case.__module__]))) except Exception as err: # pylint: disable=broad-except def runTest(self): raise err ExtraForTemplateLoadFailure = type('ExtraForTemplateLoadFailure', (vanir.tests.VanirTestCase, ), {entry.name: runTest}) tests.addTest(ExtraForTemplateLoadFailure(entry.name)) return tests
def load_tests(loader, tests, pattern): tests.addTests(loader.loadTestsFromNames(create_testcases_for_templates())) return tests