##// END OF EJS Templates
run-tests: accept '\' vs '/' path differences without '(glob)'...
Matt Harbison -
r35383:dfae1435 default
parent child Browse files
Show More
@@ -1,3019 +1,3016
1 #!/usr/bin/env python
1 #!/usr/bin/env python
2 #
2 #
3 # run-tests.py - Run a set of tests on Mercurial
3 # run-tests.py - Run a set of tests on Mercurial
4 #
4 #
5 # Copyright 2006 Matt Mackall <mpm@selenic.com>
5 # Copyright 2006 Matt Mackall <mpm@selenic.com>
6 #
6 #
7 # This software may be used and distributed according to the terms of the
7 # This software may be used and distributed according to the terms of the
8 # GNU General Public License version 2 or any later version.
8 # GNU General Public License version 2 or any later version.
9
9
10 # Modifying this script is tricky because it has many modes:
10 # Modifying this script is tricky because it has many modes:
11 # - serial (default) vs parallel (-jN, N > 1)
11 # - serial (default) vs parallel (-jN, N > 1)
12 # - no coverage (default) vs coverage (-c, -C, -s)
12 # - no coverage (default) vs coverage (-c, -C, -s)
13 # - temp install (default) vs specific hg script (--with-hg, --local)
13 # - temp install (default) vs specific hg script (--with-hg, --local)
14 # - tests are a mix of shell scripts and Python scripts
14 # - tests are a mix of shell scripts and Python scripts
15 #
15 #
16 # If you change this script, it is recommended that you ensure you
16 # If you change this script, it is recommended that you ensure you
17 # haven't broken it by running it in various modes with a representative
17 # haven't broken it by running it in various modes with a representative
18 # sample of test scripts. For example:
18 # sample of test scripts. For example:
19 #
19 #
20 # 1) serial, no coverage, temp install:
20 # 1) serial, no coverage, temp install:
21 # ./run-tests.py test-s*
21 # ./run-tests.py test-s*
22 # 2) serial, no coverage, local hg:
22 # 2) serial, no coverage, local hg:
23 # ./run-tests.py --local test-s*
23 # ./run-tests.py --local test-s*
24 # 3) serial, coverage, temp install:
24 # 3) serial, coverage, temp install:
25 # ./run-tests.py -c test-s*
25 # ./run-tests.py -c test-s*
26 # 4) serial, coverage, local hg:
26 # 4) serial, coverage, local hg:
27 # ./run-tests.py -c --local test-s* # unsupported
27 # ./run-tests.py -c --local test-s* # unsupported
28 # 5) parallel, no coverage, temp install:
28 # 5) parallel, no coverage, temp install:
29 # ./run-tests.py -j2 test-s*
29 # ./run-tests.py -j2 test-s*
30 # 6) parallel, no coverage, local hg:
30 # 6) parallel, no coverage, local hg:
31 # ./run-tests.py -j2 --local test-s*
31 # ./run-tests.py -j2 --local test-s*
32 # 7) parallel, coverage, temp install:
32 # 7) parallel, coverage, temp install:
33 # ./run-tests.py -j2 -c test-s* # currently broken
33 # ./run-tests.py -j2 -c test-s* # currently broken
34 # 8) parallel, coverage, local install:
34 # 8) parallel, coverage, local install:
35 # ./run-tests.py -j2 -c --local test-s* # unsupported (and broken)
35 # ./run-tests.py -j2 -c --local test-s* # unsupported (and broken)
36 # 9) parallel, custom tmp dir:
36 # 9) parallel, custom tmp dir:
37 # ./run-tests.py -j2 --tmpdir /tmp/myhgtests
37 # ./run-tests.py -j2 --tmpdir /tmp/myhgtests
38 # 10) parallel, pure, tests that call run-tests:
38 # 10) parallel, pure, tests that call run-tests:
39 # ./run-tests.py --pure `grep -l run-tests.py *.t`
39 # ./run-tests.py --pure `grep -l run-tests.py *.t`
40 #
40 #
41 # (You could use any subset of the tests: test-s* happens to match
41 # (You could use any subset of the tests: test-s* happens to match
42 # enough that it's worth doing parallel runs, few enough that it
42 # enough that it's worth doing parallel runs, few enough that it
43 # completes fairly quickly, includes both shell and Python scripts, and
43 # completes fairly quickly, includes both shell and Python scripts, and
44 # includes some scripts that run daemon processes.)
44 # includes some scripts that run daemon processes.)
45
45
46 from __future__ import absolute_import, print_function
46 from __future__ import absolute_import, print_function
47
47
48 import argparse
48 import argparse
49 import collections
49 import collections
50 import difflib
50 import difflib
51 import distutils.version as version
51 import distutils.version as version
52 import errno
52 import errno
53 import json
53 import json
54 import os
54 import os
55 import random
55 import random
56 import re
56 import re
57 import shutil
57 import shutil
58 import signal
58 import signal
59 import socket
59 import socket
60 import subprocess
60 import subprocess
61 import sys
61 import sys
62 import sysconfig
62 import sysconfig
63 import tempfile
63 import tempfile
64 import threading
64 import threading
65 import time
65 import time
66 import unittest
66 import unittest
67 import xml.dom.minidom as minidom
67 import xml.dom.minidom as minidom
68
68
69 try:
69 try:
70 import Queue as queue
70 import Queue as queue
71 except ImportError:
71 except ImportError:
72 import queue
72 import queue
73
73
74 try:
74 try:
75 import shlex
75 import shlex
76 shellquote = shlex.quote
76 shellquote = shlex.quote
77 except (ImportError, AttributeError):
77 except (ImportError, AttributeError):
78 import pipes
78 import pipes
79 shellquote = pipes.quote
79 shellquote = pipes.quote
80
80
81 if os.environ.get('RTUNICODEPEDANTRY', False):
81 if os.environ.get('RTUNICODEPEDANTRY', False):
82 try:
82 try:
83 reload(sys)
83 reload(sys)
84 sys.setdefaultencoding("undefined")
84 sys.setdefaultencoding("undefined")
85 except NameError:
85 except NameError:
86 pass
86 pass
87
87
88 origenviron = os.environ.copy()
88 origenviron = os.environ.copy()
89 osenvironb = getattr(os, 'environb', os.environ)
89 osenvironb = getattr(os, 'environb', os.environ)
90 processlock = threading.Lock()
90 processlock = threading.Lock()
91
91
92 pygmentspresent = False
92 pygmentspresent = False
93 # ANSI color is unsupported prior to Windows 10
93 # ANSI color is unsupported prior to Windows 10
94 if os.name != 'nt':
94 if os.name != 'nt':
95 try: # is pygments installed
95 try: # is pygments installed
96 import pygments
96 import pygments
97 import pygments.lexers as lexers
97 import pygments.lexers as lexers
98 import pygments.lexer as lexer
98 import pygments.lexer as lexer
99 import pygments.formatters as formatters
99 import pygments.formatters as formatters
100 import pygments.token as token
100 import pygments.token as token
101 import pygments.style as style
101 import pygments.style as style
102 pygmentspresent = True
102 pygmentspresent = True
103 difflexer = lexers.DiffLexer()
103 difflexer = lexers.DiffLexer()
104 terminal256formatter = formatters.Terminal256Formatter()
104 terminal256formatter = formatters.Terminal256Formatter()
105 except ImportError:
105 except ImportError:
106 pass
106 pass
107
107
108 if pygmentspresent:
108 if pygmentspresent:
109 class TestRunnerStyle(style.Style):
109 class TestRunnerStyle(style.Style):
110 default_style = ""
110 default_style = ""
111 skipped = token.string_to_tokentype("Token.Generic.Skipped")
111 skipped = token.string_to_tokentype("Token.Generic.Skipped")
112 failed = token.string_to_tokentype("Token.Generic.Failed")
112 failed = token.string_to_tokentype("Token.Generic.Failed")
113 skippedname = token.string_to_tokentype("Token.Generic.SName")
113 skippedname = token.string_to_tokentype("Token.Generic.SName")
114 failedname = token.string_to_tokentype("Token.Generic.FName")
114 failedname = token.string_to_tokentype("Token.Generic.FName")
115 styles = {
115 styles = {
116 skipped: '#e5e5e5',
116 skipped: '#e5e5e5',
117 skippedname: '#00ffff',
117 skippedname: '#00ffff',
118 failed: '#7f0000',
118 failed: '#7f0000',
119 failedname: '#ff0000',
119 failedname: '#ff0000',
120 }
120 }
121
121
122 class TestRunnerLexer(lexer.RegexLexer):
122 class TestRunnerLexer(lexer.RegexLexer):
123 tokens = {
123 tokens = {
124 'root': [
124 'root': [
125 (r'^Skipped', token.Generic.Skipped, 'skipped'),
125 (r'^Skipped', token.Generic.Skipped, 'skipped'),
126 (r'^Failed ', token.Generic.Failed, 'failed'),
126 (r'^Failed ', token.Generic.Failed, 'failed'),
127 (r'^ERROR: ', token.Generic.Failed, 'failed'),
127 (r'^ERROR: ', token.Generic.Failed, 'failed'),
128 ],
128 ],
129 'skipped': [
129 'skipped': [
130 (r'[\w-]+\.(t|py)', token.Generic.SName),
130 (r'[\w-]+\.(t|py)', token.Generic.SName),
131 (r':.*', token.Generic.Skipped),
131 (r':.*', token.Generic.Skipped),
132 ],
132 ],
133 'failed': [
133 'failed': [
134 (r'[\w-]+\.(t|py)', token.Generic.FName),
134 (r'[\w-]+\.(t|py)', token.Generic.FName),
135 (r'(:| ).*', token.Generic.Failed),
135 (r'(:| ).*', token.Generic.Failed),
136 ]
136 ]
137 }
137 }
138
138
139 runnerformatter = formatters.Terminal256Formatter(style=TestRunnerStyle)
139 runnerformatter = formatters.Terminal256Formatter(style=TestRunnerStyle)
140 runnerlexer = TestRunnerLexer()
140 runnerlexer = TestRunnerLexer()
141
141
142 if sys.version_info > (3, 5, 0):
142 if sys.version_info > (3, 5, 0):
143 PYTHON3 = True
143 PYTHON3 = True
144 xrange = range # we use xrange in one place, and we'd rather not use range
144 xrange = range # we use xrange in one place, and we'd rather not use range
145 def _bytespath(p):
145 def _bytespath(p):
146 if p is None:
146 if p is None:
147 return p
147 return p
148 return p.encode('utf-8')
148 return p.encode('utf-8')
149
149
150 def _strpath(p):
150 def _strpath(p):
151 if p is None:
151 if p is None:
152 return p
152 return p
153 return p.decode('utf-8')
153 return p.decode('utf-8')
154
154
155 elif sys.version_info >= (3, 0, 0):
155 elif sys.version_info >= (3, 0, 0):
156 print('%s is only supported on Python 3.5+ and 2.7, not %s' %
156 print('%s is only supported on Python 3.5+ and 2.7, not %s' %
157 (sys.argv[0], '.'.join(str(v) for v in sys.version_info[:3])))
157 (sys.argv[0], '.'.join(str(v) for v in sys.version_info[:3])))
158 sys.exit(70) # EX_SOFTWARE from `man 3 sysexit`
158 sys.exit(70) # EX_SOFTWARE from `man 3 sysexit`
159 else:
159 else:
160 PYTHON3 = False
160 PYTHON3 = False
161
161
162 # In python 2.x, path operations are generally done using
162 # In python 2.x, path operations are generally done using
163 # bytestrings by default, so we don't have to do any extra
163 # bytestrings by default, so we don't have to do any extra
164 # fiddling there. We define the wrapper functions anyway just to
164 # fiddling there. We define the wrapper functions anyway just to
165 # help keep code consistent between platforms.
165 # help keep code consistent between platforms.
166 def _bytespath(p):
166 def _bytespath(p):
167 return p
167 return p
168
168
169 _strpath = _bytespath
169 _strpath = _bytespath
170
170
171 # For Windows support
171 # For Windows support
172 wifexited = getattr(os, "WIFEXITED", lambda x: False)
172 wifexited = getattr(os, "WIFEXITED", lambda x: False)
173
173
174 # Whether to use IPv6
174 # Whether to use IPv6
175 def checksocketfamily(name, port=20058):
175 def checksocketfamily(name, port=20058):
176 """return true if we can listen on localhost using family=name
176 """return true if we can listen on localhost using family=name
177
177
178 name should be either 'AF_INET', or 'AF_INET6'.
178 name should be either 'AF_INET', or 'AF_INET6'.
179 port being used is okay - EADDRINUSE is considered as successful.
179 port being used is okay - EADDRINUSE is considered as successful.
180 """
180 """
181 family = getattr(socket, name, None)
181 family = getattr(socket, name, None)
182 if family is None:
182 if family is None:
183 return False
183 return False
184 try:
184 try:
185 s = socket.socket(family, socket.SOCK_STREAM)
185 s = socket.socket(family, socket.SOCK_STREAM)
186 s.bind(('localhost', port))
186 s.bind(('localhost', port))
187 s.close()
187 s.close()
188 return True
188 return True
189 except socket.error as exc:
189 except socket.error as exc:
190 if exc.errno == errno.EADDRINUSE:
190 if exc.errno == errno.EADDRINUSE:
191 return True
191 return True
192 elif exc.errno in (errno.EADDRNOTAVAIL, errno.EPROTONOSUPPORT):
192 elif exc.errno in (errno.EADDRNOTAVAIL, errno.EPROTONOSUPPORT):
193 return False
193 return False
194 else:
194 else:
195 raise
195 raise
196 else:
196 else:
197 return False
197 return False
198
198
199 # useipv6 will be set by parseargs
199 # useipv6 will be set by parseargs
200 useipv6 = None
200 useipv6 = None
201
201
202 def checkportisavailable(port):
202 def checkportisavailable(port):
203 """return true if a port seems free to bind on localhost"""
203 """return true if a port seems free to bind on localhost"""
204 if useipv6:
204 if useipv6:
205 family = socket.AF_INET6
205 family = socket.AF_INET6
206 else:
206 else:
207 family = socket.AF_INET
207 family = socket.AF_INET
208 try:
208 try:
209 s = socket.socket(family, socket.SOCK_STREAM)
209 s = socket.socket(family, socket.SOCK_STREAM)
210 s.bind(('localhost', port))
210 s.bind(('localhost', port))
211 s.close()
211 s.close()
212 return True
212 return True
213 except socket.error as exc:
213 except socket.error as exc:
214 if exc.errno not in (errno.EADDRINUSE, errno.EADDRNOTAVAIL,
214 if exc.errno not in (errno.EADDRINUSE, errno.EADDRNOTAVAIL,
215 errno.EPROTONOSUPPORT):
215 errno.EPROTONOSUPPORT):
216 raise
216 raise
217 return False
217 return False
218
218
219 closefds = os.name == 'posix'
219 closefds = os.name == 'posix'
220 def Popen4(cmd, wd, timeout, env=None):
220 def Popen4(cmd, wd, timeout, env=None):
221 processlock.acquire()
221 processlock.acquire()
222 p = subprocess.Popen(cmd, shell=True, bufsize=-1, cwd=wd, env=env,
222 p = subprocess.Popen(cmd, shell=True, bufsize=-1, cwd=wd, env=env,
223 close_fds=closefds,
223 close_fds=closefds,
224 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
224 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
225 stderr=subprocess.STDOUT)
225 stderr=subprocess.STDOUT)
226 processlock.release()
226 processlock.release()
227
227
228 p.fromchild = p.stdout
228 p.fromchild = p.stdout
229 p.tochild = p.stdin
229 p.tochild = p.stdin
230 p.childerr = p.stderr
230 p.childerr = p.stderr
231
231
232 p.timeout = False
232 p.timeout = False
233 if timeout:
233 if timeout:
234 def t():
234 def t():
235 start = time.time()
235 start = time.time()
236 while time.time() - start < timeout and p.returncode is None:
236 while time.time() - start < timeout and p.returncode is None:
237 time.sleep(.1)
237 time.sleep(.1)
238 p.timeout = True
238 p.timeout = True
239 if p.returncode is None:
239 if p.returncode is None:
240 terminate(p)
240 terminate(p)
241 threading.Thread(target=t).start()
241 threading.Thread(target=t).start()
242
242
243 return p
243 return p
244
244
245 PYTHON = _bytespath(sys.executable.replace('\\', '/'))
245 PYTHON = _bytespath(sys.executable.replace('\\', '/'))
246 IMPL_PATH = b'PYTHONPATH'
246 IMPL_PATH = b'PYTHONPATH'
247 if 'java' in sys.platform:
247 if 'java' in sys.platform:
248 IMPL_PATH = b'JYTHONPATH'
248 IMPL_PATH = b'JYTHONPATH'
249
249
250 defaults = {
250 defaults = {
251 'jobs': ('HGTEST_JOBS', 1),
251 'jobs': ('HGTEST_JOBS', 1),
252 'timeout': ('HGTEST_TIMEOUT', 180),
252 'timeout': ('HGTEST_TIMEOUT', 180),
253 'slowtimeout': ('HGTEST_SLOWTIMEOUT', 500),
253 'slowtimeout': ('HGTEST_SLOWTIMEOUT', 500),
254 'port': ('HGTEST_PORT', 20059),
254 'port': ('HGTEST_PORT', 20059),
255 'shell': ('HGTEST_SHELL', 'sh'),
255 'shell': ('HGTEST_SHELL', 'sh'),
256 }
256 }
257
257
258 def canonpath(path):
258 def canonpath(path):
259 return os.path.realpath(os.path.expanduser(path))
259 return os.path.realpath(os.path.expanduser(path))
260
260
261 def parselistfiles(files, listtype, warn=True):
261 def parselistfiles(files, listtype, warn=True):
262 entries = dict()
262 entries = dict()
263 for filename in files:
263 for filename in files:
264 try:
264 try:
265 path = os.path.expanduser(os.path.expandvars(filename))
265 path = os.path.expanduser(os.path.expandvars(filename))
266 f = open(path, "rb")
266 f = open(path, "rb")
267 except IOError as err:
267 except IOError as err:
268 if err.errno != errno.ENOENT:
268 if err.errno != errno.ENOENT:
269 raise
269 raise
270 if warn:
270 if warn:
271 print("warning: no such %s file: %s" % (listtype, filename))
271 print("warning: no such %s file: %s" % (listtype, filename))
272 continue
272 continue
273
273
274 for line in f.readlines():
274 for line in f.readlines():
275 line = line.split(b'#', 1)[0].strip()
275 line = line.split(b'#', 1)[0].strip()
276 if line:
276 if line:
277 entries[line] = filename
277 entries[line] = filename
278
278
279 f.close()
279 f.close()
280 return entries
280 return entries
281
281
282 def parsettestcases(path):
282 def parsettestcases(path):
283 """read a .t test file, return a set of test case names
283 """read a .t test file, return a set of test case names
284
284
285 If path does not exist, return an empty set.
285 If path does not exist, return an empty set.
286 """
286 """
287 cases = set()
287 cases = set()
288 try:
288 try:
289 with open(path, 'rb') as f:
289 with open(path, 'rb') as f:
290 for l in f:
290 for l in f:
291 if l.startswith(b'#testcases '):
291 if l.startswith(b'#testcases '):
292 cases.update(l[11:].split())
292 cases.update(l[11:].split())
293 except IOError as ex:
293 except IOError as ex:
294 if ex.errno != errno.ENOENT:
294 if ex.errno != errno.ENOENT:
295 raise
295 raise
296 return cases
296 return cases
297
297
298 def getparser():
298 def getparser():
299 """Obtain the OptionParser used by the CLI."""
299 """Obtain the OptionParser used by the CLI."""
300 parser = argparse.ArgumentParser(usage='%(prog)s [options] [tests]')
300 parser = argparse.ArgumentParser(usage='%(prog)s [options] [tests]')
301
301
302 selection = parser.add_argument_group('Test Selection')
302 selection = parser.add_argument_group('Test Selection')
303 selection.add_argument('--allow-slow-tests', action='store_true',
303 selection.add_argument('--allow-slow-tests', action='store_true',
304 help='allow extremely slow tests')
304 help='allow extremely slow tests')
305 selection.add_argument("--blacklist", action="append",
305 selection.add_argument("--blacklist", action="append",
306 help="skip tests listed in the specified blacklist file")
306 help="skip tests listed in the specified blacklist file")
307 selection.add_argument("--changed",
307 selection.add_argument("--changed",
308 help="run tests that are changed in parent rev or working directory")
308 help="run tests that are changed in parent rev or working directory")
309 selection.add_argument("-k", "--keywords",
309 selection.add_argument("-k", "--keywords",
310 help="run tests matching keywords")
310 help="run tests matching keywords")
311 selection.add_argument("-r", "--retest", action="store_true",
311 selection.add_argument("-r", "--retest", action="store_true",
312 help = "retest failed tests")
312 help = "retest failed tests")
313 selection.add_argument("--test-list", action="append",
313 selection.add_argument("--test-list", action="append",
314 help="read tests to run from the specified file")
314 help="read tests to run from the specified file")
315 selection.add_argument("--whitelist", action="append",
315 selection.add_argument("--whitelist", action="append",
316 help="always run tests listed in the specified whitelist file")
316 help="always run tests listed in the specified whitelist file")
317 selection.add_argument('tests', metavar='TESTS', nargs='*',
317 selection.add_argument('tests', metavar='TESTS', nargs='*',
318 help='Tests to run')
318 help='Tests to run')
319
319
320 harness = parser.add_argument_group('Test Harness Behavior')
320 harness = parser.add_argument_group('Test Harness Behavior')
321 harness.add_argument('--bisect-repo',
321 harness.add_argument('--bisect-repo',
322 metavar='bisect_repo',
322 metavar='bisect_repo',
323 help=("Path of a repo to bisect. Use together with "
323 help=("Path of a repo to bisect. Use together with "
324 "--known-good-rev"))
324 "--known-good-rev"))
325 harness.add_argument("-d", "--debug", action="store_true",
325 harness.add_argument("-d", "--debug", action="store_true",
326 help="debug mode: write output of test scripts to console"
326 help="debug mode: write output of test scripts to console"
327 " rather than capturing and diffing it (disables timeout)")
327 " rather than capturing and diffing it (disables timeout)")
328 harness.add_argument("-f", "--first", action="store_true",
328 harness.add_argument("-f", "--first", action="store_true",
329 help="exit on the first test failure")
329 help="exit on the first test failure")
330 harness.add_argument("-i", "--interactive", action="store_true",
330 harness.add_argument("-i", "--interactive", action="store_true",
331 help="prompt to accept changed output")
331 help="prompt to accept changed output")
332 harness.add_argument("-j", "--jobs", type=int,
332 harness.add_argument("-j", "--jobs", type=int,
333 help="number of jobs to run in parallel"
333 help="number of jobs to run in parallel"
334 " (default: $%s or %d)" % defaults['jobs'])
334 " (default: $%s or %d)" % defaults['jobs'])
335 harness.add_argument("--keep-tmpdir", action="store_true",
335 harness.add_argument("--keep-tmpdir", action="store_true",
336 help="keep temporary directory after running tests")
336 help="keep temporary directory after running tests")
337 harness.add_argument('--known-good-rev',
337 harness.add_argument('--known-good-rev',
338 metavar="known_good_rev",
338 metavar="known_good_rev",
339 help=("Automatically bisect any failures using this "
339 help=("Automatically bisect any failures using this "
340 "revision as a known-good revision."))
340 "revision as a known-good revision."))
341 harness.add_argument("--list-tests", action="store_true",
341 harness.add_argument("--list-tests", action="store_true",
342 help="list tests instead of running them")
342 help="list tests instead of running them")
343 harness.add_argument("--loop", action="store_true",
343 harness.add_argument("--loop", action="store_true",
344 help="loop tests repeatedly")
344 help="loop tests repeatedly")
345 harness.add_argument('--random', action="store_true",
345 harness.add_argument('--random', action="store_true",
346 help='run tests in random order')
346 help='run tests in random order')
347 harness.add_argument("-p", "--port", type=int,
347 harness.add_argument("-p", "--port", type=int,
348 help="port on which servers should listen"
348 help="port on which servers should listen"
349 " (default: $%s or %d)" % defaults['port'])
349 " (default: $%s or %d)" % defaults['port'])
350 harness.add_argument('--profile-runner', action='store_true',
350 harness.add_argument('--profile-runner', action='store_true',
351 help='run statprof on run-tests')
351 help='run statprof on run-tests')
352 harness.add_argument("-R", "--restart", action="store_true",
352 harness.add_argument("-R", "--restart", action="store_true",
353 help="restart at last error")
353 help="restart at last error")
354 harness.add_argument("--runs-per-test", type=int, dest="runs_per_test",
354 harness.add_argument("--runs-per-test", type=int, dest="runs_per_test",
355 help="run each test N times (default=1)", default=1)
355 help="run each test N times (default=1)", default=1)
356 harness.add_argument("--shell",
356 harness.add_argument("--shell",
357 help="shell to use (default: $%s or %s)" % defaults['shell'])
357 help="shell to use (default: $%s or %s)" % defaults['shell'])
358 harness.add_argument('--showchannels', action='store_true',
358 harness.add_argument('--showchannels', action='store_true',
359 help='show scheduling channels')
359 help='show scheduling channels')
360 harness.add_argument("--slowtimeout", type=int,
360 harness.add_argument("--slowtimeout", type=int,
361 help="kill errant slow tests after SLOWTIMEOUT seconds"
361 help="kill errant slow tests after SLOWTIMEOUT seconds"
362 " (default: $%s or %d)" % defaults['slowtimeout'])
362 " (default: $%s or %d)" % defaults['slowtimeout'])
363 harness.add_argument("-t", "--timeout", type=int,
363 harness.add_argument("-t", "--timeout", type=int,
364 help="kill errant tests after TIMEOUT seconds"
364 help="kill errant tests after TIMEOUT seconds"
365 " (default: $%s or %d)" % defaults['timeout'])
365 " (default: $%s or %d)" % defaults['timeout'])
366 harness.add_argument("--tmpdir",
366 harness.add_argument("--tmpdir",
367 help="run tests in the given temporary directory"
367 help="run tests in the given temporary directory"
368 " (implies --keep-tmpdir)")
368 " (implies --keep-tmpdir)")
369 harness.add_argument("-v", "--verbose", action="store_true",
369 harness.add_argument("-v", "--verbose", action="store_true",
370 help="output verbose messages")
370 help="output verbose messages")
371
371
372 hgconf = parser.add_argument_group('Mercurial Configuration')
372 hgconf = parser.add_argument_group('Mercurial Configuration')
373 hgconf.add_argument("--chg", action="store_true",
373 hgconf.add_argument("--chg", action="store_true",
374 help="install and use chg wrapper in place of hg")
374 help="install and use chg wrapper in place of hg")
375 hgconf.add_argument("--compiler",
375 hgconf.add_argument("--compiler",
376 help="compiler to build with")
376 help="compiler to build with")
377 hgconf.add_argument('--extra-config-opt', action="append", default=[],
377 hgconf.add_argument('--extra-config-opt', action="append", default=[],
378 help='set the given config opt in the test hgrc')
378 help='set the given config opt in the test hgrc')
379 hgconf.add_argument("-l", "--local", action="store_true",
379 hgconf.add_argument("-l", "--local", action="store_true",
380 help="shortcut for --with-hg=<testdir>/../hg, "
380 help="shortcut for --with-hg=<testdir>/../hg, "
381 "and --with-chg=<testdir>/../contrib/chg/chg if --chg is set")
381 "and --with-chg=<testdir>/../contrib/chg/chg if --chg is set")
382 hgconf.add_argument("--ipv6", action="store_true",
382 hgconf.add_argument("--ipv6", action="store_true",
383 help="prefer IPv6 to IPv4 for network related tests")
383 help="prefer IPv6 to IPv4 for network related tests")
384 hgconf.add_argument("--pure", action="store_true",
384 hgconf.add_argument("--pure", action="store_true",
385 help="use pure Python code instead of C extensions")
385 help="use pure Python code instead of C extensions")
386 hgconf.add_argument("-3", "--py3k-warnings", action="store_true",
386 hgconf.add_argument("-3", "--py3k-warnings", action="store_true",
387 help="enable Py3k warnings on Python 2.7+")
387 help="enable Py3k warnings on Python 2.7+")
388 hgconf.add_argument("--with-chg", metavar="CHG",
388 hgconf.add_argument("--with-chg", metavar="CHG",
389 help="use specified chg wrapper in place of hg")
389 help="use specified chg wrapper in place of hg")
390 hgconf.add_argument("--with-hg",
390 hgconf.add_argument("--with-hg",
391 metavar="HG",
391 metavar="HG",
392 help="test using specified hg script rather than a "
392 help="test using specified hg script rather than a "
393 "temporary installation")
393 "temporary installation")
394 # This option should be deleted once test-check-py3-compat.t and other
394 # This option should be deleted once test-check-py3-compat.t and other
395 # Python 3 tests run with Python 3.
395 # Python 3 tests run with Python 3.
396 hgconf.add_argument("--with-python3", metavar="PYTHON3",
396 hgconf.add_argument("--with-python3", metavar="PYTHON3",
397 help="Python 3 interpreter (if running under Python 2)"
397 help="Python 3 interpreter (if running under Python 2)"
398 " (TEMPORARY)")
398 " (TEMPORARY)")
399
399
400 reporting = parser.add_argument_group('Results Reporting')
400 reporting = parser.add_argument_group('Results Reporting')
401 reporting.add_argument("-C", "--annotate", action="store_true",
401 reporting.add_argument("-C", "--annotate", action="store_true",
402 help="output files annotated with coverage")
402 help="output files annotated with coverage")
403 reporting.add_argument("--color", choices=["always", "auto", "never"],
403 reporting.add_argument("--color", choices=["always", "auto", "never"],
404 default=os.environ.get('HGRUNTESTSCOLOR', 'auto'),
404 default=os.environ.get('HGRUNTESTSCOLOR', 'auto'),
405 help="colorisation: always|auto|never (default: auto)")
405 help="colorisation: always|auto|never (default: auto)")
406 reporting.add_argument("-c", "--cover", action="store_true",
406 reporting.add_argument("-c", "--cover", action="store_true",
407 help="print a test coverage report")
407 help="print a test coverage report")
408 reporting.add_argument('--exceptions', action='store_true',
408 reporting.add_argument('--exceptions', action='store_true',
409 help='log all exceptions and generate an exception report')
409 help='log all exceptions and generate an exception report')
410 reporting.add_argument("-H", "--htmlcov", action="store_true",
410 reporting.add_argument("-H", "--htmlcov", action="store_true",
411 help="create an HTML report of the coverage of the files")
411 help="create an HTML report of the coverage of the files")
412 reporting.add_argument("--json", action="store_true",
412 reporting.add_argument("--json", action="store_true",
413 help="store test result data in 'report.json' file")
413 help="store test result data in 'report.json' file")
414 reporting.add_argument("--outputdir",
414 reporting.add_argument("--outputdir",
415 help="directory to write error logs to (default=test directory)")
415 help="directory to write error logs to (default=test directory)")
416 reporting.add_argument("-n", "--nodiff", action="store_true",
416 reporting.add_argument("-n", "--nodiff", action="store_true",
417 help="skip showing test changes")
417 help="skip showing test changes")
418 reporting.add_argument("-S", "--noskips", action="store_true",
418 reporting.add_argument("-S", "--noskips", action="store_true",
419 help="don't report skip tests verbosely")
419 help="don't report skip tests verbosely")
420 reporting.add_argument("--time", action="store_true",
420 reporting.add_argument("--time", action="store_true",
421 help="time how long each test takes")
421 help="time how long each test takes")
422 reporting.add_argument("--view",
422 reporting.add_argument("--view",
423 help="external diff viewer")
423 help="external diff viewer")
424 reporting.add_argument("--xunit",
424 reporting.add_argument("--xunit",
425 help="record xunit results at specified path")
425 help="record xunit results at specified path")
426
426
427 for option, (envvar, default) in defaults.items():
427 for option, (envvar, default) in defaults.items():
428 defaults[option] = type(default)(os.environ.get(envvar, default))
428 defaults[option] = type(default)(os.environ.get(envvar, default))
429 parser.set_defaults(**defaults)
429 parser.set_defaults(**defaults)
430
430
431 return parser
431 return parser
432
432
433 def parseargs(args, parser):
433 def parseargs(args, parser):
434 """Parse arguments with our OptionParser and validate results."""
434 """Parse arguments with our OptionParser and validate results."""
435 options = parser.parse_args(args)
435 options = parser.parse_args(args)
436
436
437 # jython is always pure
437 # jython is always pure
438 if 'java' in sys.platform or '__pypy__' in sys.modules:
438 if 'java' in sys.platform or '__pypy__' in sys.modules:
439 options.pure = True
439 options.pure = True
440
440
441 if options.with_hg:
441 if options.with_hg:
442 options.with_hg = canonpath(_bytespath(options.with_hg))
442 options.with_hg = canonpath(_bytespath(options.with_hg))
443 if not (os.path.isfile(options.with_hg) and
443 if not (os.path.isfile(options.with_hg) and
444 os.access(options.with_hg, os.X_OK)):
444 os.access(options.with_hg, os.X_OK)):
445 parser.error('--with-hg must specify an executable hg script')
445 parser.error('--with-hg must specify an executable hg script')
446 if os.path.basename(options.with_hg) not in [b'hg', b'hg.exe']:
446 if os.path.basename(options.with_hg) not in [b'hg', b'hg.exe']:
447 sys.stderr.write('warning: --with-hg should specify an hg script\n')
447 sys.stderr.write('warning: --with-hg should specify an hg script\n')
448 if options.local:
448 if options.local:
449 testdir = os.path.dirname(_bytespath(canonpath(sys.argv[0])))
449 testdir = os.path.dirname(_bytespath(canonpath(sys.argv[0])))
450 reporootdir = os.path.dirname(testdir)
450 reporootdir = os.path.dirname(testdir)
451 pathandattrs = [(b'hg', 'with_hg')]
451 pathandattrs = [(b'hg', 'with_hg')]
452 if options.chg:
452 if options.chg:
453 pathandattrs.append((b'contrib/chg/chg', 'with_chg'))
453 pathandattrs.append((b'contrib/chg/chg', 'with_chg'))
454 for relpath, attr in pathandattrs:
454 for relpath, attr in pathandattrs:
455 binpath = os.path.join(reporootdir, relpath)
455 binpath = os.path.join(reporootdir, relpath)
456 if os.name != 'nt' and not os.access(binpath, os.X_OK):
456 if os.name != 'nt' and not os.access(binpath, os.X_OK):
457 parser.error('--local specified, but %r not found or '
457 parser.error('--local specified, but %r not found or '
458 'not executable' % binpath)
458 'not executable' % binpath)
459 setattr(options, attr, binpath)
459 setattr(options, attr, binpath)
460
460
461 if (options.chg or options.with_chg) and os.name == 'nt':
461 if (options.chg or options.with_chg) and os.name == 'nt':
462 parser.error('chg does not work on %s' % os.name)
462 parser.error('chg does not work on %s' % os.name)
463 if options.with_chg:
463 if options.with_chg:
464 options.chg = False # no installation to temporary location
464 options.chg = False # no installation to temporary location
465 options.with_chg = canonpath(_bytespath(options.with_chg))
465 options.with_chg = canonpath(_bytespath(options.with_chg))
466 if not (os.path.isfile(options.with_chg) and
466 if not (os.path.isfile(options.with_chg) and
467 os.access(options.with_chg, os.X_OK)):
467 os.access(options.with_chg, os.X_OK)):
468 parser.error('--with-chg must specify a chg executable')
468 parser.error('--with-chg must specify a chg executable')
469 if options.chg and options.with_hg:
469 if options.chg and options.with_hg:
470 # chg shares installation location with hg
470 # chg shares installation location with hg
471 parser.error('--chg does not work when --with-hg is specified '
471 parser.error('--chg does not work when --with-hg is specified '
472 '(use --with-chg instead)')
472 '(use --with-chg instead)')
473
473
474 if options.color == 'always' and not pygmentspresent:
474 if options.color == 'always' and not pygmentspresent:
475 sys.stderr.write('warning: --color=always ignored because '
475 sys.stderr.write('warning: --color=always ignored because '
476 'pygments is not installed\n')
476 'pygments is not installed\n')
477
477
478 if options.bisect_repo and not options.known_good_rev:
478 if options.bisect_repo and not options.known_good_rev:
479 parser.error("--bisect-repo cannot be used without --known-good-rev")
479 parser.error("--bisect-repo cannot be used without --known-good-rev")
480
480
481 global useipv6
481 global useipv6
482 if options.ipv6:
482 if options.ipv6:
483 useipv6 = checksocketfamily('AF_INET6')
483 useipv6 = checksocketfamily('AF_INET6')
484 else:
484 else:
485 # only use IPv6 if IPv4 is unavailable and IPv6 is available
485 # only use IPv6 if IPv4 is unavailable and IPv6 is available
486 useipv6 = ((not checksocketfamily('AF_INET'))
486 useipv6 = ((not checksocketfamily('AF_INET'))
487 and checksocketfamily('AF_INET6'))
487 and checksocketfamily('AF_INET6'))
488
488
489 options.anycoverage = options.cover or options.annotate or options.htmlcov
489 options.anycoverage = options.cover or options.annotate or options.htmlcov
490 if options.anycoverage:
490 if options.anycoverage:
491 try:
491 try:
492 import coverage
492 import coverage
493 covver = version.StrictVersion(coverage.__version__).version
493 covver = version.StrictVersion(coverage.__version__).version
494 if covver < (3, 3):
494 if covver < (3, 3):
495 parser.error('coverage options require coverage 3.3 or later')
495 parser.error('coverage options require coverage 3.3 or later')
496 except ImportError:
496 except ImportError:
497 parser.error('coverage options now require the coverage package')
497 parser.error('coverage options now require the coverage package')
498
498
499 if options.anycoverage and options.local:
499 if options.anycoverage and options.local:
500 # this needs some path mangling somewhere, I guess
500 # this needs some path mangling somewhere, I guess
501 parser.error("sorry, coverage options do not work when --local "
501 parser.error("sorry, coverage options do not work when --local "
502 "is specified")
502 "is specified")
503
503
504 if options.anycoverage and options.with_hg:
504 if options.anycoverage and options.with_hg:
505 parser.error("sorry, coverage options do not work when --with-hg "
505 parser.error("sorry, coverage options do not work when --with-hg "
506 "is specified")
506 "is specified")
507
507
508 global verbose
508 global verbose
509 if options.verbose:
509 if options.verbose:
510 verbose = ''
510 verbose = ''
511
511
512 if options.tmpdir:
512 if options.tmpdir:
513 options.tmpdir = canonpath(options.tmpdir)
513 options.tmpdir = canonpath(options.tmpdir)
514
514
515 if options.jobs < 1:
515 if options.jobs < 1:
516 parser.error('--jobs must be positive')
516 parser.error('--jobs must be positive')
517 if options.interactive and options.debug:
517 if options.interactive and options.debug:
518 parser.error("-i/--interactive and -d/--debug are incompatible")
518 parser.error("-i/--interactive and -d/--debug are incompatible")
519 if options.debug:
519 if options.debug:
520 if options.timeout != defaults['timeout']:
520 if options.timeout != defaults['timeout']:
521 sys.stderr.write(
521 sys.stderr.write(
522 'warning: --timeout option ignored with --debug\n')
522 'warning: --timeout option ignored with --debug\n')
523 if options.slowtimeout != defaults['slowtimeout']:
523 if options.slowtimeout != defaults['slowtimeout']:
524 sys.stderr.write(
524 sys.stderr.write(
525 'warning: --slowtimeout option ignored with --debug\n')
525 'warning: --slowtimeout option ignored with --debug\n')
526 options.timeout = 0
526 options.timeout = 0
527 options.slowtimeout = 0
527 options.slowtimeout = 0
528 if options.py3k_warnings:
528 if options.py3k_warnings:
529 if PYTHON3:
529 if PYTHON3:
530 parser.error(
530 parser.error(
531 '--py3k-warnings can only be used on Python 2.7')
531 '--py3k-warnings can only be used on Python 2.7')
532 if options.with_python3:
532 if options.with_python3:
533 if PYTHON3:
533 if PYTHON3:
534 parser.error('--with-python3 cannot be used when executing with '
534 parser.error('--with-python3 cannot be used when executing with '
535 'Python 3')
535 'Python 3')
536
536
537 options.with_python3 = canonpath(options.with_python3)
537 options.with_python3 = canonpath(options.with_python3)
538 # Verify Python3 executable is acceptable.
538 # Verify Python3 executable is acceptable.
539 proc = subprocess.Popen([options.with_python3, b'--version'],
539 proc = subprocess.Popen([options.with_python3, b'--version'],
540 stdout=subprocess.PIPE,
540 stdout=subprocess.PIPE,
541 stderr=subprocess.STDOUT)
541 stderr=subprocess.STDOUT)
542 out, _err = proc.communicate()
542 out, _err = proc.communicate()
543 ret = proc.wait()
543 ret = proc.wait()
544 if ret != 0:
544 if ret != 0:
545 parser.error('could not determine version of python 3')
545 parser.error('could not determine version of python 3')
546 if not out.startswith('Python '):
546 if not out.startswith('Python '):
547 parser.error('unexpected output from python3 --version: %s' %
547 parser.error('unexpected output from python3 --version: %s' %
548 out)
548 out)
549 vers = version.LooseVersion(out[len('Python '):])
549 vers = version.LooseVersion(out[len('Python '):])
550 if vers < version.LooseVersion('3.5.0'):
550 if vers < version.LooseVersion('3.5.0'):
551 parser.error('--with-python3 version must be 3.5.0 or greater; '
551 parser.error('--with-python3 version must be 3.5.0 or greater; '
552 'got %s' % out)
552 'got %s' % out)
553
553
554 if options.blacklist:
554 if options.blacklist:
555 options.blacklist = parselistfiles(options.blacklist, 'blacklist')
555 options.blacklist = parselistfiles(options.blacklist, 'blacklist')
556 if options.whitelist:
556 if options.whitelist:
557 options.whitelisted = parselistfiles(options.whitelist, 'whitelist')
557 options.whitelisted = parselistfiles(options.whitelist, 'whitelist')
558 else:
558 else:
559 options.whitelisted = {}
559 options.whitelisted = {}
560
560
561 if options.showchannels:
561 if options.showchannels:
562 options.nodiff = True
562 options.nodiff = True
563
563
564 return options
564 return options
565
565
566 def rename(src, dst):
566 def rename(src, dst):
567 """Like os.rename(), trade atomicity and opened files friendliness
567 """Like os.rename(), trade atomicity and opened files friendliness
568 for existing destination support.
568 for existing destination support.
569 """
569 """
570 shutil.copy(src, dst)
570 shutil.copy(src, dst)
571 os.remove(src)
571 os.remove(src)
572
572
573 _unified_diff = difflib.unified_diff
573 _unified_diff = difflib.unified_diff
574 if PYTHON3:
574 if PYTHON3:
575 import functools
575 import functools
576 _unified_diff = functools.partial(difflib.diff_bytes, difflib.unified_diff)
576 _unified_diff = functools.partial(difflib.diff_bytes, difflib.unified_diff)
577
577
578 def getdiff(expected, output, ref, err):
578 def getdiff(expected, output, ref, err):
579 servefail = False
579 servefail = False
580 lines = []
580 lines = []
581 for line in _unified_diff(expected, output, ref, err):
581 for line in _unified_diff(expected, output, ref, err):
582 if line.startswith(b'+++') or line.startswith(b'---'):
582 if line.startswith(b'+++') or line.startswith(b'---'):
583 line = line.replace(b'\\', b'/')
583 line = line.replace(b'\\', b'/')
584 if line.endswith(b' \n'):
584 if line.endswith(b' \n'):
585 line = line[:-2] + b'\n'
585 line = line[:-2] + b'\n'
586 lines.append(line)
586 lines.append(line)
587 if not servefail and line.startswith(
587 if not servefail and line.startswith(
588 b'+ abort: child process failed to start'):
588 b'+ abort: child process failed to start'):
589 servefail = True
589 servefail = True
590
590
591 return servefail, lines
591 return servefail, lines
592
592
593 verbose = False
593 verbose = False
594 def vlog(*msg):
594 def vlog(*msg):
595 """Log only when in verbose mode."""
595 """Log only when in verbose mode."""
596 if verbose is False:
596 if verbose is False:
597 return
597 return
598
598
599 return log(*msg)
599 return log(*msg)
600
600
601 # Bytes that break XML even in a CDATA block: control characters 0-31
601 # Bytes that break XML even in a CDATA block: control characters 0-31
602 # sans \t, \n and \r
602 # sans \t, \n and \r
603 CDATA_EVIL = re.compile(br"[\000-\010\013\014\016-\037]")
603 CDATA_EVIL = re.compile(br"[\000-\010\013\014\016-\037]")
604
604
605 # Match feature conditionalized output lines in the form, capturing the feature
605 # Match feature conditionalized output lines in the form, capturing the feature
606 # list in group 2, and the preceeding line output in group 1:
606 # list in group 2, and the preceeding line output in group 1:
607 #
607 #
608 # output..output (feature !)\n
608 # output..output (feature !)\n
609 optline = re.compile(b'(.*) \((.+?) !\)\n$')
609 optline = re.compile(b'(.*) \((.+?) !\)\n$')
610
610
611 def cdatasafe(data):
611 def cdatasafe(data):
612 """Make a string safe to include in a CDATA block.
612 """Make a string safe to include in a CDATA block.
613
613
614 Certain control characters are illegal in a CDATA block, and
614 Certain control characters are illegal in a CDATA block, and
615 there's no way to include a ]]> in a CDATA either. This function
615 there's no way to include a ]]> in a CDATA either. This function
616 replaces illegal bytes with ? and adds a space between the ]] so
616 replaces illegal bytes with ? and adds a space between the ]] so
617 that it won't break the CDATA block.
617 that it won't break the CDATA block.
618 """
618 """
619 return CDATA_EVIL.sub(b'?', data).replace(b']]>', b'] ]>')
619 return CDATA_EVIL.sub(b'?', data).replace(b']]>', b'] ]>')
620
620
621 def log(*msg):
621 def log(*msg):
622 """Log something to stdout.
622 """Log something to stdout.
623
623
624 Arguments are strings to print.
624 Arguments are strings to print.
625 """
625 """
626 with iolock:
626 with iolock:
627 if verbose:
627 if verbose:
628 print(verbose, end=' ')
628 print(verbose, end=' ')
629 for m in msg:
629 for m in msg:
630 print(m, end=' ')
630 print(m, end=' ')
631 print()
631 print()
632 sys.stdout.flush()
632 sys.stdout.flush()
633
633
634 def highlightdiff(line, color):
634 def highlightdiff(line, color):
635 if not color:
635 if not color:
636 return line
636 return line
637 assert pygmentspresent
637 assert pygmentspresent
638 return pygments.highlight(line.decode('latin1'), difflexer,
638 return pygments.highlight(line.decode('latin1'), difflexer,
639 terminal256formatter).encode('latin1')
639 terminal256formatter).encode('latin1')
640
640
641 def highlightmsg(msg, color):
641 def highlightmsg(msg, color):
642 if not color:
642 if not color:
643 return msg
643 return msg
644 assert pygmentspresent
644 assert pygmentspresent
645 return pygments.highlight(msg, runnerlexer, runnerformatter)
645 return pygments.highlight(msg, runnerlexer, runnerformatter)
646
646
647 def terminate(proc):
647 def terminate(proc):
648 """Terminate subprocess"""
648 """Terminate subprocess"""
649 vlog('# Terminating process %d' % proc.pid)
649 vlog('# Terminating process %d' % proc.pid)
650 try:
650 try:
651 proc.terminate()
651 proc.terminate()
652 except OSError:
652 except OSError:
653 pass
653 pass
654
654
655 def killdaemons(pidfile):
655 def killdaemons(pidfile):
656 import killdaemons as killmod
656 import killdaemons as killmod
657 return killmod.killdaemons(pidfile, tryhard=False, remove=True,
657 return killmod.killdaemons(pidfile, tryhard=False, remove=True,
658 logfn=vlog)
658 logfn=vlog)
659
659
660 class Test(unittest.TestCase):
660 class Test(unittest.TestCase):
661 """Encapsulates a single, runnable test.
661 """Encapsulates a single, runnable test.
662
662
663 While this class conforms to the unittest.TestCase API, it differs in that
663 While this class conforms to the unittest.TestCase API, it differs in that
664 instances need to be instantiated manually. (Typically, unittest.TestCase
664 instances need to be instantiated manually. (Typically, unittest.TestCase
665 classes are instantiated automatically by scanning modules.)
665 classes are instantiated automatically by scanning modules.)
666 """
666 """
667
667
668 # Status code reserved for skipped tests (used by hghave).
668 # Status code reserved for skipped tests (used by hghave).
669 SKIPPED_STATUS = 80
669 SKIPPED_STATUS = 80
670
670
671 def __init__(self, path, outputdir, tmpdir, keeptmpdir=False,
671 def __init__(self, path, outputdir, tmpdir, keeptmpdir=False,
672 debug=False,
672 debug=False,
673 timeout=None,
673 timeout=None,
674 startport=None, extraconfigopts=None,
674 startport=None, extraconfigopts=None,
675 py3kwarnings=False, shell=None, hgcommand=None,
675 py3kwarnings=False, shell=None, hgcommand=None,
676 slowtimeout=None, usechg=False,
676 slowtimeout=None, usechg=False,
677 useipv6=False):
677 useipv6=False):
678 """Create a test from parameters.
678 """Create a test from parameters.
679
679
680 path is the full path to the file defining the test.
680 path is the full path to the file defining the test.
681
681
682 tmpdir is the main temporary directory to use for this test.
682 tmpdir is the main temporary directory to use for this test.
683
683
684 keeptmpdir determines whether to keep the test's temporary directory
684 keeptmpdir determines whether to keep the test's temporary directory
685 after execution. It defaults to removal (False).
685 after execution. It defaults to removal (False).
686
686
687 debug mode will make the test execute verbosely, with unfiltered
687 debug mode will make the test execute verbosely, with unfiltered
688 output.
688 output.
689
689
690 timeout controls the maximum run time of the test. It is ignored when
690 timeout controls the maximum run time of the test. It is ignored when
691 debug is True. See slowtimeout for tests with #require slow.
691 debug is True. See slowtimeout for tests with #require slow.
692
692
693 slowtimeout overrides timeout if the test has #require slow.
693 slowtimeout overrides timeout if the test has #require slow.
694
694
695 startport controls the starting port number to use for this test. Each
695 startport controls the starting port number to use for this test. Each
696 test will reserve 3 port numbers for execution. It is the caller's
696 test will reserve 3 port numbers for execution. It is the caller's
697 responsibility to allocate a non-overlapping port range to Test
697 responsibility to allocate a non-overlapping port range to Test
698 instances.
698 instances.
699
699
700 extraconfigopts is an iterable of extra hgrc config options. Values
700 extraconfigopts is an iterable of extra hgrc config options. Values
701 must have the form "key=value" (something understood by hgrc). Values
701 must have the form "key=value" (something understood by hgrc). Values
702 of the form "foo.key=value" will result in "[foo] key=value".
702 of the form "foo.key=value" will result in "[foo] key=value".
703
703
704 py3kwarnings enables Py3k warnings.
704 py3kwarnings enables Py3k warnings.
705
705
706 shell is the shell to execute tests in.
706 shell is the shell to execute tests in.
707 """
707 """
708 if timeout is None:
708 if timeout is None:
709 timeout = defaults['timeout']
709 timeout = defaults['timeout']
710 if startport is None:
710 if startport is None:
711 startport = defaults['port']
711 startport = defaults['port']
712 if slowtimeout is None:
712 if slowtimeout is None:
713 slowtimeout = defaults['slowtimeout']
713 slowtimeout = defaults['slowtimeout']
714 self.path = path
714 self.path = path
715 self.bname = os.path.basename(path)
715 self.bname = os.path.basename(path)
716 self.name = _strpath(self.bname)
716 self.name = _strpath(self.bname)
717 self._testdir = os.path.dirname(path)
717 self._testdir = os.path.dirname(path)
718 self._outputdir = outputdir
718 self._outputdir = outputdir
719 self._tmpname = os.path.basename(path)
719 self._tmpname = os.path.basename(path)
720 self.errpath = os.path.join(self._outputdir, b'%s.err' % self.bname)
720 self.errpath = os.path.join(self._outputdir, b'%s.err' % self.bname)
721
721
722 self._threadtmp = tmpdir
722 self._threadtmp = tmpdir
723 self._keeptmpdir = keeptmpdir
723 self._keeptmpdir = keeptmpdir
724 self._debug = debug
724 self._debug = debug
725 self._timeout = timeout
725 self._timeout = timeout
726 self._slowtimeout = slowtimeout
726 self._slowtimeout = slowtimeout
727 self._startport = startport
727 self._startport = startport
728 self._extraconfigopts = extraconfigopts or []
728 self._extraconfigopts = extraconfigopts or []
729 self._py3kwarnings = py3kwarnings
729 self._py3kwarnings = py3kwarnings
730 self._shell = _bytespath(shell)
730 self._shell = _bytespath(shell)
731 self._hgcommand = hgcommand or b'hg'
731 self._hgcommand = hgcommand or b'hg'
732 self._usechg = usechg
732 self._usechg = usechg
733 self._useipv6 = useipv6
733 self._useipv6 = useipv6
734
734
735 self._aborted = False
735 self._aborted = False
736 self._daemonpids = []
736 self._daemonpids = []
737 self._finished = None
737 self._finished = None
738 self._ret = None
738 self._ret = None
739 self._out = None
739 self._out = None
740 self._skipped = None
740 self._skipped = None
741 self._testtmp = None
741 self._testtmp = None
742 self._chgsockdir = None
742 self._chgsockdir = None
743
743
744 self._refout = self.readrefout()
744 self._refout = self.readrefout()
745
745
746 def readrefout(self):
746 def readrefout(self):
747 """read reference output"""
747 """read reference output"""
748 # If we're not in --debug mode and reference output file exists,
748 # If we're not in --debug mode and reference output file exists,
749 # check test output against it.
749 # check test output against it.
750 if self._debug:
750 if self._debug:
751 return None # to match "out is None"
751 return None # to match "out is None"
752 elif os.path.exists(self.refpath):
752 elif os.path.exists(self.refpath):
753 with open(self.refpath, 'rb') as f:
753 with open(self.refpath, 'rb') as f:
754 return f.read().splitlines(True)
754 return f.read().splitlines(True)
755 else:
755 else:
756 return []
756 return []
757
757
758 # needed to get base class __repr__ running
758 # needed to get base class __repr__ running
759 @property
759 @property
760 def _testMethodName(self):
760 def _testMethodName(self):
761 return self.name
761 return self.name
762
762
763 def __str__(self):
763 def __str__(self):
764 return self.name
764 return self.name
765
765
766 def shortDescription(self):
766 def shortDescription(self):
767 return self.name
767 return self.name
768
768
769 def setUp(self):
769 def setUp(self):
770 """Tasks to perform before run()."""
770 """Tasks to perform before run()."""
771 self._finished = False
771 self._finished = False
772 self._ret = None
772 self._ret = None
773 self._out = None
773 self._out = None
774 self._skipped = None
774 self._skipped = None
775
775
776 try:
776 try:
777 os.mkdir(self._threadtmp)
777 os.mkdir(self._threadtmp)
778 except OSError as e:
778 except OSError as e:
779 if e.errno != errno.EEXIST:
779 if e.errno != errno.EEXIST:
780 raise
780 raise
781
781
782 name = self._tmpname
782 name = self._tmpname
783 self._testtmp = os.path.join(self._threadtmp, name)
783 self._testtmp = os.path.join(self._threadtmp, name)
784 os.mkdir(self._testtmp)
784 os.mkdir(self._testtmp)
785
785
786 # Remove any previous output files.
786 # Remove any previous output files.
787 if os.path.exists(self.errpath):
787 if os.path.exists(self.errpath):
788 try:
788 try:
789 os.remove(self.errpath)
789 os.remove(self.errpath)
790 except OSError as e:
790 except OSError as e:
791 # We might have raced another test to clean up a .err
791 # We might have raced another test to clean up a .err
792 # file, so ignore ENOENT when removing a previous .err
792 # file, so ignore ENOENT when removing a previous .err
793 # file.
793 # file.
794 if e.errno != errno.ENOENT:
794 if e.errno != errno.ENOENT:
795 raise
795 raise
796
796
797 if self._usechg:
797 if self._usechg:
798 self._chgsockdir = os.path.join(self._threadtmp,
798 self._chgsockdir = os.path.join(self._threadtmp,
799 b'%s.chgsock' % name)
799 b'%s.chgsock' % name)
800 os.mkdir(self._chgsockdir)
800 os.mkdir(self._chgsockdir)
801
801
802 def run(self, result):
802 def run(self, result):
803 """Run this test and report results against a TestResult instance."""
803 """Run this test and report results against a TestResult instance."""
804 # This function is extremely similar to unittest.TestCase.run(). Once
804 # This function is extremely similar to unittest.TestCase.run(). Once
805 # we require Python 2.7 (or at least its version of unittest), this
805 # we require Python 2.7 (or at least its version of unittest), this
806 # function can largely go away.
806 # function can largely go away.
807 self._result = result
807 self._result = result
808 result.startTest(self)
808 result.startTest(self)
809 try:
809 try:
810 try:
810 try:
811 self.setUp()
811 self.setUp()
812 except (KeyboardInterrupt, SystemExit):
812 except (KeyboardInterrupt, SystemExit):
813 self._aborted = True
813 self._aborted = True
814 raise
814 raise
815 except Exception:
815 except Exception:
816 result.addError(self, sys.exc_info())
816 result.addError(self, sys.exc_info())
817 return
817 return
818
818
819 success = False
819 success = False
820 try:
820 try:
821 self.runTest()
821 self.runTest()
822 except KeyboardInterrupt:
822 except KeyboardInterrupt:
823 self._aborted = True
823 self._aborted = True
824 raise
824 raise
825 except unittest.SkipTest as e:
825 except unittest.SkipTest as e:
826 result.addSkip(self, str(e))
826 result.addSkip(self, str(e))
827 # The base class will have already counted this as a
827 # The base class will have already counted this as a
828 # test we "ran", but we want to exclude skipped tests
828 # test we "ran", but we want to exclude skipped tests
829 # from those we count towards those run.
829 # from those we count towards those run.
830 result.testsRun -= 1
830 result.testsRun -= 1
831 except self.failureException as e:
831 except self.failureException as e:
832 # This differs from unittest in that we don't capture
832 # This differs from unittest in that we don't capture
833 # the stack trace. This is for historical reasons and
833 # the stack trace. This is for historical reasons and
834 # this decision could be revisited in the future,
834 # this decision could be revisited in the future,
835 # especially for PythonTest instances.
835 # especially for PythonTest instances.
836 if result.addFailure(self, str(e)):
836 if result.addFailure(self, str(e)):
837 success = True
837 success = True
838 except Exception:
838 except Exception:
839 result.addError(self, sys.exc_info())
839 result.addError(self, sys.exc_info())
840 else:
840 else:
841 success = True
841 success = True
842
842
843 try:
843 try:
844 self.tearDown()
844 self.tearDown()
845 except (KeyboardInterrupt, SystemExit):
845 except (KeyboardInterrupt, SystemExit):
846 self._aborted = True
846 self._aborted = True
847 raise
847 raise
848 except Exception:
848 except Exception:
849 result.addError(self, sys.exc_info())
849 result.addError(self, sys.exc_info())
850 success = False
850 success = False
851
851
852 if success:
852 if success:
853 result.addSuccess(self)
853 result.addSuccess(self)
854 finally:
854 finally:
855 result.stopTest(self, interrupted=self._aborted)
855 result.stopTest(self, interrupted=self._aborted)
856
856
857 def runTest(self):
857 def runTest(self):
858 """Run this test instance.
858 """Run this test instance.
859
859
860 This will return a tuple describing the result of the test.
860 This will return a tuple describing the result of the test.
861 """
861 """
862 env = self._getenv()
862 env = self._getenv()
863 self._genrestoreenv(env)
863 self._genrestoreenv(env)
864 self._daemonpids.append(env['DAEMON_PIDS'])
864 self._daemonpids.append(env['DAEMON_PIDS'])
865 self._createhgrc(env['HGRCPATH'])
865 self._createhgrc(env['HGRCPATH'])
866
866
867 vlog('# Test', self.name)
867 vlog('# Test', self.name)
868
868
869 ret, out = self._run(env)
869 ret, out = self._run(env)
870 self._finished = True
870 self._finished = True
871 self._ret = ret
871 self._ret = ret
872 self._out = out
872 self._out = out
873
873
874 def describe(ret):
874 def describe(ret):
875 if ret < 0:
875 if ret < 0:
876 return 'killed by signal: %d' % -ret
876 return 'killed by signal: %d' % -ret
877 return 'returned error code %d' % ret
877 return 'returned error code %d' % ret
878
878
879 self._skipped = False
879 self._skipped = False
880
880
881 if ret == self.SKIPPED_STATUS:
881 if ret == self.SKIPPED_STATUS:
882 if out is None: # Debug mode, nothing to parse.
882 if out is None: # Debug mode, nothing to parse.
883 missing = ['unknown']
883 missing = ['unknown']
884 failed = None
884 failed = None
885 else:
885 else:
886 missing, failed = TTest.parsehghaveoutput(out)
886 missing, failed = TTest.parsehghaveoutput(out)
887
887
888 if not missing:
888 if not missing:
889 missing = ['skipped']
889 missing = ['skipped']
890
890
891 if failed:
891 if failed:
892 self.fail('hg have failed checking for %s' % failed[-1])
892 self.fail('hg have failed checking for %s' % failed[-1])
893 else:
893 else:
894 self._skipped = True
894 self._skipped = True
895 raise unittest.SkipTest(missing[-1])
895 raise unittest.SkipTest(missing[-1])
896 elif ret == 'timeout':
896 elif ret == 'timeout':
897 self.fail('timed out')
897 self.fail('timed out')
898 elif ret is False:
898 elif ret is False:
899 self.fail('no result code from test')
899 self.fail('no result code from test')
900 elif out != self._refout:
900 elif out != self._refout:
901 # Diff generation may rely on written .err file.
901 # Diff generation may rely on written .err file.
902 if (ret != 0 or out != self._refout) and not self._skipped \
902 if (ret != 0 or out != self._refout) and not self._skipped \
903 and not self._debug:
903 and not self._debug:
904 f = open(self.errpath, 'wb')
904 f = open(self.errpath, 'wb')
905 for line in out:
905 for line in out:
906 f.write(line)
906 f.write(line)
907 f.close()
907 f.close()
908
908
909 # The result object handles diff calculation for us.
909 # The result object handles diff calculation for us.
910 if self._result.addOutputMismatch(self, ret, out, self._refout):
910 if self._result.addOutputMismatch(self, ret, out, self._refout):
911 # change was accepted, skip failing
911 # change was accepted, skip failing
912 return
912 return
913
913
914 if ret:
914 if ret:
915 msg = 'output changed and ' + describe(ret)
915 msg = 'output changed and ' + describe(ret)
916 else:
916 else:
917 msg = 'output changed'
917 msg = 'output changed'
918
918
919 self.fail(msg)
919 self.fail(msg)
920 elif ret:
920 elif ret:
921 self.fail(describe(ret))
921 self.fail(describe(ret))
922
922
923 def tearDown(self):
923 def tearDown(self):
924 """Tasks to perform after run()."""
924 """Tasks to perform after run()."""
925 for entry in self._daemonpids:
925 for entry in self._daemonpids:
926 killdaemons(entry)
926 killdaemons(entry)
927 self._daemonpids = []
927 self._daemonpids = []
928
928
929 if self._keeptmpdir:
929 if self._keeptmpdir:
930 log('\nKeeping testtmp dir: %s\nKeeping threadtmp dir: %s' %
930 log('\nKeeping testtmp dir: %s\nKeeping threadtmp dir: %s' %
931 (self._testtmp.decode('utf-8'),
931 (self._testtmp.decode('utf-8'),
932 self._threadtmp.decode('utf-8')))
932 self._threadtmp.decode('utf-8')))
933 else:
933 else:
934 shutil.rmtree(self._testtmp, True)
934 shutil.rmtree(self._testtmp, True)
935 shutil.rmtree(self._threadtmp, True)
935 shutil.rmtree(self._threadtmp, True)
936
936
937 if self._usechg:
937 if self._usechg:
938 # chgservers will stop automatically after they find the socket
938 # chgservers will stop automatically after they find the socket
939 # files are deleted
939 # files are deleted
940 shutil.rmtree(self._chgsockdir, True)
940 shutil.rmtree(self._chgsockdir, True)
941
941
942 if (self._ret != 0 or self._out != self._refout) and not self._skipped \
942 if (self._ret != 0 or self._out != self._refout) and not self._skipped \
943 and not self._debug and self._out:
943 and not self._debug and self._out:
944 f = open(self.errpath, 'wb')
944 f = open(self.errpath, 'wb')
945 for line in self._out:
945 for line in self._out:
946 f.write(line)
946 f.write(line)
947 f.close()
947 f.close()
948
948
949 vlog("# Ret was:", self._ret, '(%s)' % self.name)
949 vlog("# Ret was:", self._ret, '(%s)' % self.name)
950
950
951 def _run(self, env):
951 def _run(self, env):
952 # This should be implemented in child classes to run tests.
952 # This should be implemented in child classes to run tests.
953 raise unittest.SkipTest('unknown test type')
953 raise unittest.SkipTest('unknown test type')
954
954
955 def abort(self):
955 def abort(self):
956 """Terminate execution of this test."""
956 """Terminate execution of this test."""
957 self._aborted = True
957 self._aborted = True
958
958
959 def _portmap(self, i):
959 def _portmap(self, i):
960 offset = b'' if i == 0 else b'%d' % i
960 offset = b'' if i == 0 else b'%d' % i
961 return (br':%d\b' % (self._startport + i), b':$HGPORT%s' % offset)
961 return (br':%d\b' % (self._startport + i), b':$HGPORT%s' % offset)
962
962
963 def _getreplacements(self):
963 def _getreplacements(self):
964 """Obtain a mapping of text replacements to apply to test output.
964 """Obtain a mapping of text replacements to apply to test output.
965
965
966 Test output needs to be normalized so it can be compared to expected
966 Test output needs to be normalized so it can be compared to expected
967 output. This function defines how some of that normalization will
967 output. This function defines how some of that normalization will
968 occur.
968 occur.
969 """
969 """
970 r = [
970 r = [
971 # This list should be parallel to defineport in _getenv
971 # This list should be parallel to defineport in _getenv
972 self._portmap(0),
972 self._portmap(0),
973 self._portmap(1),
973 self._portmap(1),
974 self._portmap(2),
974 self._portmap(2),
975 (br'(?m)^(saved backup bundle to .*\.hg)( \(glob\))?$',
975 (br'(?m)^(saved backup bundle to .*\.hg)( \(glob\))?$',
976 br'\1 (glob)'),
976 br'\1 (glob)'),
977 (br'([^0-9])%s' % re.escape(self._localip()), br'\1$LOCALIP'),
977 (br'([^0-9])%s' % re.escape(self._localip()), br'\1$LOCALIP'),
978 (br'\bHG_TXNID=TXN:[a-f0-9]{40}\b', br'HG_TXNID=TXN:$ID$'),
978 (br'\bHG_TXNID=TXN:[a-f0-9]{40}\b', br'HG_TXNID=TXN:$ID$'),
979 ]
979 ]
980 r.append((self._escapepath(self._testtmp), b'$TESTTMP'))
980 r.append((self._escapepath(self._testtmp), b'$TESTTMP'))
981
981
982 replacementfile = os.path.join(self._testdir, b'common-pattern.py')
982 replacementfile = os.path.join(self._testdir, b'common-pattern.py')
983
983
984 if os.path.exists(replacementfile):
984 if os.path.exists(replacementfile):
985 data = {}
985 data = {}
986 with open(replacementfile, mode='rb') as source:
986 with open(replacementfile, mode='rb') as source:
987 # the intermediate 'compile' step help with debugging
987 # the intermediate 'compile' step help with debugging
988 code = compile(source.read(), replacementfile, 'exec')
988 code = compile(source.read(), replacementfile, 'exec')
989 exec(code, data)
989 exec(code, data)
990 r.extend(data.get('substitutions', ()))
990 r.extend(data.get('substitutions', ()))
991 return r
991 return r
992
992
993 def _escapepath(self, p):
993 def _escapepath(self, p):
994 if os.name == 'nt':
994 if os.name == 'nt':
995 return (
995 return (
996 (b''.join(c.isalpha() and b'[%s%s]' % (c.lower(), c.upper()) or
996 (b''.join(c.isalpha() and b'[%s%s]' % (c.lower(), c.upper()) or
997 c in b'/\\' and br'[/\\]' or c.isdigit() and c or b'\\' + c
997 c in b'/\\' and br'[/\\]' or c.isdigit() and c or b'\\' + c
998 for c in p))
998 for c in p))
999 )
999 )
1000 else:
1000 else:
1001 return re.escape(p)
1001 return re.escape(p)
1002
1002
1003 def _localip(self):
1003 def _localip(self):
1004 if self._useipv6:
1004 if self._useipv6:
1005 return b'::1'
1005 return b'::1'
1006 else:
1006 else:
1007 return b'127.0.0.1'
1007 return b'127.0.0.1'
1008
1008
1009 def _genrestoreenv(self, testenv):
1009 def _genrestoreenv(self, testenv):
1010 """Generate a script that can be used by tests to restore the original
1010 """Generate a script that can be used by tests to restore the original
1011 environment."""
1011 environment."""
1012 # Put the restoreenv script inside self._threadtmp
1012 # Put the restoreenv script inside self._threadtmp
1013 scriptpath = os.path.join(self._threadtmp, b'restoreenv.sh')
1013 scriptpath = os.path.join(self._threadtmp, b'restoreenv.sh')
1014 testenv['HGTEST_RESTOREENV'] = scriptpath
1014 testenv['HGTEST_RESTOREENV'] = scriptpath
1015
1015
1016 # Only restore environment variable names that the shell allows
1016 # Only restore environment variable names that the shell allows
1017 # us to export.
1017 # us to export.
1018 name_regex = re.compile('^[a-zA-Z][a-zA-Z0-9_]*$')
1018 name_regex = re.compile('^[a-zA-Z][a-zA-Z0-9_]*$')
1019
1019
1020 # Do not restore these variables; otherwise tests would fail.
1020 # Do not restore these variables; otherwise tests would fail.
1021 reqnames = {'PYTHON', 'TESTDIR', 'TESTTMP'}
1021 reqnames = {'PYTHON', 'TESTDIR', 'TESTTMP'}
1022
1022
1023 with open(scriptpath, 'w') as envf:
1023 with open(scriptpath, 'w') as envf:
1024 for name, value in origenviron.items():
1024 for name, value in origenviron.items():
1025 if not name_regex.match(name):
1025 if not name_regex.match(name):
1026 # Skip environment variables with unusual names not
1026 # Skip environment variables with unusual names not
1027 # allowed by most shells.
1027 # allowed by most shells.
1028 continue
1028 continue
1029 if name in reqnames:
1029 if name in reqnames:
1030 continue
1030 continue
1031 envf.write('%s=%s\n' % (name, shellquote(value)))
1031 envf.write('%s=%s\n' % (name, shellquote(value)))
1032
1032
1033 for name in testenv:
1033 for name in testenv:
1034 if name in origenviron or name in reqnames:
1034 if name in origenviron or name in reqnames:
1035 continue
1035 continue
1036 envf.write('unset %s\n' % (name,))
1036 envf.write('unset %s\n' % (name,))
1037
1037
1038 def _getenv(self):
1038 def _getenv(self):
1039 """Obtain environment variables to use during test execution."""
1039 """Obtain environment variables to use during test execution."""
1040 def defineport(i):
1040 def defineport(i):
1041 offset = '' if i == 0 else '%s' % i
1041 offset = '' if i == 0 else '%s' % i
1042 env["HGPORT%s" % offset] = '%s' % (self._startport + i)
1042 env["HGPORT%s" % offset] = '%s' % (self._startport + i)
1043 env = os.environ.copy()
1043 env = os.environ.copy()
1044 env['PYTHONUSERBASE'] = sysconfig.get_config_var('userbase')
1044 env['PYTHONUSERBASE'] = sysconfig.get_config_var('userbase')
1045 env['HGEMITWARNINGS'] = '1'
1045 env['HGEMITWARNINGS'] = '1'
1046 env['TESTTMP'] = self._testtmp
1046 env['TESTTMP'] = self._testtmp
1047 env['HOME'] = self._testtmp
1047 env['HOME'] = self._testtmp
1048 # This number should match portneeded in _getport
1048 # This number should match portneeded in _getport
1049 for port in xrange(3):
1049 for port in xrange(3):
1050 # This list should be parallel to _portmap in _getreplacements
1050 # This list should be parallel to _portmap in _getreplacements
1051 defineport(port)
1051 defineport(port)
1052 env["HGRCPATH"] = os.path.join(self._threadtmp, b'.hgrc')
1052 env["HGRCPATH"] = os.path.join(self._threadtmp, b'.hgrc')
1053 env["DAEMON_PIDS"] = os.path.join(self._threadtmp, b'daemon.pids')
1053 env["DAEMON_PIDS"] = os.path.join(self._threadtmp, b'daemon.pids')
1054 env["HGEDITOR"] = ('"' + sys.executable + '"'
1054 env["HGEDITOR"] = ('"' + sys.executable + '"'
1055 + ' -c "import sys; sys.exit(0)"')
1055 + ' -c "import sys; sys.exit(0)"')
1056 env["HGMERGE"] = "internal:merge"
1056 env["HGMERGE"] = "internal:merge"
1057 env["HGUSER"] = "test"
1057 env["HGUSER"] = "test"
1058 env["HGENCODING"] = "ascii"
1058 env["HGENCODING"] = "ascii"
1059 env["HGENCODINGMODE"] = "strict"
1059 env["HGENCODINGMODE"] = "strict"
1060 env['HGIPV6'] = str(int(self._useipv6))
1060 env['HGIPV6'] = str(int(self._useipv6))
1061
1061
1062 # LOCALIP could be ::1 or 127.0.0.1. Useful for tests that require raw
1062 # LOCALIP could be ::1 or 127.0.0.1. Useful for tests that require raw
1063 # IP addresses.
1063 # IP addresses.
1064 env['LOCALIP'] = self._localip()
1064 env['LOCALIP'] = self._localip()
1065
1065
1066 # Reset some environment variables to well-known values so that
1066 # Reset some environment variables to well-known values so that
1067 # the tests produce repeatable output.
1067 # the tests produce repeatable output.
1068 env['LANG'] = env['LC_ALL'] = env['LANGUAGE'] = 'C'
1068 env['LANG'] = env['LC_ALL'] = env['LANGUAGE'] = 'C'
1069 env['TZ'] = 'GMT'
1069 env['TZ'] = 'GMT'
1070 env["EMAIL"] = "Foo Bar <foo.bar@example.com>"
1070 env["EMAIL"] = "Foo Bar <foo.bar@example.com>"
1071 env['COLUMNS'] = '80'
1071 env['COLUMNS'] = '80'
1072 env['TERM'] = 'xterm'
1072 env['TERM'] = 'xterm'
1073
1073
1074 for k in ('HG HGPROF CDPATH GREP_OPTIONS http_proxy no_proxy ' +
1074 for k in ('HG HGPROF CDPATH GREP_OPTIONS http_proxy no_proxy ' +
1075 'HGPLAIN HGPLAINEXCEPT EDITOR VISUAL PAGER ' +
1075 'HGPLAIN HGPLAINEXCEPT EDITOR VISUAL PAGER ' +
1076 'NO_PROXY CHGDEBUG').split():
1076 'NO_PROXY CHGDEBUG').split():
1077 if k in env:
1077 if k in env:
1078 del env[k]
1078 del env[k]
1079
1079
1080 # unset env related to hooks
1080 # unset env related to hooks
1081 for k in env.keys():
1081 for k in env.keys():
1082 if k.startswith('HG_'):
1082 if k.startswith('HG_'):
1083 del env[k]
1083 del env[k]
1084
1084
1085 if self._usechg:
1085 if self._usechg:
1086 env['CHGSOCKNAME'] = os.path.join(self._chgsockdir, b'server')
1086 env['CHGSOCKNAME'] = os.path.join(self._chgsockdir, b'server')
1087
1087
1088 return env
1088 return env
1089
1089
1090 def _createhgrc(self, path):
1090 def _createhgrc(self, path):
1091 """Create an hgrc file for this test."""
1091 """Create an hgrc file for this test."""
1092 hgrc = open(path, 'wb')
1092 hgrc = open(path, 'wb')
1093 hgrc.write(b'[ui]\n')
1093 hgrc.write(b'[ui]\n')
1094 hgrc.write(b'slash = True\n')
1094 hgrc.write(b'slash = True\n')
1095 hgrc.write(b'interactive = False\n')
1095 hgrc.write(b'interactive = False\n')
1096 hgrc.write(b'mergemarkers = detailed\n')
1096 hgrc.write(b'mergemarkers = detailed\n')
1097 hgrc.write(b'promptecho = True\n')
1097 hgrc.write(b'promptecho = True\n')
1098 hgrc.write(b'[defaults]\n')
1098 hgrc.write(b'[defaults]\n')
1099 hgrc.write(b'[devel]\n')
1099 hgrc.write(b'[devel]\n')
1100 hgrc.write(b'all-warnings = true\n')
1100 hgrc.write(b'all-warnings = true\n')
1101 hgrc.write(b'default-date = 0 0\n')
1101 hgrc.write(b'default-date = 0 0\n')
1102 hgrc.write(b'[largefiles]\n')
1102 hgrc.write(b'[largefiles]\n')
1103 hgrc.write(b'usercache = %s\n' %
1103 hgrc.write(b'usercache = %s\n' %
1104 (os.path.join(self._testtmp, b'.cache/largefiles')))
1104 (os.path.join(self._testtmp, b'.cache/largefiles')))
1105 hgrc.write(b'[lfs]\n')
1105 hgrc.write(b'[lfs]\n')
1106 hgrc.write(b'usercache = %s\n' %
1106 hgrc.write(b'usercache = %s\n' %
1107 (os.path.join(self._testtmp, b'.cache/lfs')))
1107 (os.path.join(self._testtmp, b'.cache/lfs')))
1108 hgrc.write(b'[web]\n')
1108 hgrc.write(b'[web]\n')
1109 hgrc.write(b'address = localhost\n')
1109 hgrc.write(b'address = localhost\n')
1110 hgrc.write(b'ipv6 = %s\n' % str(self._useipv6).encode('ascii'))
1110 hgrc.write(b'ipv6 = %s\n' % str(self._useipv6).encode('ascii'))
1111
1111
1112 for opt in self._extraconfigopts:
1112 for opt in self._extraconfigopts:
1113 section, key = opt.encode('utf-8').split(b'.', 1)
1113 section, key = opt.encode('utf-8').split(b'.', 1)
1114 assert b'=' in key, ('extra config opt %s must '
1114 assert b'=' in key, ('extra config opt %s must '
1115 'have an = for assignment' % opt)
1115 'have an = for assignment' % opt)
1116 hgrc.write(b'[%s]\n%s\n' % (section, key))
1116 hgrc.write(b'[%s]\n%s\n' % (section, key))
1117 hgrc.close()
1117 hgrc.close()
1118
1118
1119 def fail(self, msg):
1119 def fail(self, msg):
1120 # unittest differentiates between errored and failed.
1120 # unittest differentiates between errored and failed.
1121 # Failed is denoted by AssertionError (by default at least).
1121 # Failed is denoted by AssertionError (by default at least).
1122 raise AssertionError(msg)
1122 raise AssertionError(msg)
1123
1123
1124 def _runcommand(self, cmd, env, normalizenewlines=False):
1124 def _runcommand(self, cmd, env, normalizenewlines=False):
1125 """Run command in a sub-process, capturing the output (stdout and
1125 """Run command in a sub-process, capturing the output (stdout and
1126 stderr).
1126 stderr).
1127
1127
1128 Return a tuple (exitcode, output). output is None in debug mode.
1128 Return a tuple (exitcode, output). output is None in debug mode.
1129 """
1129 """
1130 if self._debug:
1130 if self._debug:
1131 proc = subprocess.Popen(cmd, shell=True, cwd=self._testtmp,
1131 proc = subprocess.Popen(cmd, shell=True, cwd=self._testtmp,
1132 env=env)
1132 env=env)
1133 ret = proc.wait()
1133 ret = proc.wait()
1134 return (ret, None)
1134 return (ret, None)
1135
1135
1136 proc = Popen4(cmd, self._testtmp, self._timeout, env)
1136 proc = Popen4(cmd, self._testtmp, self._timeout, env)
1137 def cleanup():
1137 def cleanup():
1138 terminate(proc)
1138 terminate(proc)
1139 ret = proc.wait()
1139 ret = proc.wait()
1140 if ret == 0:
1140 if ret == 0:
1141 ret = signal.SIGTERM << 8
1141 ret = signal.SIGTERM << 8
1142 killdaemons(env['DAEMON_PIDS'])
1142 killdaemons(env['DAEMON_PIDS'])
1143 return ret
1143 return ret
1144
1144
1145 output = ''
1145 output = ''
1146 proc.tochild.close()
1146 proc.tochild.close()
1147
1147
1148 try:
1148 try:
1149 output = proc.fromchild.read()
1149 output = proc.fromchild.read()
1150 except KeyboardInterrupt:
1150 except KeyboardInterrupt:
1151 vlog('# Handling keyboard interrupt')
1151 vlog('# Handling keyboard interrupt')
1152 cleanup()
1152 cleanup()
1153 raise
1153 raise
1154
1154
1155 ret = proc.wait()
1155 ret = proc.wait()
1156 if wifexited(ret):
1156 if wifexited(ret):
1157 ret = os.WEXITSTATUS(ret)
1157 ret = os.WEXITSTATUS(ret)
1158
1158
1159 if proc.timeout:
1159 if proc.timeout:
1160 ret = 'timeout'
1160 ret = 'timeout'
1161
1161
1162 if ret:
1162 if ret:
1163 killdaemons(env['DAEMON_PIDS'])
1163 killdaemons(env['DAEMON_PIDS'])
1164
1164
1165 for s, r in self._getreplacements():
1165 for s, r in self._getreplacements():
1166 output = re.sub(s, r, output)
1166 output = re.sub(s, r, output)
1167
1167
1168 if normalizenewlines:
1168 if normalizenewlines:
1169 output = output.replace('\r\n', '\n')
1169 output = output.replace('\r\n', '\n')
1170
1170
1171 return ret, output.splitlines(True)
1171 return ret, output.splitlines(True)
1172
1172
1173 class PythonTest(Test):
1173 class PythonTest(Test):
1174 """A Python-based test."""
1174 """A Python-based test."""
1175
1175
1176 @property
1176 @property
1177 def refpath(self):
1177 def refpath(self):
1178 return os.path.join(self._testdir, b'%s.out' % self.bname)
1178 return os.path.join(self._testdir, b'%s.out' % self.bname)
1179
1179
1180 def _run(self, env):
1180 def _run(self, env):
1181 py3kswitch = self._py3kwarnings and b' -3' or b''
1181 py3kswitch = self._py3kwarnings and b' -3' or b''
1182 cmd = b'%s%s "%s"' % (PYTHON, py3kswitch, self.path)
1182 cmd = b'%s%s "%s"' % (PYTHON, py3kswitch, self.path)
1183 vlog("# Running", cmd)
1183 vlog("# Running", cmd)
1184 normalizenewlines = os.name == 'nt'
1184 normalizenewlines = os.name == 'nt'
1185 result = self._runcommand(cmd, env,
1185 result = self._runcommand(cmd, env,
1186 normalizenewlines=normalizenewlines)
1186 normalizenewlines=normalizenewlines)
1187 if self._aborted:
1187 if self._aborted:
1188 raise KeyboardInterrupt()
1188 raise KeyboardInterrupt()
1189
1189
1190 return result
1190 return result
1191
1191
1192 # Some glob patterns apply only in some circumstances, so the script
1192 # Some glob patterns apply only in some circumstances, so the script
1193 # might want to remove (glob) annotations that otherwise should be
1193 # might want to remove (glob) annotations that otherwise should be
1194 # retained.
1194 # retained.
1195 checkcodeglobpats = [
1195 checkcodeglobpats = [
1196 # On Windows it looks like \ doesn't require a (glob), but we know
1196 # On Windows it looks like \ doesn't require a (glob), but we know
1197 # better.
1197 # better.
1198 re.compile(br'^pushing to \$TESTTMP/.*[^)]$'),
1198 re.compile(br'^pushing to \$TESTTMP/.*[^)]$'),
1199 re.compile(br'^moving \S+/.*[^)]$'),
1199 re.compile(br'^moving \S+/.*[^)]$'),
1200 re.compile(br'^pulling from \$TESTTMP/.*[^)]$'),
1200 re.compile(br'^pulling from \$TESTTMP/.*[^)]$'),
1201 # Not all platforms have 127.0.0.1 as loopback (though most do),
1201 # Not all platforms have 127.0.0.1 as loopback (though most do),
1202 # so we always glob that too.
1202 # so we always glob that too.
1203 re.compile(br'.*\$LOCALIP.*$'),
1203 re.compile(br'.*\$LOCALIP.*$'),
1204 ]
1204 ]
1205
1205
1206 bchr = chr
1206 bchr = chr
1207 if PYTHON3:
1207 if PYTHON3:
1208 bchr = lambda x: bytes([x])
1208 bchr = lambda x: bytes([x])
1209
1209
1210 class TTest(Test):
1210 class TTest(Test):
1211 """A "t test" is a test backed by a .t file."""
1211 """A "t test" is a test backed by a .t file."""
1212
1212
1213 SKIPPED_PREFIX = b'skipped: '
1213 SKIPPED_PREFIX = b'skipped: '
1214 FAILED_PREFIX = b'hghave check failed: '
1214 FAILED_PREFIX = b'hghave check failed: '
1215 NEEDESCAPE = re.compile(br'[\x00-\x08\x0b-\x1f\x7f-\xff]').search
1215 NEEDESCAPE = re.compile(br'[\x00-\x08\x0b-\x1f\x7f-\xff]').search
1216
1216
1217 ESCAPESUB = re.compile(br'[\x00-\x08\x0b-\x1f\\\x7f-\xff]').sub
1217 ESCAPESUB = re.compile(br'[\x00-\x08\x0b-\x1f\\\x7f-\xff]').sub
1218 ESCAPEMAP = dict((bchr(i), br'\x%02x' % i) for i in range(256))
1218 ESCAPEMAP = dict((bchr(i), br'\x%02x' % i) for i in range(256))
1219 ESCAPEMAP.update({b'\\': b'\\\\', b'\r': br'\r'})
1219 ESCAPEMAP.update({b'\\': b'\\\\', b'\r': br'\r'})
1220
1220
1221 def __init__(self, path, *args, **kwds):
1221 def __init__(self, path, *args, **kwds):
1222 # accept an extra "case" parameter
1222 # accept an extra "case" parameter
1223 case = kwds.pop('case', None)
1223 case = kwds.pop('case', None)
1224 self._case = case
1224 self._case = case
1225 self._allcases = parsettestcases(path)
1225 self._allcases = parsettestcases(path)
1226 super(TTest, self).__init__(path, *args, **kwds)
1226 super(TTest, self).__init__(path, *args, **kwds)
1227 if case:
1227 if case:
1228 self.name = '%s (case %s)' % (self.name, _strpath(case))
1228 self.name = '%s (case %s)' % (self.name, _strpath(case))
1229 self.errpath = b'%s.%s.err' % (self.errpath[:-4], case)
1229 self.errpath = b'%s.%s.err' % (self.errpath[:-4], case)
1230 self._tmpname += b'-%s' % case
1230 self._tmpname += b'-%s' % case
1231
1231
1232 @property
1232 @property
1233 def refpath(self):
1233 def refpath(self):
1234 return os.path.join(self._testdir, self.bname)
1234 return os.path.join(self._testdir, self.bname)
1235
1235
1236 def _run(self, env):
1236 def _run(self, env):
1237 f = open(self.path, 'rb')
1237 f = open(self.path, 'rb')
1238 lines = f.readlines()
1238 lines = f.readlines()
1239 f.close()
1239 f.close()
1240
1240
1241 # .t file is both reference output and the test input, keep reference
1241 # .t file is both reference output and the test input, keep reference
1242 # output updated with the the test input. This avoids some race
1242 # output updated with the the test input. This avoids some race
1243 # conditions where the reference output does not match the actual test.
1243 # conditions where the reference output does not match the actual test.
1244 if self._refout is not None:
1244 if self._refout is not None:
1245 self._refout = lines
1245 self._refout = lines
1246
1246
1247 salt, script, after, expected = self._parsetest(lines)
1247 salt, script, after, expected = self._parsetest(lines)
1248
1248
1249 # Write out the generated script.
1249 # Write out the generated script.
1250 fname = b'%s.sh' % self._testtmp
1250 fname = b'%s.sh' % self._testtmp
1251 f = open(fname, 'wb')
1251 f = open(fname, 'wb')
1252 for l in script:
1252 for l in script:
1253 f.write(l)
1253 f.write(l)
1254 f.close()
1254 f.close()
1255
1255
1256 cmd = b'%s "%s"' % (self._shell, fname)
1256 cmd = b'%s "%s"' % (self._shell, fname)
1257 vlog("# Running", cmd)
1257 vlog("# Running", cmd)
1258
1258
1259 exitcode, output = self._runcommand(cmd, env)
1259 exitcode, output = self._runcommand(cmd, env)
1260
1260
1261 if self._aborted:
1261 if self._aborted:
1262 raise KeyboardInterrupt()
1262 raise KeyboardInterrupt()
1263
1263
1264 # Do not merge output if skipped. Return hghave message instead.
1264 # Do not merge output if skipped. Return hghave message instead.
1265 # Similarly, with --debug, output is None.
1265 # Similarly, with --debug, output is None.
1266 if exitcode == self.SKIPPED_STATUS or output is None:
1266 if exitcode == self.SKIPPED_STATUS or output is None:
1267 return exitcode, output
1267 return exitcode, output
1268
1268
1269 return self._processoutput(exitcode, output, salt, after, expected)
1269 return self._processoutput(exitcode, output, salt, after, expected)
1270
1270
1271 def _hghave(self, reqs):
1271 def _hghave(self, reqs):
1272 # TODO do something smarter when all other uses of hghave are gone.
1272 # TODO do something smarter when all other uses of hghave are gone.
1273 runtestdir = os.path.abspath(os.path.dirname(_bytespath(__file__)))
1273 runtestdir = os.path.abspath(os.path.dirname(_bytespath(__file__)))
1274 tdir = runtestdir.replace(b'\\', b'/')
1274 tdir = runtestdir.replace(b'\\', b'/')
1275 proc = Popen4(b'%s -c "%s/hghave %s"' %
1275 proc = Popen4(b'%s -c "%s/hghave %s"' %
1276 (self._shell, tdir, b' '.join(reqs)),
1276 (self._shell, tdir, b' '.join(reqs)),
1277 self._testtmp, 0, self._getenv())
1277 self._testtmp, 0, self._getenv())
1278 stdout, stderr = proc.communicate()
1278 stdout, stderr = proc.communicate()
1279 ret = proc.wait()
1279 ret = proc.wait()
1280 if wifexited(ret):
1280 if wifexited(ret):
1281 ret = os.WEXITSTATUS(ret)
1281 ret = os.WEXITSTATUS(ret)
1282 if ret == 2:
1282 if ret == 2:
1283 print(stdout.decode('utf-8'))
1283 print(stdout.decode('utf-8'))
1284 sys.exit(1)
1284 sys.exit(1)
1285
1285
1286 if ret != 0:
1286 if ret != 0:
1287 return False, stdout
1287 return False, stdout
1288
1288
1289 if b'slow' in reqs:
1289 if b'slow' in reqs:
1290 self._timeout = self._slowtimeout
1290 self._timeout = self._slowtimeout
1291 return True, None
1291 return True, None
1292
1292
1293 def _iftest(self, args):
1293 def _iftest(self, args):
1294 # implements "#if"
1294 # implements "#if"
1295 reqs = []
1295 reqs = []
1296 for arg in args:
1296 for arg in args:
1297 if arg.startswith(b'no-') and arg[3:] in self._allcases:
1297 if arg.startswith(b'no-') and arg[3:] in self._allcases:
1298 if arg[3:] == self._case:
1298 if arg[3:] == self._case:
1299 return False
1299 return False
1300 elif arg in self._allcases:
1300 elif arg in self._allcases:
1301 if arg != self._case:
1301 if arg != self._case:
1302 return False
1302 return False
1303 else:
1303 else:
1304 reqs.append(arg)
1304 reqs.append(arg)
1305 return self._hghave(reqs)[0]
1305 return self._hghave(reqs)[0]
1306
1306
1307 def _parsetest(self, lines):
1307 def _parsetest(self, lines):
1308 # We generate a shell script which outputs unique markers to line
1308 # We generate a shell script which outputs unique markers to line
1309 # up script results with our source. These markers include input
1309 # up script results with our source. These markers include input
1310 # line number and the last return code.
1310 # line number and the last return code.
1311 salt = b"SALT%d" % time.time()
1311 salt = b"SALT%d" % time.time()
1312 def addsalt(line, inpython):
1312 def addsalt(line, inpython):
1313 if inpython:
1313 if inpython:
1314 script.append(b'%s %d 0\n' % (salt, line))
1314 script.append(b'%s %d 0\n' % (salt, line))
1315 else:
1315 else:
1316 script.append(b'echo %s %d $?\n' % (salt, line))
1316 script.append(b'echo %s %d $?\n' % (salt, line))
1317
1317
1318 script = []
1318 script = []
1319
1319
1320 # After we run the shell script, we re-unify the script output
1320 # After we run the shell script, we re-unify the script output
1321 # with non-active parts of the source, with synchronization by our
1321 # with non-active parts of the source, with synchronization by our
1322 # SALT line number markers. The after table contains the non-active
1322 # SALT line number markers. The after table contains the non-active
1323 # components, ordered by line number.
1323 # components, ordered by line number.
1324 after = {}
1324 after = {}
1325
1325
1326 # Expected shell script output.
1326 # Expected shell script output.
1327 expected = {}
1327 expected = {}
1328
1328
1329 pos = prepos = -1
1329 pos = prepos = -1
1330
1330
1331 # True or False when in a true or false conditional section
1331 # True or False when in a true or false conditional section
1332 skipping = None
1332 skipping = None
1333
1333
1334 # We keep track of whether or not we're in a Python block so we
1334 # We keep track of whether or not we're in a Python block so we
1335 # can generate the surrounding doctest magic.
1335 # can generate the surrounding doctest magic.
1336 inpython = False
1336 inpython = False
1337
1337
1338 if self._debug:
1338 if self._debug:
1339 script.append(b'set -x\n')
1339 script.append(b'set -x\n')
1340 if self._hgcommand != b'hg':
1340 if self._hgcommand != b'hg':
1341 script.append(b'alias hg="%s"\n' % self._hgcommand)
1341 script.append(b'alias hg="%s"\n' % self._hgcommand)
1342 if os.getenv('MSYSTEM'):
1342 if os.getenv('MSYSTEM'):
1343 script.append(b'alias pwd="pwd -W"\n')
1343 script.append(b'alias pwd="pwd -W"\n')
1344
1344
1345 n = 0
1345 n = 0
1346 for n, l in enumerate(lines):
1346 for n, l in enumerate(lines):
1347 if not l.endswith(b'\n'):
1347 if not l.endswith(b'\n'):
1348 l += b'\n'
1348 l += b'\n'
1349 if l.startswith(b'#require'):
1349 if l.startswith(b'#require'):
1350 lsplit = l.split()
1350 lsplit = l.split()
1351 if len(lsplit) < 2 or lsplit[0] != b'#require':
1351 if len(lsplit) < 2 or lsplit[0] != b'#require':
1352 after.setdefault(pos, []).append(' !!! invalid #require\n')
1352 after.setdefault(pos, []).append(' !!! invalid #require\n')
1353 haveresult, message = self._hghave(lsplit[1:])
1353 haveresult, message = self._hghave(lsplit[1:])
1354 if not haveresult:
1354 if not haveresult:
1355 script = [b'echo "%s"\nexit 80\n' % message]
1355 script = [b'echo "%s"\nexit 80\n' % message]
1356 break
1356 break
1357 after.setdefault(pos, []).append(l)
1357 after.setdefault(pos, []).append(l)
1358 elif l.startswith(b'#if'):
1358 elif l.startswith(b'#if'):
1359 lsplit = l.split()
1359 lsplit = l.split()
1360 if len(lsplit) < 2 or lsplit[0] != b'#if':
1360 if len(lsplit) < 2 or lsplit[0] != b'#if':
1361 after.setdefault(pos, []).append(' !!! invalid #if\n')
1361 after.setdefault(pos, []).append(' !!! invalid #if\n')
1362 if skipping is not None:
1362 if skipping is not None:
1363 after.setdefault(pos, []).append(' !!! nested #if\n')
1363 after.setdefault(pos, []).append(' !!! nested #if\n')
1364 skipping = not self._iftest(lsplit[1:])
1364 skipping = not self._iftest(lsplit[1:])
1365 after.setdefault(pos, []).append(l)
1365 after.setdefault(pos, []).append(l)
1366 elif l.startswith(b'#else'):
1366 elif l.startswith(b'#else'):
1367 if skipping is None:
1367 if skipping is None:
1368 after.setdefault(pos, []).append(' !!! missing #if\n')
1368 after.setdefault(pos, []).append(' !!! missing #if\n')
1369 skipping = not skipping
1369 skipping = not skipping
1370 after.setdefault(pos, []).append(l)
1370 after.setdefault(pos, []).append(l)
1371 elif l.startswith(b'#endif'):
1371 elif l.startswith(b'#endif'):
1372 if skipping is None:
1372 if skipping is None:
1373 after.setdefault(pos, []).append(' !!! missing #if\n')
1373 after.setdefault(pos, []).append(' !!! missing #if\n')
1374 skipping = None
1374 skipping = None
1375 after.setdefault(pos, []).append(l)
1375 after.setdefault(pos, []).append(l)
1376 elif skipping:
1376 elif skipping:
1377 after.setdefault(pos, []).append(l)
1377 after.setdefault(pos, []).append(l)
1378 elif l.startswith(b' >>> '): # python inlines
1378 elif l.startswith(b' >>> '): # python inlines
1379 after.setdefault(pos, []).append(l)
1379 after.setdefault(pos, []).append(l)
1380 prepos = pos
1380 prepos = pos
1381 pos = n
1381 pos = n
1382 if not inpython:
1382 if not inpython:
1383 # We've just entered a Python block. Add the header.
1383 # We've just entered a Python block. Add the header.
1384 inpython = True
1384 inpython = True
1385 addsalt(prepos, False) # Make sure we report the exit code.
1385 addsalt(prepos, False) # Make sure we report the exit code.
1386 script.append(b'%s -m heredoctest <<EOF\n' % PYTHON)
1386 script.append(b'%s -m heredoctest <<EOF\n' % PYTHON)
1387 addsalt(n, True)
1387 addsalt(n, True)
1388 script.append(l[2:])
1388 script.append(l[2:])
1389 elif l.startswith(b' ... '): # python inlines
1389 elif l.startswith(b' ... '): # python inlines
1390 after.setdefault(prepos, []).append(l)
1390 after.setdefault(prepos, []).append(l)
1391 script.append(l[2:])
1391 script.append(l[2:])
1392 elif l.startswith(b' $ '): # commands
1392 elif l.startswith(b' $ '): # commands
1393 if inpython:
1393 if inpython:
1394 script.append(b'EOF\n')
1394 script.append(b'EOF\n')
1395 inpython = False
1395 inpython = False
1396 after.setdefault(pos, []).append(l)
1396 after.setdefault(pos, []).append(l)
1397 prepos = pos
1397 prepos = pos
1398 pos = n
1398 pos = n
1399 addsalt(n, False)
1399 addsalt(n, False)
1400 cmd = l[4:].split()
1400 cmd = l[4:].split()
1401 if len(cmd) == 2 and cmd[0] == b'cd':
1401 if len(cmd) == 2 and cmd[0] == b'cd':
1402 l = b' $ cd %s || exit 1\n' % cmd[1]
1402 l = b' $ cd %s || exit 1\n' % cmd[1]
1403 script.append(l[4:])
1403 script.append(l[4:])
1404 elif l.startswith(b' > '): # continuations
1404 elif l.startswith(b' > '): # continuations
1405 after.setdefault(prepos, []).append(l)
1405 after.setdefault(prepos, []).append(l)
1406 script.append(l[4:])
1406 script.append(l[4:])
1407 elif l.startswith(b' '): # results
1407 elif l.startswith(b' '): # results
1408 # Queue up a list of expected results.
1408 # Queue up a list of expected results.
1409 expected.setdefault(pos, []).append(l[2:])
1409 expected.setdefault(pos, []).append(l[2:])
1410 else:
1410 else:
1411 if inpython:
1411 if inpython:
1412 script.append(b'EOF\n')
1412 script.append(b'EOF\n')
1413 inpython = False
1413 inpython = False
1414 # Non-command/result. Queue up for merged output.
1414 # Non-command/result. Queue up for merged output.
1415 after.setdefault(pos, []).append(l)
1415 after.setdefault(pos, []).append(l)
1416
1416
1417 if inpython:
1417 if inpython:
1418 script.append(b'EOF\n')
1418 script.append(b'EOF\n')
1419 if skipping is not None:
1419 if skipping is not None:
1420 after.setdefault(pos, []).append(' !!! missing #endif\n')
1420 after.setdefault(pos, []).append(' !!! missing #endif\n')
1421 addsalt(n + 1, False)
1421 addsalt(n + 1, False)
1422
1422
1423 return salt, script, after, expected
1423 return salt, script, after, expected
1424
1424
1425 def _processoutput(self, exitcode, output, salt, after, expected):
1425 def _processoutput(self, exitcode, output, salt, after, expected):
1426 # Merge the script output back into a unified test.
1426 # Merge the script output back into a unified test.
1427 warnonly = 1 # 1: not yet; 2: yes; 3: for sure not
1427 warnonly = 1 # 1: not yet; 2: yes; 3: for sure not
1428 if exitcode != 0:
1428 if exitcode != 0:
1429 warnonly = 3
1429 warnonly = 3
1430
1430
1431 pos = -1
1431 pos = -1
1432 postout = []
1432 postout = []
1433 for l in output:
1433 for l in output:
1434 lout, lcmd = l, None
1434 lout, lcmd = l, None
1435 if salt in l:
1435 if salt in l:
1436 lout, lcmd = l.split(salt, 1)
1436 lout, lcmd = l.split(salt, 1)
1437
1437
1438 while lout:
1438 while lout:
1439 if not lout.endswith(b'\n'):
1439 if not lout.endswith(b'\n'):
1440 lout += b' (no-eol)\n'
1440 lout += b' (no-eol)\n'
1441
1441
1442 # Find the expected output at the current position.
1442 # Find the expected output at the current position.
1443 els = [None]
1443 els = [None]
1444 if expected.get(pos, None):
1444 if expected.get(pos, None):
1445 els = expected[pos]
1445 els = expected[pos]
1446
1446
1447 i = 0
1447 i = 0
1448 optional = []
1448 optional = []
1449 while i < len(els):
1449 while i < len(els):
1450 el = els[i]
1450 el = els[i]
1451
1451
1452 r = self.linematch(el, lout)
1452 r = self.linematch(el, lout)
1453 if isinstance(r, str):
1453 if isinstance(r, str):
1454 if r == '+glob':
1454 if r == '-glob':
1455 lout = el[:-1] + ' (glob)\n'
1456 r = '' # Warn only this line.
1457 elif r == '-glob':
1458 lout = ''.join(el.rsplit(' (glob)', 1))
1455 lout = ''.join(el.rsplit(' (glob)', 1))
1459 r = '' # Warn only this line.
1456 r = '' # Warn only this line.
1460 elif r == "retry":
1457 elif r == "retry":
1461 postout.append(b' ' + el)
1458 postout.append(b' ' + el)
1462 els.pop(i)
1459 els.pop(i)
1463 break
1460 break
1464 else:
1461 else:
1465 log('\ninfo, unknown linematch result: %r\n' % r)
1462 log('\ninfo, unknown linematch result: %r\n' % r)
1466 r = False
1463 r = False
1467 if r:
1464 if r:
1468 els.pop(i)
1465 els.pop(i)
1469 break
1466 break
1470 if el:
1467 if el:
1471 if el.endswith(b" (?)\n"):
1468 if el.endswith(b" (?)\n"):
1472 optional.append(i)
1469 optional.append(i)
1473 else:
1470 else:
1474 m = optline.match(el)
1471 m = optline.match(el)
1475 if m:
1472 if m:
1476 conditions = [
1473 conditions = [
1477 c for c in m.group(2).split(b' ')]
1474 c for c in m.group(2).split(b' ')]
1478
1475
1479 if not self._iftest(conditions):
1476 if not self._iftest(conditions):
1480 optional.append(i)
1477 optional.append(i)
1481
1478
1482 i += 1
1479 i += 1
1483
1480
1484 if r:
1481 if r:
1485 if r == "retry":
1482 if r == "retry":
1486 continue
1483 continue
1487 # clean up any optional leftovers
1484 # clean up any optional leftovers
1488 for i in optional:
1485 for i in optional:
1489 postout.append(b' ' + els[i])
1486 postout.append(b' ' + els[i])
1490 for i in reversed(optional):
1487 for i in reversed(optional):
1491 del els[i]
1488 del els[i]
1492 postout.append(b' ' + el)
1489 postout.append(b' ' + el)
1493 else:
1490 else:
1494 if self.NEEDESCAPE(lout):
1491 if self.NEEDESCAPE(lout):
1495 lout = TTest._stringescape(b'%s (esc)\n' %
1492 lout = TTest._stringescape(b'%s (esc)\n' %
1496 lout.rstrip(b'\n'))
1493 lout.rstrip(b'\n'))
1497 postout.append(b' ' + lout) # Let diff deal with it.
1494 postout.append(b' ' + lout) # Let diff deal with it.
1498 if r != '': # If line failed.
1495 if r != '': # If line failed.
1499 warnonly = 3 # for sure not
1496 warnonly = 3 # for sure not
1500 elif warnonly == 1: # Is "not yet" and line is warn only.
1497 elif warnonly == 1: # Is "not yet" and line is warn only.
1501 warnonly = 2 # Yes do warn.
1498 warnonly = 2 # Yes do warn.
1502 break
1499 break
1503 else:
1500 else:
1504 # clean up any optional leftovers
1501 # clean up any optional leftovers
1505 while expected.get(pos, None):
1502 while expected.get(pos, None):
1506 el = expected[pos].pop(0)
1503 el = expected[pos].pop(0)
1507 if el:
1504 if el:
1508 if not el.endswith(b" (?)\n"):
1505 if not el.endswith(b" (?)\n"):
1509 m = optline.match(el)
1506 m = optline.match(el)
1510 if m:
1507 if m:
1511 conditions = [c for c in m.group(2).split(b' ')]
1508 conditions = [c for c in m.group(2).split(b' ')]
1512
1509
1513 if self._iftest(conditions):
1510 if self._iftest(conditions):
1514 # Don't append as optional line
1511 # Don't append as optional line
1515 continue
1512 continue
1516 else:
1513 else:
1517 continue
1514 continue
1518 postout.append(b' ' + el)
1515 postout.append(b' ' + el)
1519
1516
1520 if lcmd:
1517 if lcmd:
1521 # Add on last return code.
1518 # Add on last return code.
1522 ret = int(lcmd.split()[1])
1519 ret = int(lcmd.split()[1])
1523 if ret != 0:
1520 if ret != 0:
1524 postout.append(b' [%d]\n' % ret)
1521 postout.append(b' [%d]\n' % ret)
1525 if pos in after:
1522 if pos in after:
1526 # Merge in non-active test bits.
1523 # Merge in non-active test bits.
1527 postout += after.pop(pos)
1524 postout += after.pop(pos)
1528 pos = int(lcmd.split()[0])
1525 pos = int(lcmd.split()[0])
1529
1526
1530 if pos in after:
1527 if pos in after:
1531 postout += after.pop(pos)
1528 postout += after.pop(pos)
1532
1529
1533 if warnonly == 2:
1530 if warnonly == 2:
1534 exitcode = False # Set exitcode to warned.
1531 exitcode = False # Set exitcode to warned.
1535
1532
1536 return exitcode, postout
1533 return exitcode, postout
1537
1534
1538 @staticmethod
1535 @staticmethod
1539 def rematch(el, l):
1536 def rematch(el, l):
1540 try:
1537 try:
1541 el = b'(?:' + el + b')'
1538 el = b'(?:' + el + b')'
1542 # use \Z to ensure that the regex matches to the end of the string
1539 # use \Z to ensure that the regex matches to the end of the string
1543 if os.name == 'nt':
1540 if os.name == 'nt':
1544 return re.match(el + br'\r?\n\Z', l)
1541 return re.match(el + br'\r?\n\Z', l)
1545 return re.match(el + br'\n\Z', l)
1542 return re.match(el + br'\n\Z', l)
1546 except re.error:
1543 except re.error:
1547 # el is an invalid regex
1544 # el is an invalid regex
1548 return False
1545 return False
1549
1546
1550 @staticmethod
1547 @staticmethod
1551 def globmatch(el, l):
1548 def globmatch(el, l):
1552 # The only supported special characters are * and ? plus / which also
1549 # The only supported special characters are * and ? plus / which also
1553 # matches \ on windows. Escaping of these characters is supported.
1550 # matches \ on windows. Escaping of these characters is supported.
1554 if el + b'\n' == l:
1551 if el + b'\n' == l:
1555 if os.altsep:
1552 if os.altsep:
1556 # matching on "/" is not needed for this line
1553 # matching on "/" is not needed for this line
1557 for pat in checkcodeglobpats:
1554 for pat in checkcodeglobpats:
1558 if pat.match(el):
1555 if pat.match(el):
1559 return True
1556 return True
1560 return b'-glob'
1557 return b'-glob'
1561 return True
1558 return True
1562 el = el.replace(b'$LOCALIP', b'*')
1559 el = el.replace(b'$LOCALIP', b'*')
1563 i, n = 0, len(el)
1560 i, n = 0, len(el)
1564 res = b''
1561 res = b''
1565 while i < n:
1562 while i < n:
1566 c = el[i:i + 1]
1563 c = el[i:i + 1]
1567 i += 1
1564 i += 1
1568 if c == b'\\' and i < n and el[i:i + 1] in b'*?\\/':
1565 if c == b'\\' and i < n and el[i:i + 1] in b'*?\\/':
1569 res += el[i - 1:i + 1]
1566 res += el[i - 1:i + 1]
1570 i += 1
1567 i += 1
1571 elif c == b'*':
1568 elif c == b'*':
1572 res += b'.*'
1569 res += b'.*'
1573 elif c == b'?':
1570 elif c == b'?':
1574 res += b'.'
1571 res += b'.'
1575 elif c == b'/' and os.altsep:
1572 elif c == b'/' and os.altsep:
1576 res += b'[/\\\\]'
1573 res += b'[/\\\\]'
1577 else:
1574 else:
1578 res += re.escape(c)
1575 res += re.escape(c)
1579 return TTest.rematch(res, l)
1576 return TTest.rematch(res, l)
1580
1577
1581 def linematch(self, el, l):
1578 def linematch(self, el, l):
1582 retry = False
1579 retry = False
1583 if el == l: # perfect match (fast)
1580 if el == l: # perfect match (fast)
1584 return True
1581 return True
1585 if el:
1582 if el:
1586 if el.endswith(b" (?)\n"):
1583 if el.endswith(b" (?)\n"):
1587 retry = "retry"
1584 retry = "retry"
1588 el = el[:-5] + b"\n"
1585 el = el[:-5] + b"\n"
1589 else:
1586 else:
1590 m = optline.match(el)
1587 m = optline.match(el)
1591 if m:
1588 if m:
1592 conditions = [c for c in m.group(2).split(b' ')]
1589 conditions = [c for c in m.group(2).split(b' ')]
1593
1590
1594 el = m.group(1) + b"\n"
1591 el = m.group(1) + b"\n"
1595 if not self._iftest(conditions):
1592 if not self._iftest(conditions):
1596 retry = "retry" # Not required by listed features
1593 retry = "retry" # Not required by listed features
1597
1594
1598 if el.endswith(b" (esc)\n"):
1595 if el.endswith(b" (esc)\n"):
1599 if PYTHON3:
1596 if PYTHON3:
1600 el = el[:-7].decode('unicode_escape') + '\n'
1597 el = el[:-7].decode('unicode_escape') + '\n'
1601 el = el.encode('utf-8')
1598 el = el.encode('utf-8')
1602 else:
1599 else:
1603 el = el[:-7].decode('string-escape') + '\n'
1600 el = el[:-7].decode('string-escape') + '\n'
1604 if el == l or os.name == 'nt' and el[:-1] + b'\r\n' == l:
1601 if el == l or os.name == 'nt' and el[:-1] + b'\r\n' == l:
1605 return True
1602 return True
1606 if el.endswith(b" (re)\n"):
1603 if el.endswith(b" (re)\n"):
1607 return TTest.rematch(el[:-6], l) or retry
1604 return TTest.rematch(el[:-6], l) or retry
1608 if el.endswith(b" (glob)\n"):
1605 if el.endswith(b" (glob)\n"):
1609 # ignore '(glob)' added to l by 'replacements'
1606 # ignore '(glob)' added to l by 'replacements'
1610 if l.endswith(b" (glob)\n"):
1607 if l.endswith(b" (glob)\n"):
1611 l = l[:-8] + b"\n"
1608 l = l[:-8] + b"\n"
1612 return TTest.globmatch(el[:-8], l) or retry
1609 return TTest.globmatch(el[:-8], l) or retry
1613 if os.altsep:
1610 if os.altsep:
1614 _l = l.replace(b'\\', b'/')
1611 _l = l.replace(b'\\', b'/')
1615 if el == _l or os.name == 'nt' and el[:-1] + b'\r\n' == _l:
1612 if el == _l or os.name == 'nt' and el[:-1] + b'\r\n' == _l:
1616 return b'+glob'
1613 return True
1617 return retry
1614 return retry
1618
1615
1619 @staticmethod
1616 @staticmethod
1620 def parsehghaveoutput(lines):
1617 def parsehghaveoutput(lines):
1621 '''Parse hghave log lines.
1618 '''Parse hghave log lines.
1622
1619
1623 Return tuple of lists (missing, failed):
1620 Return tuple of lists (missing, failed):
1624 * the missing/unknown features
1621 * the missing/unknown features
1625 * the features for which existence check failed'''
1622 * the features for which existence check failed'''
1626 missing = []
1623 missing = []
1627 failed = []
1624 failed = []
1628 for line in lines:
1625 for line in lines:
1629 if line.startswith(TTest.SKIPPED_PREFIX):
1626 if line.startswith(TTest.SKIPPED_PREFIX):
1630 line = line.splitlines()[0]
1627 line = line.splitlines()[0]
1631 missing.append(line[len(TTest.SKIPPED_PREFIX):].decode('utf-8'))
1628 missing.append(line[len(TTest.SKIPPED_PREFIX):].decode('utf-8'))
1632 elif line.startswith(TTest.FAILED_PREFIX):
1629 elif line.startswith(TTest.FAILED_PREFIX):
1633 line = line.splitlines()[0]
1630 line = line.splitlines()[0]
1634 failed.append(line[len(TTest.FAILED_PREFIX):].decode('utf-8'))
1631 failed.append(line[len(TTest.FAILED_PREFIX):].decode('utf-8'))
1635
1632
1636 return missing, failed
1633 return missing, failed
1637
1634
1638 @staticmethod
1635 @staticmethod
1639 def _escapef(m):
1636 def _escapef(m):
1640 return TTest.ESCAPEMAP[m.group(0)]
1637 return TTest.ESCAPEMAP[m.group(0)]
1641
1638
1642 @staticmethod
1639 @staticmethod
1643 def _stringescape(s):
1640 def _stringescape(s):
1644 return TTest.ESCAPESUB(TTest._escapef, s)
1641 return TTest.ESCAPESUB(TTest._escapef, s)
1645
1642
1646 iolock = threading.RLock()
1643 iolock = threading.RLock()
1647
1644
1648 class TestResult(unittest._TextTestResult):
1645 class TestResult(unittest._TextTestResult):
1649 """Holds results when executing via unittest."""
1646 """Holds results when executing via unittest."""
1650 # Don't worry too much about accessing the non-public _TextTestResult.
1647 # Don't worry too much about accessing the non-public _TextTestResult.
1651 # It is relatively common in Python testing tools.
1648 # It is relatively common in Python testing tools.
1652 def __init__(self, options, *args, **kwargs):
1649 def __init__(self, options, *args, **kwargs):
1653 super(TestResult, self).__init__(*args, **kwargs)
1650 super(TestResult, self).__init__(*args, **kwargs)
1654
1651
1655 self._options = options
1652 self._options = options
1656
1653
1657 # unittest.TestResult didn't have skipped until 2.7. We need to
1654 # unittest.TestResult didn't have skipped until 2.7. We need to
1658 # polyfill it.
1655 # polyfill it.
1659 self.skipped = []
1656 self.skipped = []
1660
1657
1661 # We have a custom "ignored" result that isn't present in any Python
1658 # We have a custom "ignored" result that isn't present in any Python
1662 # unittest implementation. It is very similar to skipped. It may make
1659 # unittest implementation. It is very similar to skipped. It may make
1663 # sense to map it into skip some day.
1660 # sense to map it into skip some day.
1664 self.ignored = []
1661 self.ignored = []
1665
1662
1666 self.times = []
1663 self.times = []
1667 self._firststarttime = None
1664 self._firststarttime = None
1668 # Data stored for the benefit of generating xunit reports.
1665 # Data stored for the benefit of generating xunit reports.
1669 self.successes = []
1666 self.successes = []
1670 self.faildata = {}
1667 self.faildata = {}
1671
1668
1672 if options.color == 'auto':
1669 if options.color == 'auto':
1673 self.color = pygmentspresent and self.stream.isatty()
1670 self.color = pygmentspresent and self.stream.isatty()
1674 elif options.color == 'never':
1671 elif options.color == 'never':
1675 self.color = False
1672 self.color = False
1676 else: # 'always', for testing purposes
1673 else: # 'always', for testing purposes
1677 self.color = pygmentspresent
1674 self.color = pygmentspresent
1678
1675
1679 def addFailure(self, test, reason):
1676 def addFailure(self, test, reason):
1680 self.failures.append((test, reason))
1677 self.failures.append((test, reason))
1681
1678
1682 if self._options.first:
1679 if self._options.first:
1683 self.stop()
1680 self.stop()
1684 else:
1681 else:
1685 with iolock:
1682 with iolock:
1686 if reason == "timed out":
1683 if reason == "timed out":
1687 self.stream.write('t')
1684 self.stream.write('t')
1688 else:
1685 else:
1689 if not self._options.nodiff:
1686 if not self._options.nodiff:
1690 self.stream.write('\n')
1687 self.stream.write('\n')
1691 # Exclude the '\n' from highlighting to lex correctly
1688 # Exclude the '\n' from highlighting to lex correctly
1692 formatted = 'ERROR: %s output changed\n' % test
1689 formatted = 'ERROR: %s output changed\n' % test
1693 self.stream.write(highlightmsg(formatted, self.color))
1690 self.stream.write(highlightmsg(formatted, self.color))
1694 self.stream.write('!')
1691 self.stream.write('!')
1695
1692
1696 self.stream.flush()
1693 self.stream.flush()
1697
1694
1698 def addSuccess(self, test):
1695 def addSuccess(self, test):
1699 with iolock:
1696 with iolock:
1700 super(TestResult, self).addSuccess(test)
1697 super(TestResult, self).addSuccess(test)
1701 self.successes.append(test)
1698 self.successes.append(test)
1702
1699
1703 def addError(self, test, err):
1700 def addError(self, test, err):
1704 super(TestResult, self).addError(test, err)
1701 super(TestResult, self).addError(test, err)
1705 if self._options.first:
1702 if self._options.first:
1706 self.stop()
1703 self.stop()
1707
1704
1708 # Polyfill.
1705 # Polyfill.
1709 def addSkip(self, test, reason):
1706 def addSkip(self, test, reason):
1710 self.skipped.append((test, reason))
1707 self.skipped.append((test, reason))
1711 with iolock:
1708 with iolock:
1712 if self.showAll:
1709 if self.showAll:
1713 self.stream.writeln('skipped %s' % reason)
1710 self.stream.writeln('skipped %s' % reason)
1714 else:
1711 else:
1715 self.stream.write('s')
1712 self.stream.write('s')
1716 self.stream.flush()
1713 self.stream.flush()
1717
1714
1718 def addIgnore(self, test, reason):
1715 def addIgnore(self, test, reason):
1719 self.ignored.append((test, reason))
1716 self.ignored.append((test, reason))
1720 with iolock:
1717 with iolock:
1721 if self.showAll:
1718 if self.showAll:
1722 self.stream.writeln('ignored %s' % reason)
1719 self.stream.writeln('ignored %s' % reason)
1723 else:
1720 else:
1724 if reason not in ('not retesting', "doesn't match keyword"):
1721 if reason not in ('not retesting', "doesn't match keyword"):
1725 self.stream.write('i')
1722 self.stream.write('i')
1726 else:
1723 else:
1727 self.testsRun += 1
1724 self.testsRun += 1
1728 self.stream.flush()
1725 self.stream.flush()
1729
1726
1730 def addOutputMismatch(self, test, ret, got, expected):
1727 def addOutputMismatch(self, test, ret, got, expected):
1731 """Record a mismatch in test output for a particular test."""
1728 """Record a mismatch in test output for a particular test."""
1732 if self.shouldStop:
1729 if self.shouldStop:
1733 # don't print, some other test case already failed and
1730 # don't print, some other test case already failed and
1734 # printed, we're just stale and probably failed due to our
1731 # printed, we're just stale and probably failed due to our
1735 # temp dir getting cleaned up.
1732 # temp dir getting cleaned up.
1736 return
1733 return
1737
1734
1738 accepted = False
1735 accepted = False
1739 lines = []
1736 lines = []
1740
1737
1741 with iolock:
1738 with iolock:
1742 if self._options.nodiff:
1739 if self._options.nodiff:
1743 pass
1740 pass
1744 elif self._options.view:
1741 elif self._options.view:
1745 v = self._options.view
1742 v = self._options.view
1746 if PYTHON3:
1743 if PYTHON3:
1747 v = _bytespath(v)
1744 v = _bytespath(v)
1748 os.system(b"%s %s %s" %
1745 os.system(b"%s %s %s" %
1749 (v, test.refpath, test.errpath))
1746 (v, test.refpath, test.errpath))
1750 else:
1747 else:
1751 servefail, lines = getdiff(expected, got,
1748 servefail, lines = getdiff(expected, got,
1752 test.refpath, test.errpath)
1749 test.refpath, test.errpath)
1753 if servefail:
1750 if servefail:
1754 raise test.failureException(
1751 raise test.failureException(
1755 'server failed to start (HGPORT=%s)' % test._startport)
1752 'server failed to start (HGPORT=%s)' % test._startport)
1756 else:
1753 else:
1757 self.stream.write('\n')
1754 self.stream.write('\n')
1758 for line in lines:
1755 for line in lines:
1759 line = highlightdiff(line, self.color)
1756 line = highlightdiff(line, self.color)
1760 if PYTHON3:
1757 if PYTHON3:
1761 self.stream.flush()
1758 self.stream.flush()
1762 self.stream.buffer.write(line)
1759 self.stream.buffer.write(line)
1763 self.stream.buffer.flush()
1760 self.stream.buffer.flush()
1764 else:
1761 else:
1765 self.stream.write(line)
1762 self.stream.write(line)
1766 self.stream.flush()
1763 self.stream.flush()
1767
1764
1768 # handle interactive prompt without releasing iolock
1765 # handle interactive prompt without releasing iolock
1769 if self._options.interactive:
1766 if self._options.interactive:
1770 if test.readrefout() != expected:
1767 if test.readrefout() != expected:
1771 self.stream.write(
1768 self.stream.write(
1772 'Reference output has changed (run again to prompt '
1769 'Reference output has changed (run again to prompt '
1773 'changes)')
1770 'changes)')
1774 else:
1771 else:
1775 self.stream.write('Accept this change? [n] ')
1772 self.stream.write('Accept this change? [n] ')
1776 answer = sys.stdin.readline().strip()
1773 answer = sys.stdin.readline().strip()
1777 if answer.lower() in ('y', 'yes'):
1774 if answer.lower() in ('y', 'yes'):
1778 if test.path.endswith(b'.t'):
1775 if test.path.endswith(b'.t'):
1779 rename(test.errpath, test.path)
1776 rename(test.errpath, test.path)
1780 else:
1777 else:
1781 rename(test.errpath, '%s.out' % test.path)
1778 rename(test.errpath, '%s.out' % test.path)
1782 accepted = True
1779 accepted = True
1783 if not accepted:
1780 if not accepted:
1784 self.faildata[test.name] = b''.join(lines)
1781 self.faildata[test.name] = b''.join(lines)
1785
1782
1786 return accepted
1783 return accepted
1787
1784
1788 def startTest(self, test):
1785 def startTest(self, test):
1789 super(TestResult, self).startTest(test)
1786 super(TestResult, self).startTest(test)
1790
1787
1791 # os.times module computes the user time and system time spent by
1788 # os.times module computes the user time and system time spent by
1792 # child's processes along with real elapsed time taken by a process.
1789 # child's processes along with real elapsed time taken by a process.
1793 # This module has one limitation. It can only work for Linux user
1790 # This module has one limitation. It can only work for Linux user
1794 # and not for Windows.
1791 # and not for Windows.
1795 test.started = os.times()
1792 test.started = os.times()
1796 if self._firststarttime is None: # thread racy but irrelevant
1793 if self._firststarttime is None: # thread racy but irrelevant
1797 self._firststarttime = test.started[4]
1794 self._firststarttime = test.started[4]
1798
1795
1799 def stopTest(self, test, interrupted=False):
1796 def stopTest(self, test, interrupted=False):
1800 super(TestResult, self).stopTest(test)
1797 super(TestResult, self).stopTest(test)
1801
1798
1802 test.stopped = os.times()
1799 test.stopped = os.times()
1803
1800
1804 starttime = test.started
1801 starttime = test.started
1805 endtime = test.stopped
1802 endtime = test.stopped
1806 origin = self._firststarttime
1803 origin = self._firststarttime
1807 self.times.append((test.name,
1804 self.times.append((test.name,
1808 endtime[2] - starttime[2], # user space CPU time
1805 endtime[2] - starttime[2], # user space CPU time
1809 endtime[3] - starttime[3], # sys space CPU time
1806 endtime[3] - starttime[3], # sys space CPU time
1810 endtime[4] - starttime[4], # real time
1807 endtime[4] - starttime[4], # real time
1811 starttime[4] - origin, # start date in run context
1808 starttime[4] - origin, # start date in run context
1812 endtime[4] - origin, # end date in run context
1809 endtime[4] - origin, # end date in run context
1813 ))
1810 ))
1814
1811
1815 if interrupted:
1812 if interrupted:
1816 with iolock:
1813 with iolock:
1817 self.stream.writeln('INTERRUPTED: %s (after %d seconds)' % (
1814 self.stream.writeln('INTERRUPTED: %s (after %d seconds)' % (
1818 test.name, self.times[-1][3]))
1815 test.name, self.times[-1][3]))
1819
1816
1820 class TestSuite(unittest.TestSuite):
1817 class TestSuite(unittest.TestSuite):
1821 """Custom unittest TestSuite that knows how to execute Mercurial tests."""
1818 """Custom unittest TestSuite that knows how to execute Mercurial tests."""
1822
1819
1823 def __init__(self, testdir, jobs=1, whitelist=None, blacklist=None,
1820 def __init__(self, testdir, jobs=1, whitelist=None, blacklist=None,
1824 retest=False, keywords=None, loop=False, runs_per_test=1,
1821 retest=False, keywords=None, loop=False, runs_per_test=1,
1825 loadtest=None, showchannels=False,
1822 loadtest=None, showchannels=False,
1826 *args, **kwargs):
1823 *args, **kwargs):
1827 """Create a new instance that can run tests with a configuration.
1824 """Create a new instance that can run tests with a configuration.
1828
1825
1829 testdir specifies the directory where tests are executed from. This
1826 testdir specifies the directory where tests are executed from. This
1830 is typically the ``tests`` directory from Mercurial's source
1827 is typically the ``tests`` directory from Mercurial's source
1831 repository.
1828 repository.
1832
1829
1833 jobs specifies the number of jobs to run concurrently. Each test
1830 jobs specifies the number of jobs to run concurrently. Each test
1834 executes on its own thread. Tests actually spawn new processes, so
1831 executes on its own thread. Tests actually spawn new processes, so
1835 state mutation should not be an issue.
1832 state mutation should not be an issue.
1836
1833
1837 If there is only one job, it will use the main thread.
1834 If there is only one job, it will use the main thread.
1838
1835
1839 whitelist and blacklist denote tests that have been whitelisted and
1836 whitelist and blacklist denote tests that have been whitelisted and
1840 blacklisted, respectively. These arguments don't belong in TestSuite.
1837 blacklisted, respectively. These arguments don't belong in TestSuite.
1841 Instead, whitelist and blacklist should be handled by the thing that
1838 Instead, whitelist and blacklist should be handled by the thing that
1842 populates the TestSuite with tests. They are present to preserve
1839 populates the TestSuite with tests. They are present to preserve
1843 backwards compatible behavior which reports skipped tests as part
1840 backwards compatible behavior which reports skipped tests as part
1844 of the results.
1841 of the results.
1845
1842
1846 retest denotes whether to retest failed tests. This arguably belongs
1843 retest denotes whether to retest failed tests. This arguably belongs
1847 outside of TestSuite.
1844 outside of TestSuite.
1848
1845
1849 keywords denotes key words that will be used to filter which tests
1846 keywords denotes key words that will be used to filter which tests
1850 to execute. This arguably belongs outside of TestSuite.
1847 to execute. This arguably belongs outside of TestSuite.
1851
1848
1852 loop denotes whether to loop over tests forever.
1849 loop denotes whether to loop over tests forever.
1853 """
1850 """
1854 super(TestSuite, self).__init__(*args, **kwargs)
1851 super(TestSuite, self).__init__(*args, **kwargs)
1855
1852
1856 self._jobs = jobs
1853 self._jobs = jobs
1857 self._whitelist = whitelist
1854 self._whitelist = whitelist
1858 self._blacklist = blacklist
1855 self._blacklist = blacklist
1859 self._retest = retest
1856 self._retest = retest
1860 self._keywords = keywords
1857 self._keywords = keywords
1861 self._loop = loop
1858 self._loop = loop
1862 self._runs_per_test = runs_per_test
1859 self._runs_per_test = runs_per_test
1863 self._loadtest = loadtest
1860 self._loadtest = loadtest
1864 self._showchannels = showchannels
1861 self._showchannels = showchannels
1865
1862
1866 def run(self, result):
1863 def run(self, result):
1867 # We have a number of filters that need to be applied. We do this
1864 # We have a number of filters that need to be applied. We do this
1868 # here instead of inside Test because it makes the running logic for
1865 # here instead of inside Test because it makes the running logic for
1869 # Test simpler.
1866 # Test simpler.
1870 tests = []
1867 tests = []
1871 num_tests = [0]
1868 num_tests = [0]
1872 for test in self._tests:
1869 for test in self._tests:
1873 def get():
1870 def get():
1874 num_tests[0] += 1
1871 num_tests[0] += 1
1875 if getattr(test, 'should_reload', False):
1872 if getattr(test, 'should_reload', False):
1876 return self._loadtest(test, num_tests[0])
1873 return self._loadtest(test, num_tests[0])
1877 return test
1874 return test
1878 if not os.path.exists(test.path):
1875 if not os.path.exists(test.path):
1879 result.addSkip(test, "Doesn't exist")
1876 result.addSkip(test, "Doesn't exist")
1880 continue
1877 continue
1881
1878
1882 if not (self._whitelist and test.bname in self._whitelist):
1879 if not (self._whitelist and test.bname in self._whitelist):
1883 if self._blacklist and test.bname in self._blacklist:
1880 if self._blacklist and test.bname in self._blacklist:
1884 result.addSkip(test, 'blacklisted')
1881 result.addSkip(test, 'blacklisted')
1885 continue
1882 continue
1886
1883
1887 if self._retest and not os.path.exists(test.errpath):
1884 if self._retest and not os.path.exists(test.errpath):
1888 result.addIgnore(test, 'not retesting')
1885 result.addIgnore(test, 'not retesting')
1889 continue
1886 continue
1890
1887
1891 if self._keywords:
1888 if self._keywords:
1892 f = open(test.path, 'rb')
1889 f = open(test.path, 'rb')
1893 t = f.read().lower() + test.bname.lower()
1890 t = f.read().lower() + test.bname.lower()
1894 f.close()
1891 f.close()
1895 ignored = False
1892 ignored = False
1896 for k in self._keywords.lower().split():
1893 for k in self._keywords.lower().split():
1897 if k not in t:
1894 if k not in t:
1898 result.addIgnore(test, "doesn't match keyword")
1895 result.addIgnore(test, "doesn't match keyword")
1899 ignored = True
1896 ignored = True
1900 break
1897 break
1901
1898
1902 if ignored:
1899 if ignored:
1903 continue
1900 continue
1904 for _ in xrange(self._runs_per_test):
1901 for _ in xrange(self._runs_per_test):
1905 tests.append(get())
1902 tests.append(get())
1906
1903
1907 runtests = list(tests)
1904 runtests = list(tests)
1908 done = queue.Queue()
1905 done = queue.Queue()
1909 running = 0
1906 running = 0
1910
1907
1911 channels = [""] * self._jobs
1908 channels = [""] * self._jobs
1912
1909
1913 def job(test, result):
1910 def job(test, result):
1914 for n, v in enumerate(channels):
1911 for n, v in enumerate(channels):
1915 if not v:
1912 if not v:
1916 channel = n
1913 channel = n
1917 break
1914 break
1918 else:
1915 else:
1919 raise ValueError('Could not find output channel')
1916 raise ValueError('Could not find output channel')
1920 channels[channel] = "=" + test.name[5:].split(".")[0]
1917 channels[channel] = "=" + test.name[5:].split(".")[0]
1921 try:
1918 try:
1922 test(result)
1919 test(result)
1923 done.put(None)
1920 done.put(None)
1924 except KeyboardInterrupt:
1921 except KeyboardInterrupt:
1925 pass
1922 pass
1926 except: # re-raises
1923 except: # re-raises
1927 done.put(('!', test, 'run-test raised an error, see traceback'))
1924 done.put(('!', test, 'run-test raised an error, see traceback'))
1928 raise
1925 raise
1929 finally:
1926 finally:
1930 try:
1927 try:
1931 channels[channel] = ''
1928 channels[channel] = ''
1932 except IndexError:
1929 except IndexError:
1933 pass
1930 pass
1934
1931
1935 def stat():
1932 def stat():
1936 count = 0
1933 count = 0
1937 while channels:
1934 while channels:
1938 d = '\n%03s ' % count
1935 d = '\n%03s ' % count
1939 for n, v in enumerate(channels):
1936 for n, v in enumerate(channels):
1940 if v:
1937 if v:
1941 d += v[0]
1938 d += v[0]
1942 channels[n] = v[1:] or '.'
1939 channels[n] = v[1:] or '.'
1943 else:
1940 else:
1944 d += ' '
1941 d += ' '
1945 d += ' '
1942 d += ' '
1946 with iolock:
1943 with iolock:
1947 sys.stdout.write(d + ' ')
1944 sys.stdout.write(d + ' ')
1948 sys.stdout.flush()
1945 sys.stdout.flush()
1949 for x in xrange(10):
1946 for x in xrange(10):
1950 if channels:
1947 if channels:
1951 time.sleep(.1)
1948 time.sleep(.1)
1952 count += 1
1949 count += 1
1953
1950
1954 stoppedearly = False
1951 stoppedearly = False
1955
1952
1956 if self._showchannels:
1953 if self._showchannels:
1957 statthread = threading.Thread(target=stat, name="stat")
1954 statthread = threading.Thread(target=stat, name="stat")
1958 statthread.start()
1955 statthread.start()
1959
1956
1960 try:
1957 try:
1961 while tests or running:
1958 while tests or running:
1962 if not done.empty() or running == self._jobs or not tests:
1959 if not done.empty() or running == self._jobs or not tests:
1963 try:
1960 try:
1964 done.get(True, 1)
1961 done.get(True, 1)
1965 running -= 1
1962 running -= 1
1966 if result and result.shouldStop:
1963 if result and result.shouldStop:
1967 stoppedearly = True
1964 stoppedearly = True
1968 break
1965 break
1969 except queue.Empty:
1966 except queue.Empty:
1970 continue
1967 continue
1971 if tests and not running == self._jobs:
1968 if tests and not running == self._jobs:
1972 test = tests.pop(0)
1969 test = tests.pop(0)
1973 if self._loop:
1970 if self._loop:
1974 if getattr(test, 'should_reload', False):
1971 if getattr(test, 'should_reload', False):
1975 num_tests[0] += 1
1972 num_tests[0] += 1
1976 tests.append(
1973 tests.append(
1977 self._loadtest(test, num_tests[0]))
1974 self._loadtest(test, num_tests[0]))
1978 else:
1975 else:
1979 tests.append(test)
1976 tests.append(test)
1980 if self._jobs == 1:
1977 if self._jobs == 1:
1981 job(test, result)
1978 job(test, result)
1982 else:
1979 else:
1983 t = threading.Thread(target=job, name=test.name,
1980 t = threading.Thread(target=job, name=test.name,
1984 args=(test, result))
1981 args=(test, result))
1985 t.start()
1982 t.start()
1986 running += 1
1983 running += 1
1987
1984
1988 # If we stop early we still need to wait on started tests to
1985 # If we stop early we still need to wait on started tests to
1989 # finish. Otherwise, there is a race between the test completing
1986 # finish. Otherwise, there is a race between the test completing
1990 # and the test's cleanup code running. This could result in the
1987 # and the test's cleanup code running. This could result in the
1991 # test reporting incorrect.
1988 # test reporting incorrect.
1992 if stoppedearly:
1989 if stoppedearly:
1993 while running:
1990 while running:
1994 try:
1991 try:
1995 done.get(True, 1)
1992 done.get(True, 1)
1996 running -= 1
1993 running -= 1
1997 except queue.Empty:
1994 except queue.Empty:
1998 continue
1995 continue
1999 except KeyboardInterrupt:
1996 except KeyboardInterrupt:
2000 for test in runtests:
1997 for test in runtests:
2001 test.abort()
1998 test.abort()
2002
1999
2003 channels = []
2000 channels = []
2004
2001
2005 return result
2002 return result
2006
2003
2007 # Save the most recent 5 wall-clock runtimes of each test to a
2004 # Save the most recent 5 wall-clock runtimes of each test to a
2008 # human-readable text file named .testtimes. Tests are sorted
2005 # human-readable text file named .testtimes. Tests are sorted
2009 # alphabetically, while times for each test are listed from oldest to
2006 # alphabetically, while times for each test are listed from oldest to
2010 # newest.
2007 # newest.
2011
2008
2012 def loadtimes(outputdir):
2009 def loadtimes(outputdir):
2013 times = []
2010 times = []
2014 try:
2011 try:
2015 with open(os.path.join(outputdir, b'.testtimes-')) as fp:
2012 with open(os.path.join(outputdir, b'.testtimes-')) as fp:
2016 for line in fp:
2013 for line in fp:
2017 ts = line.split()
2014 ts = line.split()
2018 times.append((ts[0], [float(t) for t in ts[1:]]))
2015 times.append((ts[0], [float(t) for t in ts[1:]]))
2019 except IOError as err:
2016 except IOError as err:
2020 if err.errno != errno.ENOENT:
2017 if err.errno != errno.ENOENT:
2021 raise
2018 raise
2022 return times
2019 return times
2023
2020
2024 def savetimes(outputdir, result):
2021 def savetimes(outputdir, result):
2025 saved = dict(loadtimes(outputdir))
2022 saved = dict(loadtimes(outputdir))
2026 maxruns = 5
2023 maxruns = 5
2027 skipped = set([str(t[0]) for t in result.skipped])
2024 skipped = set([str(t[0]) for t in result.skipped])
2028 for tdata in result.times:
2025 for tdata in result.times:
2029 test, real = tdata[0], tdata[3]
2026 test, real = tdata[0], tdata[3]
2030 if test not in skipped:
2027 if test not in skipped:
2031 ts = saved.setdefault(test, [])
2028 ts = saved.setdefault(test, [])
2032 ts.append(real)
2029 ts.append(real)
2033 ts[:] = ts[-maxruns:]
2030 ts[:] = ts[-maxruns:]
2034
2031
2035 fd, tmpname = tempfile.mkstemp(prefix=b'.testtimes',
2032 fd, tmpname = tempfile.mkstemp(prefix=b'.testtimes',
2036 dir=outputdir, text=True)
2033 dir=outputdir, text=True)
2037 with os.fdopen(fd, 'w') as fp:
2034 with os.fdopen(fd, 'w') as fp:
2038 for name, ts in sorted(saved.items()):
2035 for name, ts in sorted(saved.items()):
2039 fp.write('%s %s\n' % (name, ' '.join(['%.3f' % (t,) for t in ts])))
2036 fp.write('%s %s\n' % (name, ' '.join(['%.3f' % (t,) for t in ts])))
2040 timepath = os.path.join(outputdir, b'.testtimes')
2037 timepath = os.path.join(outputdir, b'.testtimes')
2041 try:
2038 try:
2042 os.unlink(timepath)
2039 os.unlink(timepath)
2043 except OSError:
2040 except OSError:
2044 pass
2041 pass
2045 try:
2042 try:
2046 os.rename(tmpname, timepath)
2043 os.rename(tmpname, timepath)
2047 except OSError:
2044 except OSError:
2048 pass
2045 pass
2049
2046
2050 class TextTestRunner(unittest.TextTestRunner):
2047 class TextTestRunner(unittest.TextTestRunner):
2051 """Custom unittest test runner that uses appropriate settings."""
2048 """Custom unittest test runner that uses appropriate settings."""
2052
2049
2053 def __init__(self, runner, *args, **kwargs):
2050 def __init__(self, runner, *args, **kwargs):
2054 super(TextTestRunner, self).__init__(*args, **kwargs)
2051 super(TextTestRunner, self).__init__(*args, **kwargs)
2055
2052
2056 self._runner = runner
2053 self._runner = runner
2057
2054
2058 def listtests(self, test):
2055 def listtests(self, test):
2059 result = TestResult(self._runner.options, self.stream,
2056 result = TestResult(self._runner.options, self.stream,
2060 self.descriptions, 0)
2057 self.descriptions, 0)
2061 test = sorted(test, key=lambda t: t.name)
2058 test = sorted(test, key=lambda t: t.name)
2062 for t in test:
2059 for t in test:
2063 print(t.name)
2060 print(t.name)
2064 result.addSuccess(t)
2061 result.addSuccess(t)
2065
2062
2066 if self._runner.options.xunit:
2063 if self._runner.options.xunit:
2067 with open(self._runner.options.xunit, "wb") as xuf:
2064 with open(self._runner.options.xunit, "wb") as xuf:
2068 self._writexunit(result, xuf)
2065 self._writexunit(result, xuf)
2069
2066
2070 if self._runner.options.json:
2067 if self._runner.options.json:
2071 jsonpath = os.path.join(self._runner._outputdir, b'report.json')
2068 jsonpath = os.path.join(self._runner._outputdir, b'report.json')
2072 with open(jsonpath, 'w') as fp:
2069 with open(jsonpath, 'w') as fp:
2073 self._writejson(result, fp)
2070 self._writejson(result, fp)
2074
2071
2075 return result
2072 return result
2076
2073
2077 def run(self, test):
2074 def run(self, test):
2078 result = TestResult(self._runner.options, self.stream,
2075 result = TestResult(self._runner.options, self.stream,
2079 self.descriptions, self.verbosity)
2076 self.descriptions, self.verbosity)
2080
2077
2081 test(result)
2078 test(result)
2082
2079
2083 failed = len(result.failures)
2080 failed = len(result.failures)
2084 skipped = len(result.skipped)
2081 skipped = len(result.skipped)
2085 ignored = len(result.ignored)
2082 ignored = len(result.ignored)
2086
2083
2087 with iolock:
2084 with iolock:
2088 self.stream.writeln('')
2085 self.stream.writeln('')
2089
2086
2090 if not self._runner.options.noskips:
2087 if not self._runner.options.noskips:
2091 for test, msg in result.skipped:
2088 for test, msg in result.skipped:
2092 formatted = 'Skipped %s: %s\n' % (test.name, msg)
2089 formatted = 'Skipped %s: %s\n' % (test.name, msg)
2093 self.stream.write(highlightmsg(formatted, result.color))
2090 self.stream.write(highlightmsg(formatted, result.color))
2094 for test, msg in result.failures:
2091 for test, msg in result.failures:
2095 formatted = 'Failed %s: %s\n' % (test.name, msg)
2092 formatted = 'Failed %s: %s\n' % (test.name, msg)
2096 self.stream.write(highlightmsg(formatted, result.color))
2093 self.stream.write(highlightmsg(formatted, result.color))
2097 for test, msg in result.errors:
2094 for test, msg in result.errors:
2098 self.stream.writeln('Errored %s: %s' % (test.name, msg))
2095 self.stream.writeln('Errored %s: %s' % (test.name, msg))
2099
2096
2100 if self._runner.options.xunit:
2097 if self._runner.options.xunit:
2101 with open(self._runner.options.xunit, "wb") as xuf:
2098 with open(self._runner.options.xunit, "wb") as xuf:
2102 self._writexunit(result, xuf)
2099 self._writexunit(result, xuf)
2103
2100
2104 if self._runner.options.json:
2101 if self._runner.options.json:
2105 jsonpath = os.path.join(self._runner._outputdir, b'report.json')
2102 jsonpath = os.path.join(self._runner._outputdir, b'report.json')
2106 with open(jsonpath, 'w') as fp:
2103 with open(jsonpath, 'w') as fp:
2107 self._writejson(result, fp)
2104 self._writejson(result, fp)
2108
2105
2109 self._runner._checkhglib('Tested')
2106 self._runner._checkhglib('Tested')
2110
2107
2111 savetimes(self._runner._outputdir, result)
2108 savetimes(self._runner._outputdir, result)
2112
2109
2113 if failed and self._runner.options.known_good_rev:
2110 if failed and self._runner.options.known_good_rev:
2114 self._bisecttests(t for t, m in result.failures)
2111 self._bisecttests(t for t, m in result.failures)
2115 self.stream.writeln(
2112 self.stream.writeln(
2116 '# Ran %d tests, %d skipped, %d failed.'
2113 '# Ran %d tests, %d skipped, %d failed.'
2117 % (result.testsRun, skipped + ignored, failed))
2114 % (result.testsRun, skipped + ignored, failed))
2118 if failed:
2115 if failed:
2119 self.stream.writeln('python hash seed: %s' %
2116 self.stream.writeln('python hash seed: %s' %
2120 os.environ['PYTHONHASHSEED'])
2117 os.environ['PYTHONHASHSEED'])
2121 if self._runner.options.time:
2118 if self._runner.options.time:
2122 self.printtimes(result.times)
2119 self.printtimes(result.times)
2123
2120
2124 if self._runner.options.exceptions:
2121 if self._runner.options.exceptions:
2125 exceptions = aggregateexceptions(
2122 exceptions = aggregateexceptions(
2126 os.path.join(self._runner._outputdir, b'exceptions'))
2123 os.path.join(self._runner._outputdir, b'exceptions'))
2127 total = sum(exceptions.values())
2124 total = sum(exceptions.values())
2128
2125
2129 self.stream.writeln('Exceptions Report:')
2126 self.stream.writeln('Exceptions Report:')
2130 self.stream.writeln('%d total from %d frames' %
2127 self.stream.writeln('%d total from %d frames' %
2131 (total, len(exceptions)))
2128 (total, len(exceptions)))
2132 for (frame, line, exc), count in exceptions.most_common():
2129 for (frame, line, exc), count in exceptions.most_common():
2133 self.stream.writeln('%d\t%s: %s' % (count, frame, exc))
2130 self.stream.writeln('%d\t%s: %s' % (count, frame, exc))
2134
2131
2135 self.stream.flush()
2132 self.stream.flush()
2136
2133
2137 return result
2134 return result
2138
2135
2139 def _bisecttests(self, tests):
2136 def _bisecttests(self, tests):
2140 bisectcmd = ['hg', 'bisect']
2137 bisectcmd = ['hg', 'bisect']
2141 bisectrepo = self._runner.options.bisect_repo
2138 bisectrepo = self._runner.options.bisect_repo
2142 if bisectrepo:
2139 if bisectrepo:
2143 bisectcmd.extend(['-R', os.path.abspath(bisectrepo)])
2140 bisectcmd.extend(['-R', os.path.abspath(bisectrepo)])
2144 def pread(args):
2141 def pread(args):
2145 env = os.environ.copy()
2142 env = os.environ.copy()
2146 env['HGPLAIN'] = '1'
2143 env['HGPLAIN'] = '1'
2147 p = subprocess.Popen(args, stderr=subprocess.STDOUT,
2144 p = subprocess.Popen(args, stderr=subprocess.STDOUT,
2148 stdout=subprocess.PIPE, env=env)
2145 stdout=subprocess.PIPE, env=env)
2149 data = p.stdout.read()
2146 data = p.stdout.read()
2150 p.wait()
2147 p.wait()
2151 return data
2148 return data
2152 for test in tests:
2149 for test in tests:
2153 pread(bisectcmd + ['--reset']),
2150 pread(bisectcmd + ['--reset']),
2154 pread(bisectcmd + ['--bad', '.'])
2151 pread(bisectcmd + ['--bad', '.'])
2155 pread(bisectcmd + ['--good', self._runner.options.known_good_rev])
2152 pread(bisectcmd + ['--good', self._runner.options.known_good_rev])
2156 # TODO: we probably need to forward more options
2153 # TODO: we probably need to forward more options
2157 # that alter hg's behavior inside the tests.
2154 # that alter hg's behavior inside the tests.
2158 opts = ''
2155 opts = ''
2159 withhg = self._runner.options.with_hg
2156 withhg = self._runner.options.with_hg
2160 if withhg:
2157 if withhg:
2161 opts += ' --with-hg=%s ' % shellquote(_strpath(withhg))
2158 opts += ' --with-hg=%s ' % shellquote(_strpath(withhg))
2162 rtc = '%s %s %s %s' % (sys.executable, sys.argv[0], opts,
2159 rtc = '%s %s %s %s' % (sys.executable, sys.argv[0], opts,
2163 test)
2160 test)
2164 data = pread(bisectcmd + ['--command', rtc])
2161 data = pread(bisectcmd + ['--command', rtc])
2165 m = re.search(
2162 m = re.search(
2166 (br'\nThe first (?P<goodbad>bad|good) revision '
2163 (br'\nThe first (?P<goodbad>bad|good) revision '
2167 br'is:\nchangeset: +\d+:(?P<node>[a-f0-9]+)\n.*\n'
2164 br'is:\nchangeset: +\d+:(?P<node>[a-f0-9]+)\n.*\n'
2168 br'summary: +(?P<summary>[^\n]+)\n'),
2165 br'summary: +(?P<summary>[^\n]+)\n'),
2169 data, (re.MULTILINE | re.DOTALL))
2166 data, (re.MULTILINE | re.DOTALL))
2170 if m is None:
2167 if m is None:
2171 self.stream.writeln(
2168 self.stream.writeln(
2172 'Failed to identify failure point for %s' % test)
2169 'Failed to identify failure point for %s' % test)
2173 continue
2170 continue
2174 dat = m.groupdict()
2171 dat = m.groupdict()
2175 verb = 'broken' if dat['goodbad'] == 'bad' else 'fixed'
2172 verb = 'broken' if dat['goodbad'] == 'bad' else 'fixed'
2176 self.stream.writeln(
2173 self.stream.writeln(
2177 '%s %s by %s (%s)' % (
2174 '%s %s by %s (%s)' % (
2178 test, verb, dat['node'], dat['summary']))
2175 test, verb, dat['node'], dat['summary']))
2179
2176
2180 def printtimes(self, times):
2177 def printtimes(self, times):
2181 # iolock held by run
2178 # iolock held by run
2182 self.stream.writeln('# Producing time report')
2179 self.stream.writeln('# Producing time report')
2183 times.sort(key=lambda t: (t[3]))
2180 times.sort(key=lambda t: (t[3]))
2184 cols = '%7.3f %7.3f %7.3f %7.3f %7.3f %s'
2181 cols = '%7.3f %7.3f %7.3f %7.3f %7.3f %s'
2185 self.stream.writeln('%-7s %-7s %-7s %-7s %-7s %s' %
2182 self.stream.writeln('%-7s %-7s %-7s %-7s %-7s %s' %
2186 ('start', 'end', 'cuser', 'csys', 'real', 'Test'))
2183 ('start', 'end', 'cuser', 'csys', 'real', 'Test'))
2187 for tdata in times:
2184 for tdata in times:
2188 test = tdata[0]
2185 test = tdata[0]
2189 cuser, csys, real, start, end = tdata[1:6]
2186 cuser, csys, real, start, end = tdata[1:6]
2190 self.stream.writeln(cols % (start, end, cuser, csys, real, test))
2187 self.stream.writeln(cols % (start, end, cuser, csys, real, test))
2191
2188
2192 @staticmethod
2189 @staticmethod
2193 def _writexunit(result, outf):
2190 def _writexunit(result, outf):
2194 # See http://llg.cubic.org/docs/junit/ for a reference.
2191 # See http://llg.cubic.org/docs/junit/ for a reference.
2195 timesd = dict((t[0], t[3]) for t in result.times)
2192 timesd = dict((t[0], t[3]) for t in result.times)
2196 doc = minidom.Document()
2193 doc = minidom.Document()
2197 s = doc.createElement('testsuite')
2194 s = doc.createElement('testsuite')
2198 s.setAttribute('name', 'run-tests')
2195 s.setAttribute('name', 'run-tests')
2199 s.setAttribute('tests', str(result.testsRun))
2196 s.setAttribute('tests', str(result.testsRun))
2200 s.setAttribute('errors', "0") # TODO
2197 s.setAttribute('errors', "0") # TODO
2201 s.setAttribute('failures', str(len(result.failures)))
2198 s.setAttribute('failures', str(len(result.failures)))
2202 s.setAttribute('skipped', str(len(result.skipped) +
2199 s.setAttribute('skipped', str(len(result.skipped) +
2203 len(result.ignored)))
2200 len(result.ignored)))
2204 doc.appendChild(s)
2201 doc.appendChild(s)
2205 for tc in result.successes:
2202 for tc in result.successes:
2206 t = doc.createElement('testcase')
2203 t = doc.createElement('testcase')
2207 t.setAttribute('name', tc.name)
2204 t.setAttribute('name', tc.name)
2208 tctime = timesd.get(tc.name)
2205 tctime = timesd.get(tc.name)
2209 if tctime is not None:
2206 if tctime is not None:
2210 t.setAttribute('time', '%.3f' % tctime)
2207 t.setAttribute('time', '%.3f' % tctime)
2211 s.appendChild(t)
2208 s.appendChild(t)
2212 for tc, err in sorted(result.faildata.items()):
2209 for tc, err in sorted(result.faildata.items()):
2213 t = doc.createElement('testcase')
2210 t = doc.createElement('testcase')
2214 t.setAttribute('name', tc)
2211 t.setAttribute('name', tc)
2215 tctime = timesd.get(tc)
2212 tctime = timesd.get(tc)
2216 if tctime is not None:
2213 if tctime is not None:
2217 t.setAttribute('time', '%.3f' % tctime)
2214 t.setAttribute('time', '%.3f' % tctime)
2218 # createCDATASection expects a unicode or it will
2215 # createCDATASection expects a unicode or it will
2219 # convert using default conversion rules, which will
2216 # convert using default conversion rules, which will
2220 # fail if string isn't ASCII.
2217 # fail if string isn't ASCII.
2221 err = cdatasafe(err).decode('utf-8', 'replace')
2218 err = cdatasafe(err).decode('utf-8', 'replace')
2222 cd = doc.createCDATASection(err)
2219 cd = doc.createCDATASection(err)
2223 # Use 'failure' here instead of 'error' to match errors = 0,
2220 # Use 'failure' here instead of 'error' to match errors = 0,
2224 # failures = len(result.failures) in the testsuite element.
2221 # failures = len(result.failures) in the testsuite element.
2225 failelem = doc.createElement('failure')
2222 failelem = doc.createElement('failure')
2226 failelem.setAttribute('message', 'output changed')
2223 failelem.setAttribute('message', 'output changed')
2227 failelem.setAttribute('type', 'output-mismatch')
2224 failelem.setAttribute('type', 'output-mismatch')
2228 failelem.appendChild(cd)
2225 failelem.appendChild(cd)
2229 t.appendChild(failelem)
2226 t.appendChild(failelem)
2230 s.appendChild(t)
2227 s.appendChild(t)
2231 for tc, message in result.skipped:
2228 for tc, message in result.skipped:
2232 # According to the schema, 'skipped' has no attributes. So store
2229 # According to the schema, 'skipped' has no attributes. So store
2233 # the skip message as a text node instead.
2230 # the skip message as a text node instead.
2234 t = doc.createElement('testcase')
2231 t = doc.createElement('testcase')
2235 t.setAttribute('name', tc.name)
2232 t.setAttribute('name', tc.name)
2236 binmessage = message.encode('utf-8')
2233 binmessage = message.encode('utf-8')
2237 message = cdatasafe(binmessage).decode('utf-8', 'replace')
2234 message = cdatasafe(binmessage).decode('utf-8', 'replace')
2238 cd = doc.createCDATASection(message)
2235 cd = doc.createCDATASection(message)
2239 skipelem = doc.createElement('skipped')
2236 skipelem = doc.createElement('skipped')
2240 skipelem.appendChild(cd)
2237 skipelem.appendChild(cd)
2241 t.appendChild(skipelem)
2238 t.appendChild(skipelem)
2242 s.appendChild(t)
2239 s.appendChild(t)
2243 outf.write(doc.toprettyxml(indent=' ', encoding='utf-8'))
2240 outf.write(doc.toprettyxml(indent=' ', encoding='utf-8'))
2244
2241
2245 @staticmethod
2242 @staticmethod
2246 def _writejson(result, outf):
2243 def _writejson(result, outf):
2247 timesd = {}
2244 timesd = {}
2248 for tdata in result.times:
2245 for tdata in result.times:
2249 test = tdata[0]
2246 test = tdata[0]
2250 timesd[test] = tdata[1:]
2247 timesd[test] = tdata[1:]
2251
2248
2252 outcome = {}
2249 outcome = {}
2253 groups = [('success', ((tc, None)
2250 groups = [('success', ((tc, None)
2254 for tc in result.successes)),
2251 for tc in result.successes)),
2255 ('failure', result.failures),
2252 ('failure', result.failures),
2256 ('skip', result.skipped)]
2253 ('skip', result.skipped)]
2257 for res, testcases in groups:
2254 for res, testcases in groups:
2258 for tc, __ in testcases:
2255 for tc, __ in testcases:
2259 if tc.name in timesd:
2256 if tc.name in timesd:
2260 diff = result.faildata.get(tc.name, b'')
2257 diff = result.faildata.get(tc.name, b'')
2261 try:
2258 try:
2262 diff = diff.decode('unicode_escape')
2259 diff = diff.decode('unicode_escape')
2263 except UnicodeDecodeError as e:
2260 except UnicodeDecodeError as e:
2264 diff = '%r decoding diff, sorry' % e
2261 diff = '%r decoding diff, sorry' % e
2265 tres = {'result': res,
2262 tres = {'result': res,
2266 'time': ('%0.3f' % timesd[tc.name][2]),
2263 'time': ('%0.3f' % timesd[tc.name][2]),
2267 'cuser': ('%0.3f' % timesd[tc.name][0]),
2264 'cuser': ('%0.3f' % timesd[tc.name][0]),
2268 'csys': ('%0.3f' % timesd[tc.name][1]),
2265 'csys': ('%0.3f' % timesd[tc.name][1]),
2269 'start': ('%0.3f' % timesd[tc.name][3]),
2266 'start': ('%0.3f' % timesd[tc.name][3]),
2270 'end': ('%0.3f' % timesd[tc.name][4]),
2267 'end': ('%0.3f' % timesd[tc.name][4]),
2271 'diff': diff,
2268 'diff': diff,
2272 }
2269 }
2273 else:
2270 else:
2274 # blacklisted test
2271 # blacklisted test
2275 tres = {'result': res}
2272 tres = {'result': res}
2276
2273
2277 outcome[tc.name] = tres
2274 outcome[tc.name] = tres
2278 jsonout = json.dumps(outcome, sort_keys=True, indent=4,
2275 jsonout = json.dumps(outcome, sort_keys=True, indent=4,
2279 separators=(',', ': '))
2276 separators=(',', ': '))
2280 outf.writelines(("testreport =", jsonout))
2277 outf.writelines(("testreport =", jsonout))
2281
2278
2282 class TestRunner(object):
2279 class TestRunner(object):
2283 """Holds context for executing tests.
2280 """Holds context for executing tests.
2284
2281
2285 Tests rely on a lot of state. This object holds it for them.
2282 Tests rely on a lot of state. This object holds it for them.
2286 """
2283 """
2287
2284
2288 # Programs required to run tests.
2285 # Programs required to run tests.
2289 REQUIREDTOOLS = [
2286 REQUIREDTOOLS = [
2290 b'diff',
2287 b'diff',
2291 b'grep',
2288 b'grep',
2292 b'unzip',
2289 b'unzip',
2293 b'gunzip',
2290 b'gunzip',
2294 b'bunzip2',
2291 b'bunzip2',
2295 b'sed',
2292 b'sed',
2296 ]
2293 ]
2297
2294
2298 # Maps file extensions to test class.
2295 # Maps file extensions to test class.
2299 TESTTYPES = [
2296 TESTTYPES = [
2300 (b'.py', PythonTest),
2297 (b'.py', PythonTest),
2301 (b'.t', TTest),
2298 (b'.t', TTest),
2302 ]
2299 ]
2303
2300
2304 def __init__(self):
2301 def __init__(self):
2305 self.options = None
2302 self.options = None
2306 self._hgroot = None
2303 self._hgroot = None
2307 self._testdir = None
2304 self._testdir = None
2308 self._outputdir = None
2305 self._outputdir = None
2309 self._hgtmp = None
2306 self._hgtmp = None
2310 self._installdir = None
2307 self._installdir = None
2311 self._bindir = None
2308 self._bindir = None
2312 self._tmpbinddir = None
2309 self._tmpbinddir = None
2313 self._pythondir = None
2310 self._pythondir = None
2314 self._coveragefile = None
2311 self._coveragefile = None
2315 self._createdfiles = []
2312 self._createdfiles = []
2316 self._hgcommand = None
2313 self._hgcommand = None
2317 self._hgpath = None
2314 self._hgpath = None
2318 self._portoffset = 0
2315 self._portoffset = 0
2319 self._ports = {}
2316 self._ports = {}
2320
2317
2321 def run(self, args, parser=None):
2318 def run(self, args, parser=None):
2322 """Run the test suite."""
2319 """Run the test suite."""
2323 oldmask = os.umask(0o22)
2320 oldmask = os.umask(0o22)
2324 try:
2321 try:
2325 parser = parser or getparser()
2322 parser = parser or getparser()
2326 options = parseargs(args, parser)
2323 options = parseargs(args, parser)
2327 tests = [_bytespath(a) for a in options.tests]
2324 tests = [_bytespath(a) for a in options.tests]
2328 if options.test_list is not None:
2325 if options.test_list is not None:
2329 for listfile in options.test_list:
2326 for listfile in options.test_list:
2330 with open(listfile, 'rb') as f:
2327 with open(listfile, 'rb') as f:
2331 tests.extend(t for t in f.read().splitlines() if t)
2328 tests.extend(t for t in f.read().splitlines() if t)
2332 self.options = options
2329 self.options = options
2333
2330
2334 self._checktools()
2331 self._checktools()
2335 testdescs = self.findtests(tests)
2332 testdescs = self.findtests(tests)
2336 if options.profile_runner:
2333 if options.profile_runner:
2337 import statprof
2334 import statprof
2338 statprof.start()
2335 statprof.start()
2339 result = self._run(testdescs)
2336 result = self._run(testdescs)
2340 if options.profile_runner:
2337 if options.profile_runner:
2341 statprof.stop()
2338 statprof.stop()
2342 statprof.display()
2339 statprof.display()
2343 return result
2340 return result
2344
2341
2345 finally:
2342 finally:
2346 os.umask(oldmask)
2343 os.umask(oldmask)
2347
2344
2348 def _run(self, testdescs):
2345 def _run(self, testdescs):
2349 if self.options.random:
2346 if self.options.random:
2350 random.shuffle(testdescs)
2347 random.shuffle(testdescs)
2351 else:
2348 else:
2352 # keywords for slow tests
2349 # keywords for slow tests
2353 slow = {b'svn': 10,
2350 slow = {b'svn': 10,
2354 b'cvs': 10,
2351 b'cvs': 10,
2355 b'hghave': 10,
2352 b'hghave': 10,
2356 b'largefiles-update': 10,
2353 b'largefiles-update': 10,
2357 b'run-tests': 10,
2354 b'run-tests': 10,
2358 b'corruption': 10,
2355 b'corruption': 10,
2359 b'race': 10,
2356 b'race': 10,
2360 b'i18n': 10,
2357 b'i18n': 10,
2361 b'check': 100,
2358 b'check': 100,
2362 b'gendoc': 100,
2359 b'gendoc': 100,
2363 b'contrib-perf': 200,
2360 b'contrib-perf': 200,
2364 }
2361 }
2365 perf = {}
2362 perf = {}
2366 def sortkey(f):
2363 def sortkey(f):
2367 # run largest tests first, as they tend to take the longest
2364 # run largest tests first, as they tend to take the longest
2368 f = f['path']
2365 f = f['path']
2369 try:
2366 try:
2370 return perf[f]
2367 return perf[f]
2371 except KeyError:
2368 except KeyError:
2372 try:
2369 try:
2373 val = -os.stat(f).st_size
2370 val = -os.stat(f).st_size
2374 except OSError as e:
2371 except OSError as e:
2375 if e.errno != errno.ENOENT:
2372 if e.errno != errno.ENOENT:
2376 raise
2373 raise
2377 perf[f] = -1e9 # file does not exist, tell early
2374 perf[f] = -1e9 # file does not exist, tell early
2378 return -1e9
2375 return -1e9
2379 for kw, mul in slow.items():
2376 for kw, mul in slow.items():
2380 if kw in f:
2377 if kw in f:
2381 val *= mul
2378 val *= mul
2382 if f.endswith(b'.py'):
2379 if f.endswith(b'.py'):
2383 val /= 10.0
2380 val /= 10.0
2384 perf[f] = val / 1000.0
2381 perf[f] = val / 1000.0
2385 return perf[f]
2382 return perf[f]
2386 testdescs.sort(key=sortkey)
2383 testdescs.sort(key=sortkey)
2387
2384
2388 self._testdir = osenvironb[b'TESTDIR'] = getattr(
2385 self._testdir = osenvironb[b'TESTDIR'] = getattr(
2389 os, 'getcwdb', os.getcwd)()
2386 os, 'getcwdb', os.getcwd)()
2390 # assume all tests in same folder for now
2387 # assume all tests in same folder for now
2391 if testdescs:
2388 if testdescs:
2392 pathname = os.path.dirname(testdescs[0]['path'])
2389 pathname = os.path.dirname(testdescs[0]['path'])
2393 if pathname:
2390 if pathname:
2394 osenvironb[b'TESTDIR'] = os.path.join(osenvironb[b'TESTDIR'],
2391 osenvironb[b'TESTDIR'] = os.path.join(osenvironb[b'TESTDIR'],
2395 pathname)
2392 pathname)
2396 if self.options.outputdir:
2393 if self.options.outputdir:
2397 self._outputdir = canonpath(_bytespath(self.options.outputdir))
2394 self._outputdir = canonpath(_bytespath(self.options.outputdir))
2398 else:
2395 else:
2399 self._outputdir = self._testdir
2396 self._outputdir = self._testdir
2400 if testdescs and pathname:
2397 if testdescs and pathname:
2401 self._outputdir = os.path.join(self._outputdir, pathname)
2398 self._outputdir = os.path.join(self._outputdir, pathname)
2402
2399
2403 if 'PYTHONHASHSEED' not in os.environ:
2400 if 'PYTHONHASHSEED' not in os.environ:
2404 # use a random python hash seed all the time
2401 # use a random python hash seed all the time
2405 # we do the randomness ourself to know what seed is used
2402 # we do the randomness ourself to know what seed is used
2406 os.environ['PYTHONHASHSEED'] = str(random.getrandbits(32))
2403 os.environ['PYTHONHASHSEED'] = str(random.getrandbits(32))
2407
2404
2408 if self.options.tmpdir:
2405 if self.options.tmpdir:
2409 self.options.keep_tmpdir = True
2406 self.options.keep_tmpdir = True
2410 tmpdir = _bytespath(self.options.tmpdir)
2407 tmpdir = _bytespath(self.options.tmpdir)
2411 if os.path.exists(tmpdir):
2408 if os.path.exists(tmpdir):
2412 # Meaning of tmpdir has changed since 1.3: we used to create
2409 # Meaning of tmpdir has changed since 1.3: we used to create
2413 # HGTMP inside tmpdir; now HGTMP is tmpdir. So fail if
2410 # HGTMP inside tmpdir; now HGTMP is tmpdir. So fail if
2414 # tmpdir already exists.
2411 # tmpdir already exists.
2415 print("error: temp dir %r already exists" % tmpdir)
2412 print("error: temp dir %r already exists" % tmpdir)
2416 return 1
2413 return 1
2417
2414
2418 # Automatically removing tmpdir sounds convenient, but could
2415 # Automatically removing tmpdir sounds convenient, but could
2419 # really annoy anyone in the habit of using "--tmpdir=/tmp"
2416 # really annoy anyone in the habit of using "--tmpdir=/tmp"
2420 # or "--tmpdir=$HOME".
2417 # or "--tmpdir=$HOME".
2421 #vlog("# Removing temp dir", tmpdir)
2418 #vlog("# Removing temp dir", tmpdir)
2422 #shutil.rmtree(tmpdir)
2419 #shutil.rmtree(tmpdir)
2423 os.makedirs(tmpdir)
2420 os.makedirs(tmpdir)
2424 else:
2421 else:
2425 d = None
2422 d = None
2426 if os.name == 'nt':
2423 if os.name == 'nt':
2427 # without this, we get the default temp dir location, but
2424 # without this, we get the default temp dir location, but
2428 # in all lowercase, which causes troubles with paths (issue3490)
2425 # in all lowercase, which causes troubles with paths (issue3490)
2429 d = osenvironb.get(b'TMP', None)
2426 d = osenvironb.get(b'TMP', None)
2430 tmpdir = tempfile.mkdtemp(b'', b'hgtests.', d)
2427 tmpdir = tempfile.mkdtemp(b'', b'hgtests.', d)
2431
2428
2432 self._hgtmp = osenvironb[b'HGTMP'] = (
2429 self._hgtmp = osenvironb[b'HGTMP'] = (
2433 os.path.realpath(tmpdir))
2430 os.path.realpath(tmpdir))
2434
2431
2435 if self.options.with_hg:
2432 if self.options.with_hg:
2436 self._installdir = None
2433 self._installdir = None
2437 whg = self.options.with_hg
2434 whg = self.options.with_hg
2438 self._bindir = os.path.dirname(os.path.realpath(whg))
2435 self._bindir = os.path.dirname(os.path.realpath(whg))
2439 assert isinstance(self._bindir, bytes)
2436 assert isinstance(self._bindir, bytes)
2440 self._hgcommand = os.path.basename(whg)
2437 self._hgcommand = os.path.basename(whg)
2441 self._tmpbindir = os.path.join(self._hgtmp, b'install', b'bin')
2438 self._tmpbindir = os.path.join(self._hgtmp, b'install', b'bin')
2442 os.makedirs(self._tmpbindir)
2439 os.makedirs(self._tmpbindir)
2443
2440
2444 # This looks redundant with how Python initializes sys.path from
2441 # This looks redundant with how Python initializes sys.path from
2445 # the location of the script being executed. Needed because the
2442 # the location of the script being executed. Needed because the
2446 # "hg" specified by --with-hg is not the only Python script
2443 # "hg" specified by --with-hg is not the only Python script
2447 # executed in the test suite that needs to import 'mercurial'
2444 # executed in the test suite that needs to import 'mercurial'
2448 # ... which means it's not really redundant at all.
2445 # ... which means it's not really redundant at all.
2449 self._pythondir = self._bindir
2446 self._pythondir = self._bindir
2450 else:
2447 else:
2451 self._installdir = os.path.join(self._hgtmp, b"install")
2448 self._installdir = os.path.join(self._hgtmp, b"install")
2452 self._bindir = os.path.join(self._installdir, b"bin")
2449 self._bindir = os.path.join(self._installdir, b"bin")
2453 self._hgcommand = b'hg'
2450 self._hgcommand = b'hg'
2454 self._tmpbindir = self._bindir
2451 self._tmpbindir = self._bindir
2455 self._pythondir = os.path.join(self._installdir, b"lib", b"python")
2452 self._pythondir = os.path.join(self._installdir, b"lib", b"python")
2456
2453
2457 # set CHGHG, then replace "hg" command by "chg"
2454 # set CHGHG, then replace "hg" command by "chg"
2458 chgbindir = self._bindir
2455 chgbindir = self._bindir
2459 if self.options.chg or self.options.with_chg:
2456 if self.options.chg or self.options.with_chg:
2460 osenvironb[b'CHGHG'] = os.path.join(self._bindir, self._hgcommand)
2457 osenvironb[b'CHGHG'] = os.path.join(self._bindir, self._hgcommand)
2461 else:
2458 else:
2462 osenvironb.pop(b'CHGHG', None) # drop flag for hghave
2459 osenvironb.pop(b'CHGHG', None) # drop flag for hghave
2463 if self.options.chg:
2460 if self.options.chg:
2464 self._hgcommand = b'chg'
2461 self._hgcommand = b'chg'
2465 elif self.options.with_chg:
2462 elif self.options.with_chg:
2466 chgbindir = os.path.dirname(os.path.realpath(self.options.with_chg))
2463 chgbindir = os.path.dirname(os.path.realpath(self.options.with_chg))
2467 self._hgcommand = os.path.basename(self.options.with_chg)
2464 self._hgcommand = os.path.basename(self.options.with_chg)
2468
2465
2469 osenvironb[b"BINDIR"] = self._bindir
2466 osenvironb[b"BINDIR"] = self._bindir
2470 osenvironb[b"PYTHON"] = PYTHON
2467 osenvironb[b"PYTHON"] = PYTHON
2471
2468
2472 if self.options.with_python3:
2469 if self.options.with_python3:
2473 osenvironb[b'PYTHON3'] = self.options.with_python3
2470 osenvironb[b'PYTHON3'] = self.options.with_python3
2474
2471
2475 fileb = _bytespath(__file__)
2472 fileb = _bytespath(__file__)
2476 runtestdir = os.path.abspath(os.path.dirname(fileb))
2473 runtestdir = os.path.abspath(os.path.dirname(fileb))
2477 osenvironb[b'RUNTESTDIR'] = runtestdir
2474 osenvironb[b'RUNTESTDIR'] = runtestdir
2478 if PYTHON3:
2475 if PYTHON3:
2479 sepb = _bytespath(os.pathsep)
2476 sepb = _bytespath(os.pathsep)
2480 else:
2477 else:
2481 sepb = os.pathsep
2478 sepb = os.pathsep
2482 path = [self._bindir, runtestdir] + osenvironb[b"PATH"].split(sepb)
2479 path = [self._bindir, runtestdir] + osenvironb[b"PATH"].split(sepb)
2483 if os.path.islink(__file__):
2480 if os.path.islink(__file__):
2484 # test helper will likely be at the end of the symlink
2481 # test helper will likely be at the end of the symlink
2485 realfile = os.path.realpath(fileb)
2482 realfile = os.path.realpath(fileb)
2486 realdir = os.path.abspath(os.path.dirname(realfile))
2483 realdir = os.path.abspath(os.path.dirname(realfile))
2487 path.insert(2, realdir)
2484 path.insert(2, realdir)
2488 if chgbindir != self._bindir:
2485 if chgbindir != self._bindir:
2489 path.insert(1, chgbindir)
2486 path.insert(1, chgbindir)
2490 if self._testdir != runtestdir:
2487 if self._testdir != runtestdir:
2491 path = [self._testdir] + path
2488 path = [self._testdir] + path
2492 if self._tmpbindir != self._bindir:
2489 if self._tmpbindir != self._bindir:
2493 path = [self._tmpbindir] + path
2490 path = [self._tmpbindir] + path
2494 osenvironb[b"PATH"] = sepb.join(path)
2491 osenvironb[b"PATH"] = sepb.join(path)
2495
2492
2496 # Include TESTDIR in PYTHONPATH so that out-of-tree extensions
2493 # Include TESTDIR in PYTHONPATH so that out-of-tree extensions
2497 # can run .../tests/run-tests.py test-foo where test-foo
2494 # can run .../tests/run-tests.py test-foo where test-foo
2498 # adds an extension to HGRC. Also include run-test.py directory to
2495 # adds an extension to HGRC. Also include run-test.py directory to
2499 # import modules like heredoctest.
2496 # import modules like heredoctest.
2500 pypath = [self._pythondir, self._testdir, runtestdir]
2497 pypath = [self._pythondir, self._testdir, runtestdir]
2501 # We have to augment PYTHONPATH, rather than simply replacing
2498 # We have to augment PYTHONPATH, rather than simply replacing
2502 # it, in case external libraries are only available via current
2499 # it, in case external libraries are only available via current
2503 # PYTHONPATH. (In particular, the Subversion bindings on OS X
2500 # PYTHONPATH. (In particular, the Subversion bindings on OS X
2504 # are in /opt/subversion.)
2501 # are in /opt/subversion.)
2505 oldpypath = osenvironb.get(IMPL_PATH)
2502 oldpypath = osenvironb.get(IMPL_PATH)
2506 if oldpypath:
2503 if oldpypath:
2507 pypath.append(oldpypath)
2504 pypath.append(oldpypath)
2508 osenvironb[IMPL_PATH] = sepb.join(pypath)
2505 osenvironb[IMPL_PATH] = sepb.join(pypath)
2509
2506
2510 if self.options.pure:
2507 if self.options.pure:
2511 os.environ["HGTEST_RUN_TESTS_PURE"] = "--pure"
2508 os.environ["HGTEST_RUN_TESTS_PURE"] = "--pure"
2512 os.environ["HGMODULEPOLICY"] = "py"
2509 os.environ["HGMODULEPOLICY"] = "py"
2513
2510
2514 if self.options.allow_slow_tests:
2511 if self.options.allow_slow_tests:
2515 os.environ["HGTEST_SLOW"] = "slow"
2512 os.environ["HGTEST_SLOW"] = "slow"
2516 elif 'HGTEST_SLOW' in os.environ:
2513 elif 'HGTEST_SLOW' in os.environ:
2517 del os.environ['HGTEST_SLOW']
2514 del os.environ['HGTEST_SLOW']
2518
2515
2519 self._coveragefile = os.path.join(self._testdir, b'.coverage')
2516 self._coveragefile = os.path.join(self._testdir, b'.coverage')
2520
2517
2521 if self.options.exceptions:
2518 if self.options.exceptions:
2522 exceptionsdir = os.path.join(self._outputdir, b'exceptions')
2519 exceptionsdir = os.path.join(self._outputdir, b'exceptions')
2523 try:
2520 try:
2524 os.makedirs(exceptionsdir)
2521 os.makedirs(exceptionsdir)
2525 except OSError as e:
2522 except OSError as e:
2526 if e.errno != errno.EEXIST:
2523 if e.errno != errno.EEXIST:
2527 raise
2524 raise
2528
2525
2529 # Remove all existing exception reports.
2526 # Remove all existing exception reports.
2530 for f in os.listdir(exceptionsdir):
2527 for f in os.listdir(exceptionsdir):
2531 os.unlink(os.path.join(exceptionsdir, f))
2528 os.unlink(os.path.join(exceptionsdir, f))
2532
2529
2533 osenvironb[b'HGEXCEPTIONSDIR'] = exceptionsdir
2530 osenvironb[b'HGEXCEPTIONSDIR'] = exceptionsdir
2534 logexceptions = os.path.join(self._testdir, b'logexceptions.py')
2531 logexceptions = os.path.join(self._testdir, b'logexceptions.py')
2535 self.options.extra_config_opt.append(
2532 self.options.extra_config_opt.append(
2536 'extensions.logexceptions=%s' % logexceptions.decode('utf-8'))
2533 'extensions.logexceptions=%s' % logexceptions.decode('utf-8'))
2537
2534
2538 vlog("# Using TESTDIR", self._testdir)
2535 vlog("# Using TESTDIR", self._testdir)
2539 vlog("# Using RUNTESTDIR", osenvironb[b'RUNTESTDIR'])
2536 vlog("# Using RUNTESTDIR", osenvironb[b'RUNTESTDIR'])
2540 vlog("# Using HGTMP", self._hgtmp)
2537 vlog("# Using HGTMP", self._hgtmp)
2541 vlog("# Using PATH", os.environ["PATH"])
2538 vlog("# Using PATH", os.environ["PATH"])
2542 vlog("# Using", IMPL_PATH, osenvironb[IMPL_PATH])
2539 vlog("# Using", IMPL_PATH, osenvironb[IMPL_PATH])
2543 vlog("# Writing to directory", self._outputdir)
2540 vlog("# Writing to directory", self._outputdir)
2544
2541
2545 try:
2542 try:
2546 return self._runtests(testdescs) or 0
2543 return self._runtests(testdescs) or 0
2547 finally:
2544 finally:
2548 time.sleep(.1)
2545 time.sleep(.1)
2549 self._cleanup()
2546 self._cleanup()
2550
2547
2551 def findtests(self, args):
2548 def findtests(self, args):
2552 """Finds possible test files from arguments.
2549 """Finds possible test files from arguments.
2553
2550
2554 If you wish to inject custom tests into the test harness, this would
2551 If you wish to inject custom tests into the test harness, this would
2555 be a good function to monkeypatch or override in a derived class.
2552 be a good function to monkeypatch or override in a derived class.
2556 """
2553 """
2557 if not args:
2554 if not args:
2558 if self.options.changed:
2555 if self.options.changed:
2559 proc = Popen4('hg st --rev "%s" -man0 .' %
2556 proc = Popen4('hg st --rev "%s" -man0 .' %
2560 self.options.changed, None, 0)
2557 self.options.changed, None, 0)
2561 stdout, stderr = proc.communicate()
2558 stdout, stderr = proc.communicate()
2562 args = stdout.strip(b'\0').split(b'\0')
2559 args = stdout.strip(b'\0').split(b'\0')
2563 else:
2560 else:
2564 args = os.listdir(b'.')
2561 args = os.listdir(b'.')
2565
2562
2566 expanded_args = []
2563 expanded_args = []
2567 for arg in args:
2564 for arg in args:
2568 if os.path.isdir(arg):
2565 if os.path.isdir(arg):
2569 if not arg.endswith(b'/'):
2566 if not arg.endswith(b'/'):
2570 arg += b'/'
2567 arg += b'/'
2571 expanded_args.extend([arg + a for a in os.listdir(arg)])
2568 expanded_args.extend([arg + a for a in os.listdir(arg)])
2572 else:
2569 else:
2573 expanded_args.append(arg)
2570 expanded_args.append(arg)
2574 args = expanded_args
2571 args = expanded_args
2575
2572
2576 tests = []
2573 tests = []
2577 for t in args:
2574 for t in args:
2578 if not (os.path.basename(t).startswith(b'test-')
2575 if not (os.path.basename(t).startswith(b'test-')
2579 and (t.endswith(b'.py') or t.endswith(b'.t'))):
2576 and (t.endswith(b'.py') or t.endswith(b'.t'))):
2580 continue
2577 continue
2581 if t.endswith(b'.t'):
2578 if t.endswith(b'.t'):
2582 # .t file may contain multiple test cases
2579 # .t file may contain multiple test cases
2583 cases = sorted(parsettestcases(t))
2580 cases = sorted(parsettestcases(t))
2584 if cases:
2581 if cases:
2585 tests += [{'path': t, 'case': c} for c in sorted(cases)]
2582 tests += [{'path': t, 'case': c} for c in sorted(cases)]
2586 else:
2583 else:
2587 tests.append({'path': t})
2584 tests.append({'path': t})
2588 else:
2585 else:
2589 tests.append({'path': t})
2586 tests.append({'path': t})
2590 return tests
2587 return tests
2591
2588
2592 def _runtests(self, testdescs):
2589 def _runtests(self, testdescs):
2593 def _reloadtest(test, i):
2590 def _reloadtest(test, i):
2594 # convert a test back to its description dict
2591 # convert a test back to its description dict
2595 desc = {'path': test.path}
2592 desc = {'path': test.path}
2596 case = getattr(test, '_case', None)
2593 case = getattr(test, '_case', None)
2597 if case:
2594 if case:
2598 desc['case'] = case
2595 desc['case'] = case
2599 return self._gettest(desc, i)
2596 return self._gettest(desc, i)
2600
2597
2601 try:
2598 try:
2602 if self.options.restart:
2599 if self.options.restart:
2603 orig = list(testdescs)
2600 orig = list(testdescs)
2604 while testdescs:
2601 while testdescs:
2605 desc = testdescs[0]
2602 desc = testdescs[0]
2606 # desc['path'] is a relative path
2603 # desc['path'] is a relative path
2607 if 'case' in desc:
2604 if 'case' in desc:
2608 errpath = b'%s.%s.err' % (desc['path'], desc['case'])
2605 errpath = b'%s.%s.err' % (desc['path'], desc['case'])
2609 else:
2606 else:
2610 errpath = b'%s.err' % desc['path']
2607 errpath = b'%s.err' % desc['path']
2611 errpath = os.path.join(self._outputdir, errpath)
2608 errpath = os.path.join(self._outputdir, errpath)
2612 if os.path.exists(errpath):
2609 if os.path.exists(errpath):
2613 break
2610 break
2614 testdescs.pop(0)
2611 testdescs.pop(0)
2615 if not testdescs:
2612 if not testdescs:
2616 print("running all tests")
2613 print("running all tests")
2617 testdescs = orig
2614 testdescs = orig
2618
2615
2619 tests = [self._gettest(d, i) for i, d in enumerate(testdescs)]
2616 tests = [self._gettest(d, i) for i, d in enumerate(testdescs)]
2620
2617
2621 failed = False
2618 failed = False
2622 kws = self.options.keywords
2619 kws = self.options.keywords
2623 if kws is not None and PYTHON3:
2620 if kws is not None and PYTHON3:
2624 kws = kws.encode('utf-8')
2621 kws = kws.encode('utf-8')
2625
2622
2626 suite = TestSuite(self._testdir,
2623 suite = TestSuite(self._testdir,
2627 jobs=self.options.jobs,
2624 jobs=self.options.jobs,
2628 whitelist=self.options.whitelisted,
2625 whitelist=self.options.whitelisted,
2629 blacklist=self.options.blacklist,
2626 blacklist=self.options.blacklist,
2630 retest=self.options.retest,
2627 retest=self.options.retest,
2631 keywords=kws,
2628 keywords=kws,
2632 loop=self.options.loop,
2629 loop=self.options.loop,
2633 runs_per_test=self.options.runs_per_test,
2630 runs_per_test=self.options.runs_per_test,
2634 showchannels=self.options.showchannels,
2631 showchannels=self.options.showchannels,
2635 tests=tests, loadtest=_reloadtest)
2632 tests=tests, loadtest=_reloadtest)
2636 verbosity = 1
2633 verbosity = 1
2637 if self.options.verbose:
2634 if self.options.verbose:
2638 verbosity = 2
2635 verbosity = 2
2639 runner = TextTestRunner(self, verbosity=verbosity)
2636 runner = TextTestRunner(self, verbosity=verbosity)
2640
2637
2641 if self.options.list_tests:
2638 if self.options.list_tests:
2642 result = runner.listtests(suite)
2639 result = runner.listtests(suite)
2643 else:
2640 else:
2644 if self._installdir:
2641 if self._installdir:
2645 self._installhg()
2642 self._installhg()
2646 self._checkhglib("Testing")
2643 self._checkhglib("Testing")
2647 else:
2644 else:
2648 self._usecorrectpython()
2645 self._usecorrectpython()
2649 if self.options.chg:
2646 if self.options.chg:
2650 assert self._installdir
2647 assert self._installdir
2651 self._installchg()
2648 self._installchg()
2652
2649
2653 result = runner.run(suite)
2650 result = runner.run(suite)
2654
2651
2655 if result.failures:
2652 if result.failures:
2656 failed = True
2653 failed = True
2657
2654
2658 if self.options.anycoverage:
2655 if self.options.anycoverage:
2659 self._outputcoverage()
2656 self._outputcoverage()
2660 except KeyboardInterrupt:
2657 except KeyboardInterrupt:
2661 failed = True
2658 failed = True
2662 print("\ninterrupted!")
2659 print("\ninterrupted!")
2663
2660
2664 if failed:
2661 if failed:
2665 return 1
2662 return 1
2666
2663
2667 def _getport(self, count):
2664 def _getport(self, count):
2668 port = self._ports.get(count) # do we have a cached entry?
2665 port = self._ports.get(count) # do we have a cached entry?
2669 if port is None:
2666 if port is None:
2670 portneeded = 3
2667 portneeded = 3
2671 # above 100 tries we just give up and let test reports failure
2668 # above 100 tries we just give up and let test reports failure
2672 for tries in xrange(100):
2669 for tries in xrange(100):
2673 allfree = True
2670 allfree = True
2674 port = self.options.port + self._portoffset
2671 port = self.options.port + self._portoffset
2675 for idx in xrange(portneeded):
2672 for idx in xrange(portneeded):
2676 if not checkportisavailable(port + idx):
2673 if not checkportisavailable(port + idx):
2677 allfree = False
2674 allfree = False
2678 break
2675 break
2679 self._portoffset += portneeded
2676 self._portoffset += portneeded
2680 if allfree:
2677 if allfree:
2681 break
2678 break
2682 self._ports[count] = port
2679 self._ports[count] = port
2683 return port
2680 return port
2684
2681
2685 def _gettest(self, testdesc, count):
2682 def _gettest(self, testdesc, count):
2686 """Obtain a Test by looking at its filename.
2683 """Obtain a Test by looking at its filename.
2687
2684
2688 Returns a Test instance. The Test may not be runnable if it doesn't
2685 Returns a Test instance. The Test may not be runnable if it doesn't
2689 map to a known type.
2686 map to a known type.
2690 """
2687 """
2691 path = testdesc['path']
2688 path = testdesc['path']
2692 lctest = path.lower()
2689 lctest = path.lower()
2693 testcls = Test
2690 testcls = Test
2694
2691
2695 for ext, cls in self.TESTTYPES:
2692 for ext, cls in self.TESTTYPES:
2696 if lctest.endswith(ext):
2693 if lctest.endswith(ext):
2697 testcls = cls
2694 testcls = cls
2698 break
2695 break
2699
2696
2700 refpath = os.path.join(self._testdir, path)
2697 refpath = os.path.join(self._testdir, path)
2701 tmpdir = os.path.join(self._hgtmp, b'child%d' % count)
2698 tmpdir = os.path.join(self._hgtmp, b'child%d' % count)
2702
2699
2703 # extra keyword parameters. 'case' is used by .t tests
2700 # extra keyword parameters. 'case' is used by .t tests
2704 kwds = dict((k, testdesc[k]) for k in ['case'] if k in testdesc)
2701 kwds = dict((k, testdesc[k]) for k in ['case'] if k in testdesc)
2705
2702
2706 t = testcls(refpath, self._outputdir, tmpdir,
2703 t = testcls(refpath, self._outputdir, tmpdir,
2707 keeptmpdir=self.options.keep_tmpdir,
2704 keeptmpdir=self.options.keep_tmpdir,
2708 debug=self.options.debug,
2705 debug=self.options.debug,
2709 timeout=self.options.timeout,
2706 timeout=self.options.timeout,
2710 startport=self._getport(count),
2707 startport=self._getport(count),
2711 extraconfigopts=self.options.extra_config_opt,
2708 extraconfigopts=self.options.extra_config_opt,
2712 py3kwarnings=self.options.py3k_warnings,
2709 py3kwarnings=self.options.py3k_warnings,
2713 shell=self.options.shell,
2710 shell=self.options.shell,
2714 hgcommand=self._hgcommand,
2711 hgcommand=self._hgcommand,
2715 usechg=bool(self.options.with_chg or self.options.chg),
2712 usechg=bool(self.options.with_chg or self.options.chg),
2716 useipv6=useipv6, **kwds)
2713 useipv6=useipv6, **kwds)
2717 t.should_reload = True
2714 t.should_reload = True
2718 return t
2715 return t
2719
2716
2720 def _cleanup(self):
2717 def _cleanup(self):
2721 """Clean up state from this test invocation."""
2718 """Clean up state from this test invocation."""
2722 if self.options.keep_tmpdir:
2719 if self.options.keep_tmpdir:
2723 return
2720 return
2724
2721
2725 vlog("# Cleaning up HGTMP", self._hgtmp)
2722 vlog("# Cleaning up HGTMP", self._hgtmp)
2726 shutil.rmtree(self._hgtmp, True)
2723 shutil.rmtree(self._hgtmp, True)
2727 for f in self._createdfiles:
2724 for f in self._createdfiles:
2728 try:
2725 try:
2729 os.remove(f)
2726 os.remove(f)
2730 except OSError:
2727 except OSError:
2731 pass
2728 pass
2732
2729
2733 def _usecorrectpython(self):
2730 def _usecorrectpython(self):
2734 """Configure the environment to use the appropriate Python in tests."""
2731 """Configure the environment to use the appropriate Python in tests."""
2735 # Tests must use the same interpreter as us or bad things will happen.
2732 # Tests must use the same interpreter as us or bad things will happen.
2736 pyexename = sys.platform == 'win32' and b'python.exe' or b'python'
2733 pyexename = sys.platform == 'win32' and b'python.exe' or b'python'
2737 if getattr(os, 'symlink', None):
2734 if getattr(os, 'symlink', None):
2738 vlog("# Making python executable in test path a symlink to '%s'" %
2735 vlog("# Making python executable in test path a symlink to '%s'" %
2739 sys.executable)
2736 sys.executable)
2740 mypython = os.path.join(self._tmpbindir, pyexename)
2737 mypython = os.path.join(self._tmpbindir, pyexename)
2741 try:
2738 try:
2742 if os.readlink(mypython) == sys.executable:
2739 if os.readlink(mypython) == sys.executable:
2743 return
2740 return
2744 os.unlink(mypython)
2741 os.unlink(mypython)
2745 except OSError as err:
2742 except OSError as err:
2746 if err.errno != errno.ENOENT:
2743 if err.errno != errno.ENOENT:
2747 raise
2744 raise
2748 if self._findprogram(pyexename) != sys.executable:
2745 if self._findprogram(pyexename) != sys.executable:
2749 try:
2746 try:
2750 os.symlink(sys.executable, mypython)
2747 os.symlink(sys.executable, mypython)
2751 self._createdfiles.append(mypython)
2748 self._createdfiles.append(mypython)
2752 except OSError as err:
2749 except OSError as err:
2753 # child processes may race, which is harmless
2750 # child processes may race, which is harmless
2754 if err.errno != errno.EEXIST:
2751 if err.errno != errno.EEXIST:
2755 raise
2752 raise
2756 else:
2753 else:
2757 exedir, exename = os.path.split(sys.executable)
2754 exedir, exename = os.path.split(sys.executable)
2758 vlog("# Modifying search path to find %s as %s in '%s'" %
2755 vlog("# Modifying search path to find %s as %s in '%s'" %
2759 (exename, pyexename, exedir))
2756 (exename, pyexename, exedir))
2760 path = os.environ['PATH'].split(os.pathsep)
2757 path = os.environ['PATH'].split(os.pathsep)
2761 while exedir in path:
2758 while exedir in path:
2762 path.remove(exedir)
2759 path.remove(exedir)
2763 os.environ['PATH'] = os.pathsep.join([exedir] + path)
2760 os.environ['PATH'] = os.pathsep.join([exedir] + path)
2764 if not self._findprogram(pyexename):
2761 if not self._findprogram(pyexename):
2765 print("WARNING: Cannot find %s in search path" % pyexename)
2762 print("WARNING: Cannot find %s in search path" % pyexename)
2766
2763
2767 def _installhg(self):
2764 def _installhg(self):
2768 """Install hg into the test environment.
2765 """Install hg into the test environment.
2769
2766
2770 This will also configure hg with the appropriate testing settings.
2767 This will also configure hg with the appropriate testing settings.
2771 """
2768 """
2772 vlog("# Performing temporary installation of HG")
2769 vlog("# Performing temporary installation of HG")
2773 installerrs = os.path.join(self._hgtmp, b"install.err")
2770 installerrs = os.path.join(self._hgtmp, b"install.err")
2774 compiler = ''
2771 compiler = ''
2775 if self.options.compiler:
2772 if self.options.compiler:
2776 compiler = '--compiler ' + self.options.compiler
2773 compiler = '--compiler ' + self.options.compiler
2777 if self.options.pure:
2774 if self.options.pure:
2778 pure = b"--pure"
2775 pure = b"--pure"
2779 else:
2776 else:
2780 pure = b""
2777 pure = b""
2781
2778
2782 # Run installer in hg root
2779 # Run installer in hg root
2783 script = os.path.realpath(sys.argv[0])
2780 script = os.path.realpath(sys.argv[0])
2784 exe = sys.executable
2781 exe = sys.executable
2785 if PYTHON3:
2782 if PYTHON3:
2786 compiler = _bytespath(compiler)
2783 compiler = _bytespath(compiler)
2787 script = _bytespath(script)
2784 script = _bytespath(script)
2788 exe = _bytespath(exe)
2785 exe = _bytespath(exe)
2789 hgroot = os.path.dirname(os.path.dirname(script))
2786 hgroot = os.path.dirname(os.path.dirname(script))
2790 self._hgroot = hgroot
2787 self._hgroot = hgroot
2791 os.chdir(hgroot)
2788 os.chdir(hgroot)
2792 nohome = b'--home=""'
2789 nohome = b'--home=""'
2793 if os.name == 'nt':
2790 if os.name == 'nt':
2794 # The --home="" trick works only on OS where os.sep == '/'
2791 # The --home="" trick works only on OS where os.sep == '/'
2795 # because of a distutils convert_path() fast-path. Avoid it at
2792 # because of a distutils convert_path() fast-path. Avoid it at
2796 # least on Windows for now, deal with .pydistutils.cfg bugs
2793 # least on Windows for now, deal with .pydistutils.cfg bugs
2797 # when they happen.
2794 # when they happen.
2798 nohome = b''
2795 nohome = b''
2799 cmd = (b'%(exe)s setup.py %(pure)s clean --all'
2796 cmd = (b'%(exe)s setup.py %(pure)s clean --all'
2800 b' build %(compiler)s --build-base="%(base)s"'
2797 b' build %(compiler)s --build-base="%(base)s"'
2801 b' install --force --prefix="%(prefix)s"'
2798 b' install --force --prefix="%(prefix)s"'
2802 b' --install-lib="%(libdir)s"'
2799 b' --install-lib="%(libdir)s"'
2803 b' --install-scripts="%(bindir)s" %(nohome)s >%(logfile)s 2>&1'
2800 b' --install-scripts="%(bindir)s" %(nohome)s >%(logfile)s 2>&1'
2804 % {b'exe': exe, b'pure': pure,
2801 % {b'exe': exe, b'pure': pure,
2805 b'compiler': compiler,
2802 b'compiler': compiler,
2806 b'base': os.path.join(self._hgtmp, b"build"),
2803 b'base': os.path.join(self._hgtmp, b"build"),
2807 b'prefix': self._installdir, b'libdir': self._pythondir,
2804 b'prefix': self._installdir, b'libdir': self._pythondir,
2808 b'bindir': self._bindir,
2805 b'bindir': self._bindir,
2809 b'nohome': nohome, b'logfile': installerrs})
2806 b'nohome': nohome, b'logfile': installerrs})
2810
2807
2811 # setuptools requires install directories to exist.
2808 # setuptools requires install directories to exist.
2812 def makedirs(p):
2809 def makedirs(p):
2813 try:
2810 try:
2814 os.makedirs(p)
2811 os.makedirs(p)
2815 except OSError as e:
2812 except OSError as e:
2816 if e.errno != errno.EEXIST:
2813 if e.errno != errno.EEXIST:
2817 raise
2814 raise
2818 makedirs(self._pythondir)
2815 makedirs(self._pythondir)
2819 makedirs(self._bindir)
2816 makedirs(self._bindir)
2820
2817
2821 vlog("# Running", cmd)
2818 vlog("# Running", cmd)
2822 if os.system(cmd) == 0:
2819 if os.system(cmd) == 0:
2823 if not self.options.verbose:
2820 if not self.options.verbose:
2824 try:
2821 try:
2825 os.remove(installerrs)
2822 os.remove(installerrs)
2826 except OSError as e:
2823 except OSError as e:
2827 if e.errno != errno.ENOENT:
2824 if e.errno != errno.ENOENT:
2828 raise
2825 raise
2829 else:
2826 else:
2830 f = open(installerrs, 'rb')
2827 f = open(installerrs, 'rb')
2831 for line in f:
2828 for line in f:
2832 if PYTHON3:
2829 if PYTHON3:
2833 sys.stdout.buffer.write(line)
2830 sys.stdout.buffer.write(line)
2834 else:
2831 else:
2835 sys.stdout.write(line)
2832 sys.stdout.write(line)
2836 f.close()
2833 f.close()
2837 sys.exit(1)
2834 sys.exit(1)
2838 os.chdir(self._testdir)
2835 os.chdir(self._testdir)
2839
2836
2840 self._usecorrectpython()
2837 self._usecorrectpython()
2841
2838
2842 if self.options.py3k_warnings and not self.options.anycoverage:
2839 if self.options.py3k_warnings and not self.options.anycoverage:
2843 vlog("# Updating hg command to enable Py3k Warnings switch")
2840 vlog("# Updating hg command to enable Py3k Warnings switch")
2844 f = open(os.path.join(self._bindir, 'hg'), 'rb')
2841 f = open(os.path.join(self._bindir, 'hg'), 'rb')
2845 lines = [line.rstrip() for line in f]
2842 lines = [line.rstrip() for line in f]
2846 lines[0] += ' -3'
2843 lines[0] += ' -3'
2847 f.close()
2844 f.close()
2848 f = open(os.path.join(self._bindir, 'hg'), 'wb')
2845 f = open(os.path.join(self._bindir, 'hg'), 'wb')
2849 for line in lines:
2846 for line in lines:
2850 f.write(line + '\n')
2847 f.write(line + '\n')
2851 f.close()
2848 f.close()
2852
2849
2853 hgbat = os.path.join(self._bindir, b'hg.bat')
2850 hgbat = os.path.join(self._bindir, b'hg.bat')
2854 if os.path.isfile(hgbat):
2851 if os.path.isfile(hgbat):
2855 # hg.bat expects to be put in bin/scripts while run-tests.py
2852 # hg.bat expects to be put in bin/scripts while run-tests.py
2856 # installation layout put it in bin/ directly. Fix it
2853 # installation layout put it in bin/ directly. Fix it
2857 f = open(hgbat, 'rb')
2854 f = open(hgbat, 'rb')
2858 data = f.read()
2855 data = f.read()
2859 f.close()
2856 f.close()
2860 if b'"%~dp0..\python" "%~dp0hg" %*' in data:
2857 if b'"%~dp0..\python" "%~dp0hg" %*' in data:
2861 data = data.replace(b'"%~dp0..\python" "%~dp0hg" %*',
2858 data = data.replace(b'"%~dp0..\python" "%~dp0hg" %*',
2862 b'"%~dp0python" "%~dp0hg" %*')
2859 b'"%~dp0python" "%~dp0hg" %*')
2863 f = open(hgbat, 'wb')
2860 f = open(hgbat, 'wb')
2864 f.write(data)
2861 f.write(data)
2865 f.close()
2862 f.close()
2866 else:
2863 else:
2867 print('WARNING: cannot fix hg.bat reference to python.exe')
2864 print('WARNING: cannot fix hg.bat reference to python.exe')
2868
2865
2869 if self.options.anycoverage:
2866 if self.options.anycoverage:
2870 custom = os.path.join(self._testdir, 'sitecustomize.py')
2867 custom = os.path.join(self._testdir, 'sitecustomize.py')
2871 target = os.path.join(self._pythondir, 'sitecustomize.py')
2868 target = os.path.join(self._pythondir, 'sitecustomize.py')
2872 vlog('# Installing coverage trigger to %s' % target)
2869 vlog('# Installing coverage trigger to %s' % target)
2873 shutil.copyfile(custom, target)
2870 shutil.copyfile(custom, target)
2874 rc = os.path.join(self._testdir, '.coveragerc')
2871 rc = os.path.join(self._testdir, '.coveragerc')
2875 vlog('# Installing coverage rc to %s' % rc)
2872 vlog('# Installing coverage rc to %s' % rc)
2876 os.environ['COVERAGE_PROCESS_START'] = rc
2873 os.environ['COVERAGE_PROCESS_START'] = rc
2877 covdir = os.path.join(self._installdir, '..', 'coverage')
2874 covdir = os.path.join(self._installdir, '..', 'coverage')
2878 try:
2875 try:
2879 os.mkdir(covdir)
2876 os.mkdir(covdir)
2880 except OSError as e:
2877 except OSError as e:
2881 if e.errno != errno.EEXIST:
2878 if e.errno != errno.EEXIST:
2882 raise
2879 raise
2883
2880
2884 os.environ['COVERAGE_DIR'] = covdir
2881 os.environ['COVERAGE_DIR'] = covdir
2885
2882
2886 def _checkhglib(self, verb):
2883 def _checkhglib(self, verb):
2887 """Ensure that the 'mercurial' package imported by python is
2884 """Ensure that the 'mercurial' package imported by python is
2888 the one we expect it to be. If not, print a warning to stderr."""
2885 the one we expect it to be. If not, print a warning to stderr."""
2889 if ((self._bindir == self._pythondir) and
2886 if ((self._bindir == self._pythondir) and
2890 (self._bindir != self._tmpbindir)):
2887 (self._bindir != self._tmpbindir)):
2891 # The pythondir has been inferred from --with-hg flag.
2888 # The pythondir has been inferred from --with-hg flag.
2892 # We cannot expect anything sensible here.
2889 # We cannot expect anything sensible here.
2893 return
2890 return
2894 expecthg = os.path.join(self._pythondir, b'mercurial')
2891 expecthg = os.path.join(self._pythondir, b'mercurial')
2895 actualhg = self._gethgpath()
2892 actualhg = self._gethgpath()
2896 if os.path.abspath(actualhg) != os.path.abspath(expecthg):
2893 if os.path.abspath(actualhg) != os.path.abspath(expecthg):
2897 sys.stderr.write('warning: %s with unexpected mercurial lib: %s\n'
2894 sys.stderr.write('warning: %s with unexpected mercurial lib: %s\n'
2898 ' (expected %s)\n'
2895 ' (expected %s)\n'
2899 % (verb, actualhg, expecthg))
2896 % (verb, actualhg, expecthg))
2900 def _gethgpath(self):
2897 def _gethgpath(self):
2901 """Return the path to the mercurial package that is actually found by
2898 """Return the path to the mercurial package that is actually found by
2902 the current Python interpreter."""
2899 the current Python interpreter."""
2903 if self._hgpath is not None:
2900 if self._hgpath is not None:
2904 return self._hgpath
2901 return self._hgpath
2905
2902
2906 cmd = b'%s -c "import mercurial; print (mercurial.__path__[0])"'
2903 cmd = b'%s -c "import mercurial; print (mercurial.__path__[0])"'
2907 cmd = cmd % PYTHON
2904 cmd = cmd % PYTHON
2908 if PYTHON3:
2905 if PYTHON3:
2909 cmd = _strpath(cmd)
2906 cmd = _strpath(cmd)
2910 pipe = os.popen(cmd)
2907 pipe = os.popen(cmd)
2911 try:
2908 try:
2912 self._hgpath = _bytespath(pipe.read().strip())
2909 self._hgpath = _bytespath(pipe.read().strip())
2913 finally:
2910 finally:
2914 pipe.close()
2911 pipe.close()
2915
2912
2916 return self._hgpath
2913 return self._hgpath
2917
2914
2918 def _installchg(self):
2915 def _installchg(self):
2919 """Install chg into the test environment"""
2916 """Install chg into the test environment"""
2920 vlog('# Performing temporary installation of CHG')
2917 vlog('# Performing temporary installation of CHG')
2921 assert os.path.dirname(self._bindir) == self._installdir
2918 assert os.path.dirname(self._bindir) == self._installdir
2922 assert self._hgroot, 'must be called after _installhg()'
2919 assert self._hgroot, 'must be called after _installhg()'
2923 cmd = (b'"%(make)s" clean install PREFIX="%(prefix)s"'
2920 cmd = (b'"%(make)s" clean install PREFIX="%(prefix)s"'
2924 % {b'make': 'make', # TODO: switch by option or environment?
2921 % {b'make': 'make', # TODO: switch by option or environment?
2925 b'prefix': self._installdir})
2922 b'prefix': self._installdir})
2926 cwd = os.path.join(self._hgroot, b'contrib', b'chg')
2923 cwd = os.path.join(self._hgroot, b'contrib', b'chg')
2927 vlog("# Running", cmd)
2924 vlog("# Running", cmd)
2928 proc = subprocess.Popen(cmd, shell=True, cwd=cwd,
2925 proc = subprocess.Popen(cmd, shell=True, cwd=cwd,
2929 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
2926 stdin=subprocess.PIPE, stdout=subprocess.PIPE,
2930 stderr=subprocess.STDOUT)
2927 stderr=subprocess.STDOUT)
2931 out, _err = proc.communicate()
2928 out, _err = proc.communicate()
2932 if proc.returncode != 0:
2929 if proc.returncode != 0:
2933 if PYTHON3:
2930 if PYTHON3:
2934 sys.stdout.buffer.write(out)
2931 sys.stdout.buffer.write(out)
2935 else:
2932 else:
2936 sys.stdout.write(out)
2933 sys.stdout.write(out)
2937 sys.exit(1)
2934 sys.exit(1)
2938
2935
2939 def _outputcoverage(self):
2936 def _outputcoverage(self):
2940 """Produce code coverage output."""
2937 """Produce code coverage output."""
2941 import coverage
2938 import coverage
2942 coverage = coverage.coverage
2939 coverage = coverage.coverage
2943
2940
2944 vlog('# Producing coverage report')
2941 vlog('# Producing coverage report')
2945 # chdir is the easiest way to get short, relative paths in the
2942 # chdir is the easiest way to get short, relative paths in the
2946 # output.
2943 # output.
2947 os.chdir(self._hgroot)
2944 os.chdir(self._hgroot)
2948 covdir = os.path.join(self._installdir, '..', 'coverage')
2945 covdir = os.path.join(self._installdir, '..', 'coverage')
2949 cov = coverage(data_file=os.path.join(covdir, 'cov'))
2946 cov = coverage(data_file=os.path.join(covdir, 'cov'))
2950
2947
2951 # Map install directory paths back to source directory.
2948 # Map install directory paths back to source directory.
2952 cov.config.paths['srcdir'] = ['.', self._pythondir]
2949 cov.config.paths['srcdir'] = ['.', self._pythondir]
2953
2950
2954 cov.combine()
2951 cov.combine()
2955
2952
2956 omit = [os.path.join(x, '*') for x in [self._bindir, self._testdir]]
2953 omit = [os.path.join(x, '*') for x in [self._bindir, self._testdir]]
2957 cov.report(ignore_errors=True, omit=omit)
2954 cov.report(ignore_errors=True, omit=omit)
2958
2955
2959 if self.options.htmlcov:
2956 if self.options.htmlcov:
2960 htmldir = os.path.join(self._outputdir, 'htmlcov')
2957 htmldir = os.path.join(self._outputdir, 'htmlcov')
2961 cov.html_report(directory=htmldir, omit=omit)
2958 cov.html_report(directory=htmldir, omit=omit)
2962 if self.options.annotate:
2959 if self.options.annotate:
2963 adir = os.path.join(self._outputdir, 'annotated')
2960 adir = os.path.join(self._outputdir, 'annotated')
2964 if not os.path.isdir(adir):
2961 if not os.path.isdir(adir):
2965 os.mkdir(adir)
2962 os.mkdir(adir)
2966 cov.annotate(directory=adir, omit=omit)
2963 cov.annotate(directory=adir, omit=omit)
2967
2964
2968 def _findprogram(self, program):
2965 def _findprogram(self, program):
2969 """Search PATH for a executable program"""
2966 """Search PATH for a executable program"""
2970 dpb = _bytespath(os.defpath)
2967 dpb = _bytespath(os.defpath)
2971 sepb = _bytespath(os.pathsep)
2968 sepb = _bytespath(os.pathsep)
2972 for p in osenvironb.get(b'PATH', dpb).split(sepb):
2969 for p in osenvironb.get(b'PATH', dpb).split(sepb):
2973 name = os.path.join(p, program)
2970 name = os.path.join(p, program)
2974 if os.name == 'nt' or os.access(name, os.X_OK):
2971 if os.name == 'nt' or os.access(name, os.X_OK):
2975 return name
2972 return name
2976 return None
2973 return None
2977
2974
2978 def _checktools(self):
2975 def _checktools(self):
2979 """Ensure tools required to run tests are present."""
2976 """Ensure tools required to run tests are present."""
2980 for p in self.REQUIREDTOOLS:
2977 for p in self.REQUIREDTOOLS:
2981 if os.name == 'nt' and not p.endswith('.exe'):
2978 if os.name == 'nt' and not p.endswith('.exe'):
2982 p += '.exe'
2979 p += '.exe'
2983 found = self._findprogram(p)
2980 found = self._findprogram(p)
2984 if found:
2981 if found:
2985 vlog("# Found prerequisite", p, "at", found)
2982 vlog("# Found prerequisite", p, "at", found)
2986 else:
2983 else:
2987 print("WARNING: Did not find prerequisite tool: %s " %
2984 print("WARNING: Did not find prerequisite tool: %s " %
2988 p.decode("utf-8"))
2985 p.decode("utf-8"))
2989
2986
2990 def aggregateexceptions(path):
2987 def aggregateexceptions(path):
2991 exceptions = collections.Counter()
2988 exceptions = collections.Counter()
2992
2989
2993 for f in os.listdir(path):
2990 for f in os.listdir(path):
2994 with open(os.path.join(path, f), 'rb') as fh:
2991 with open(os.path.join(path, f), 'rb') as fh:
2995 data = fh.read().split(b'\0')
2992 data = fh.read().split(b'\0')
2996 if len(data) != 4:
2993 if len(data) != 4:
2997 continue
2994 continue
2998
2995
2999 exc, mainframe, hgframe, hgline = data
2996 exc, mainframe, hgframe, hgline = data
3000 exc = exc.decode('utf-8')
2997 exc = exc.decode('utf-8')
3001 mainframe = mainframe.decode('utf-8')
2998 mainframe = mainframe.decode('utf-8')
3002 hgframe = hgframe.decode('utf-8')
2999 hgframe = hgframe.decode('utf-8')
3003 hgline = hgline.decode('utf-8')
3000 hgline = hgline.decode('utf-8')
3004 exceptions[(hgframe, hgline, exc)] += 1
3001 exceptions[(hgframe, hgline, exc)] += 1
3005
3002
3006 return exceptions
3003 return exceptions
3007
3004
3008 if __name__ == '__main__':
3005 if __name__ == '__main__':
3009 runner = TestRunner()
3006 runner = TestRunner()
3010
3007
3011 try:
3008 try:
3012 import msvcrt
3009 import msvcrt
3013 msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
3010 msvcrt.setmode(sys.stdin.fileno(), os.O_BINARY)
3014 msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
3011 msvcrt.setmode(sys.stdout.fileno(), os.O_BINARY)
3015 msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
3012 msvcrt.setmode(sys.stderr.fileno(), os.O_BINARY)
3016 except ImportError:
3013 except ImportError:
3017 pass
3014 pass
3018
3015
3019 sys.exit(runner.run(sys.argv[1:]))
3016 sys.exit(runner.run(sys.argv[1:]))
@@ -1,108 +1,108
1 """test line matching with some failing examples and some which warn
1 """test line matching with some failing examples and some which warn
2
2
3 run-test.t only checks positive matches and can not see warnings
3 run-test.t only checks positive matches and can not see warnings
4 (both by design)
4 (both by design)
5 """
5 """
6 from __future__ import absolute_import, print_function
6 from __future__ import absolute_import, print_function
7
7
8 import doctest
8 import doctest
9 import os
9 import os
10 import re
10 import re
11 # this is hack to make sure no escape characters are inserted into the output
11 # this is hack to make sure no escape characters are inserted into the output
12 if 'TERM' in os.environ:
12 if 'TERM' in os.environ:
13 del os.environ['TERM']
13 del os.environ['TERM']
14 run_tests = __import__('run-tests')
14 run_tests = __import__('run-tests')
15
15
16 def prn(ex):
16 def prn(ex):
17 m = ex.args[0]
17 m = ex.args[0]
18 if isinstance(m, str):
18 if isinstance(m, str):
19 print(m)
19 print(m)
20 else:
20 else:
21 print(m.decode('utf-8'))
21 print(m.decode('utf-8'))
22
22
23 def lm(expected, output):
23 def lm(expected, output):
24 r"""check if output matches expected
24 r"""check if output matches expected
25
25
26 does it generally work?
26 does it generally work?
27 >>> lm(b'H*e (glob)\n', b'Here\n')
27 >>> lm(b'H*e (glob)\n', b'Here\n')
28 True
28 True
29
29
30 fail on bad test data
30 fail on bad test data
31 >>> try: lm(b'a\n',b'a')
31 >>> try: lm(b'a\n',b'a')
32 ... except AssertionError as ex: print(ex)
32 ... except AssertionError as ex: print(ex)
33 missing newline
33 missing newline
34 >>> try: lm(b'single backslash\n', b'single \backslash\n')
34 >>> try: lm(b'single backslash\n', b'single \backslash\n')
35 ... except AssertionError as ex: prn(ex)
35 ... except AssertionError as ex: prn(ex)
36 single backslash or unknown char
36 single backslash or unknown char
37 """
37 """
38 assert (expected.endswith(b'\n')
38 assert (expected.endswith(b'\n')
39 and output.endswith(b'\n')), 'missing newline'
39 and output.endswith(b'\n')), 'missing newline'
40 assert not re.search(br'[^ \w\\/\r\n()*?]', expected + output), \
40 assert not re.search(br'[^ \w\\/\r\n()*?]', expected + output), \
41 b'single backslash or unknown char'
41 b'single backslash or unknown char'
42 test = run_tests.TTest(b'test-run-test.t', b'.', b'.')
42 test = run_tests.TTest(b'test-run-test.t', b'.', b'.')
43 match = test.linematch(expected, output)
43 match = test.linematch(expected, output)
44 if isinstance(match, str):
44 if isinstance(match, str):
45 return 'special: ' + match
45 return 'special: ' + match
46 elif isinstance(match, bytes):
46 elif isinstance(match, bytes):
47 return 'special: ' + match.decode('utf-8')
47 return 'special: ' + match.decode('utf-8')
48 else:
48 else:
49 return bool(match) # do not return match object
49 return bool(match) # do not return match object
50
50
51 def wintests():
51 def wintests():
52 r"""test matching like running on windows
52 r"""test matching like running on windows
53
53
54 enable windows matching on any os
54 enable windows matching on any os
55 >>> _osaltsep = os.altsep
55 >>> _osaltsep = os.altsep
56 >>> os.altsep = True
56 >>> os.altsep = True
57 >>> _osname = os.name
57 >>> _osname = os.name
58 >>> os.name = 'nt'
58 >>> os.name = 'nt'
59
59
60 valid match on windows
60 valid match on windows
61 >>> lm(b'g/a*/d (glob)\n', b'g\\abc/d\n')
61 >>> lm(b'g/a*/d (glob)\n', b'g\\abc/d\n')
62 True
62 True
63
63
64 direct matching, glob unnecessary
64 direct matching, glob unnecessary
65 >>> lm(b'g/b (glob)\n', b'g/b\n')
65 >>> lm(b'g/b (glob)\n', b'g/b\n')
66 'special: -glob'
66 'special: -glob'
67
67
68 missing glob
68 missing glob
69 >>> lm(b'/g/c/d/fg\n', b'\\g\\c\\d/fg\n')
69 >>> lm(b'/g/c/d/fg\n', b'\\g\\c\\d/fg\n')
70 'special: +glob'
70 True
71 >>> lm(b'/g/c/d/fg\n', b'\\g\\c\\d\\fg\r\n')
71 >>> lm(b'/g/c/d/fg\n', b'\\g\\c\\d\\fg\r\n')
72 'special: +glob'
72 True
73
73
74 restore os.altsep
74 restore os.altsep
75 >>> os.altsep = _osaltsep
75 >>> os.altsep = _osaltsep
76 >>> os.name = _osname
76 >>> os.name = _osname
77 """
77 """
78 pass
78 pass
79
79
80 def otherostests():
80 def otherostests():
81 r"""test matching like running on non-windows os
81 r"""test matching like running on non-windows os
82
82
83 disable windows matching on any os
83 disable windows matching on any os
84 >>> _osaltsep = os.altsep
84 >>> _osaltsep = os.altsep
85 >>> os.altsep = False
85 >>> os.altsep = False
86 >>> _osname = os.name
86 >>> _osname = os.name
87 >>> os.name = 'nt'
87 >>> os.name = 'nt'
88
88
89 backslash does not match slash
89 backslash does not match slash
90 >>> lm(b'h/a* (glob)\n', b'h\\ab\n')
90 >>> lm(b'h/a* (glob)\n', b'h\\ab\n')
91 False
91 False
92
92
93 direct matching glob can not be recognized
93 direct matching glob can not be recognized
94 >>> lm(b'h/b (glob)\n', b'h/b\n')
94 >>> lm(b'h/b (glob)\n', b'h/b\n')
95 True
95 True
96
96
97 missing glob can not not be recognized
97 missing glob can not not be recognized
98 >>> lm(b'/h/c/df/g/\n', b'\\h/c\\df/g\\\n')
98 >>> lm(b'/h/c/df/g/\n', b'\\h/c\\df/g\\\n')
99 False
99 False
100
100
101 restore os.altsep
101 restore os.altsep
102 >>> os.altsep = _osaltsep
102 >>> os.altsep = _osaltsep
103 >>> os.name = _osname
103 >>> os.name = _osname
104 """
104 """
105 pass
105 pass
106
106
107 if __name__ == '__main__':
107 if __name__ == '__main__':
108 doctest.testmod()
108 doctest.testmod()
General Comments 0
You need to be logged in to leave comments. Login now