| 1 | n/a | """Run the project's test suite.""" |
|---|
| 2 | n/a | |
|---|
| 3 | n/a | import os |
|---|
| 4 | n/a | import sys |
|---|
| 5 | n/a | import logging |
|---|
| 6 | n/a | import unittest |
|---|
| 7 | n/a | |
|---|
| 8 | n/a | from packaging import logger |
|---|
| 9 | n/a | from packaging.command.cmd import Command |
|---|
| 10 | n/a | from packaging.database import get_distribution |
|---|
| 11 | n/a | from packaging.errors import PackagingOptionError |
|---|
| 12 | n/a | from packaging.util import resolve_name |
|---|
| 13 | n/a | |
|---|
| 14 | n/a | |
|---|
| 15 | n/a | class test(Command): |
|---|
| 16 | n/a | |
|---|
| 17 | n/a | description = "run the project's test suite" |
|---|
| 18 | n/a | |
|---|
| 19 | n/a | user_options = [ |
|---|
| 20 | n/a | ('suite=', 's', |
|---|
| 21 | n/a | "test suite to run (for example: 'some_module.test_suite')"), |
|---|
| 22 | n/a | ('runner=', None, |
|---|
| 23 | n/a | "test runner to be called."), |
|---|
| 24 | n/a | ('tests-require=', None, |
|---|
| 25 | n/a | "list of distributions required to run the test suite."), |
|---|
| 26 | n/a | ] |
|---|
| 27 | n/a | |
|---|
| 28 | n/a | def initialize_options(self): |
|---|
| 29 | n/a | self.suite = None |
|---|
| 30 | n/a | self.runner = None |
|---|
| 31 | n/a | self.tests_require = [] |
|---|
| 32 | n/a | |
|---|
| 33 | n/a | def finalize_options(self): |
|---|
| 34 | n/a | self.build_lib = self.get_finalized_command("build").build_lib |
|---|
| 35 | n/a | for requirement in self.tests_require: |
|---|
| 36 | n/a | if get_distribution(requirement) is None: |
|---|
| 37 | n/a | logger.warning("test dependency %s is not installed, " |
|---|
| 38 | n/a | "tests may fail", requirement) |
|---|
| 39 | n/a | if (not self.suite and not self.runner and |
|---|
| 40 | n/a | self.get_ut_with_discovery() is None): |
|---|
| 41 | n/a | raise PackagingOptionError( |
|---|
| 42 | n/a | "no test discovery available, please give a 'suite' or " |
|---|
| 43 | n/a | "'runner' option or install unittest2") |
|---|
| 44 | n/a | |
|---|
| 45 | n/a | def get_ut_with_discovery(self): |
|---|
| 46 | n/a | if hasattr(unittest.TestLoader, "discover"): |
|---|
| 47 | n/a | return unittest |
|---|
| 48 | n/a | else: |
|---|
| 49 | n/a | try: |
|---|
| 50 | n/a | import unittest2 |
|---|
| 51 | n/a | return unittest2 |
|---|
| 52 | n/a | except ImportError: |
|---|
| 53 | n/a | return None |
|---|
| 54 | n/a | |
|---|
| 55 | n/a | def run(self): |
|---|
| 56 | n/a | prev_syspath = sys.path[:] |
|---|
| 57 | n/a | try: |
|---|
| 58 | n/a | # build release |
|---|
| 59 | n/a | build = self.reinitialize_command('build') |
|---|
| 60 | n/a | self.run_command('build') |
|---|
| 61 | n/a | sys.path.insert(0, build.build_lib) |
|---|
| 62 | n/a | |
|---|
| 63 | n/a | # XXX maybe we could pass the verbose argument of pysetup here |
|---|
| 64 | n/a | logger = logging.getLogger('packaging') |
|---|
| 65 | n/a | verbose = logger.getEffectiveLevel() >= logging.DEBUG |
|---|
| 66 | n/a | verbosity = verbose + 1 |
|---|
| 67 | n/a | |
|---|
| 68 | n/a | # run the tests |
|---|
| 69 | n/a | if self.runner: |
|---|
| 70 | n/a | resolve_name(self.runner)() |
|---|
| 71 | n/a | elif self.suite: |
|---|
| 72 | n/a | runner = unittest.TextTestRunner(verbosity=verbosity) |
|---|
| 73 | n/a | runner.run(resolve_name(self.suite)()) |
|---|
| 74 | n/a | elif self.get_ut_with_discovery(): |
|---|
| 75 | n/a | ut = self.get_ut_with_discovery() |
|---|
| 76 | n/a | test_suite = ut.TestLoader().discover(os.curdir) |
|---|
| 77 | n/a | runner = ut.TextTestRunner(verbosity=verbosity) |
|---|
| 78 | n/a | runner.run(test_suite) |
|---|
| 79 | n/a | finally: |
|---|
| 80 | n/a | sys.path[:] = prev_syspath |
|---|
