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