mirror of
https://github.com/python/cpython.git
synced 2025-10-30 21:21:22 +00:00
119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
from setuptools import Command
|
|
from distutils.errors import DistutilsOptionError
|
|
import sys
|
|
from pkg_resources import *
|
|
from unittest import TestLoader, main
|
|
|
|
class ScanningLoader(TestLoader):
|
|
|
|
def loadTestsFromModule(self, module):
|
|
"""Return a suite of all tests cases contained in the given module
|
|
|
|
If the module is a package, load tests from all the modules in it.
|
|
If the module has an ``additional_tests`` function, call it and add
|
|
the return value to the tests.
|
|
"""
|
|
tests = []
|
|
if module.__name__!='setuptools.tests.doctest': # ugh
|
|
tests.append(TestLoader.loadTestsFromModule(self,module))
|
|
|
|
if hasattr(module, "additional_tests"):
|
|
tests.append(module.additional_tests())
|
|
|
|
if hasattr(module, '__path__'):
|
|
for file in resource_listdir(module.__name__, ''):
|
|
if file.endswith('.py') and file!='__init__.py':
|
|
submodule = module.__name__+'.'+file[:-3]
|
|
else:
|
|
if resource_exists(
|
|
module.__name__, file+'/__init__.py'
|
|
):
|
|
submodule = module.__name__+'.'+file
|
|
else:
|
|
continue
|
|
tests.append(self.loadTestsFromName(submodule))
|
|
|
|
if len(tests)!=1:
|
|
return self.suiteClass(tests)
|
|
else:
|
|
return tests[0] # don't create a nested suite for only one return
|
|
|
|
|
|
class test(Command):
|
|
|
|
"""Command to run unit tests after in-place build"""
|
|
|
|
description = "run unit tests after in-place build"
|
|
|
|
user_options = [
|
|
('test-module=','m', "Run 'test_suite' in specified module"),
|
|
('test-suite=','s',
|
|
"Test suite to run (e.g. 'some_module.test_suite')"),
|
|
]
|
|
|
|
def initialize_options(self):
|
|
self.test_suite = None
|
|
self.test_module = None
|
|
self.test_loader = None
|
|
|
|
|
|
def finalize_options(self):
|
|
|
|
if self.test_suite is None:
|
|
if self.test_module is None:
|
|
self.test_suite = self.distribution.test_suite
|
|
else:
|
|
self.test_suite = self.test_module+".test_suite"
|
|
elif self.test_module:
|
|
raise DistutilsOptionError(
|
|
"You may specify a module or a suite, but not both"
|
|
)
|
|
|
|
self.test_args = [self.test_suite]
|
|
|
|
if self.verbose:
|
|
self.test_args.insert(0,'--verbose')
|
|
if self.test_loader is None:
|
|
self.test_loader = getattr(self.distribution,'test_loader',None)
|
|
if self.test_loader is None:
|
|
self.test_loader = "setuptools.command.test:ScanningLoader"
|
|
|
|
|
|
|
|
def run(self):
|
|
# Ensure metadata is up-to-date
|
|
self.run_command('egg_info')
|
|
|
|
# Build extensions in-place
|
|
self.reinitialize_command('build_ext', inplace=1)
|
|
self.run_command('build_ext')
|
|
|
|
if self.distribution.tests_require:
|
|
self.distribution.fetch_build_eggs(self.distribution.tests_require)
|
|
|
|
if self.test_suite:
|
|
cmd = ' '.join(self.test_args)
|
|
if self.dry_run:
|
|
self.announce('skipping "unittest %s" (dry run)' % cmd)
|
|
else:
|
|
self.announce('running "unittest %s"' % cmd)
|
|
self.run_tests()
|
|
|
|
|
|
def run_tests(self):
|
|
import unittest
|
|
old_path = sys.path[:]
|
|
ei_cmd = self.get_finalized_command("egg_info")
|
|
path_item = normalize_path(ei_cmd.egg_base)
|
|
metadata = PathMetadata(
|
|
path_item, normalize_path(ei_cmd.egg_info)
|
|
)
|
|
dist = Distribution(path_item, metadata, project_name=ei_cmd.egg_name)
|
|
working_set.add(dist)
|
|
require(str(dist.as_requirement()))
|
|
loader_ep = EntryPoint.parse("x="+self.test_loader)
|
|
loader_class = loader_ep.load(require=False)
|
|
unittest.main(
|
|
None, None, [unittest.__file__]+self.test_args,
|
|
testLoader = loader_class()
|
|
)
|