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