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