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
