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