1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
#!/usr/bin/env python
"""Tests for distutils2.
The tests for distutils2 are defined in the distutils2.tests package.
"""
import sys
from os.path import dirname, islink, realpath, join, abspath
from optparse import OptionParser
COVERAGE_FILE = join(dirname(abspath(__file__)), '.coverage')
def get_coverage():
""" Return a usable coverage object. """
# deferred import because coverage is optional
import coverage
cov = getattr(coverage, "the_coverage", None)
if not cov:
cov = coverage.coverage(COVERAGE_FILE)
return cov
def ignore_prefixes(module):
""" Return a list of prefixes to ignore in the coverage report if
we want to completely skip `module`.
"""
# A function like that is needed because some GNU/Linux
# distributions, such a Ubuntu, really like to build link farm in
# /usr/lib in order to save a few bytes on the disk.
dirnames = [dirname(module.__file__)]
pymod = module.__file__.rstrip("c")
if islink(pymod):
dirnames.append(dirname(realpath(pymod)))
return dirnames
def parse_opts():
parser = OptionParser(usage="%prog [OPTIONS]",
description="run the distutils2 unittests")
parser.add_option("-q", "--quiet", help="do not print verbose messages",
action="store_true", default=False)
parser.add_option("-c", "--coverage", action="store_true", default=False,
help="produce a coverage report at the end of the run")
parser.add_option("-r", "--report", action="store_true", default=False,
help="produce a coverage report from the last test run")
parser.add_option("-m", "--show-missing", action="store_true",
default=False,
help=("Show line numbers of statements in each module "
"that weren't executed."))
opts, args = parser.parse_args()
return opts, args
def coverage_report(opts):
from distutils2.tests.support import unittest
cov = get_coverage()
if hasattr(cov, "load"):
# running coverage 3.x
cov.load()
morfs = None
else:
# running coverage 2.x
cov.cache = COVERAGE_FILE
cov.restore()
morfs = [m for m in list(cov.cexecuted.keys()) if "distutils2" in m]
prefixes = ["runtests", "distutils2/tests", "distutils2/_backport"]
prefixes += ignore_prefixes(unittest)
try:
import docutils
prefixes += ignore_prefixes(docutils)
except ImportError:
# that module is completely optional
pass
try:
import roman
prefixes += ignore_prefixes(roman)
except ImportError:
# that module is also completely optional
pass
try:
cov.report(morfs,
omit_prefixes=prefixes,
show_missing=opts.show_missing)
except TypeError:
# Coverage 3.4 turned `omit_prefixes` into a list of globbing patterns
cov.report(morfs,
omit=[p + "*" for p in prefixes],
show_missing=opts.show_missing)
def test_main():
opts, args = parse_opts()
verbose = not opts.quiet
ret = 0
if opts.coverage:
cov = get_coverage()
cov.erase()
cov.start()
if not opts.report:
ret = run_tests(verbose)
if opts.coverage:
cov.stop()
cov.save()
if opts.report or opts.coverage:
coverage_report(opts)
return ret
def run_tests(verbose):
# do NOT import those at the top level, coverage will be inaccurate if
# modules are imported before its magic is started
from distutils2.tests import run_unittest, test_suite, reap_children, TestFailed
from distutils2._backport.tests import test_suite as btest_suite
try:
try:
run_unittest([test_suite(), btest_suite()], verbose_=verbose)
return 0
except TestFailed:
return 1
finally:
reap_children()
if __name__ == "__main__":
if sys.version < '2.5':
try:
from distutils2._backport import hashlib
except ImportError:
import subprocess
subprocess.call([sys.executable, 'setup.py', 'build_ext'])
sys.exit(test_main())
|
