mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
dfda82e492
Summary: Mostly empty lines removed and added. A few bugfixes on excessive line splitting. Reviewed By: quark-zju Differential Revision: D8199128 fbshipit-source-id: 90c1616061bfd7cfbba0b75f03f89683340374d5
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
# Translate run-tests.py tests to Python standard unittests
|
|
#
|
|
# Copyright 2018 Facebook, Inc.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2 or any later version.
|
|
from __future__ import absolute_import, division, print_function, unicode_literals
|
|
|
|
import contextlib
|
|
import glob
|
|
import os
|
|
import random
|
|
import re
|
|
import shlex
|
|
import subprocess
|
|
import unittest
|
|
|
|
|
|
try:
|
|
shlex_quote = shlex.quote # Python 3.3 and up
|
|
except AttributeError:
|
|
shlex_quote = subprocess.list2cmdline
|
|
|
|
|
|
@contextlib.contextmanager
|
|
def chdir(path):
|
|
oldpwd = os.getcwd()
|
|
try:
|
|
os.chdir(path)
|
|
yield
|
|
finally:
|
|
os.chdir(oldpwd)
|
|
|
|
|
|
def gettestmethod(name, port):
|
|
def runsingletest(self):
|
|
with chdir(self._runtests_dir):
|
|
env = os.environ.copy()
|
|
python = os.environ.get("HGTEST_PYTHON")
|
|
args = [python, "run-tests.py", "--port", "%s" % port]
|
|
args += os.getenv("HGTEST_RUNTESTS_ARGS", "").split()
|
|
hgpath = os.environ.get("HGTEST_HG")
|
|
if hgpath:
|
|
args.append("--with-hg=%s" % hgpath)
|
|
chgpath = os.environ.get("HGTEST_CHG")
|
|
if chgpath:
|
|
env["CHG"] = chgpath
|
|
# set HGDATAPATH
|
|
datapath = os.path.join(self._runtests_dir, "../mercurial")
|
|
env["HGDATAPATH"] = datapath
|
|
# set HGPYTHONPATH since PYTHONPATH might be discarded
|
|
pythonpath = os.pathsep.join([self._runtests_dir])
|
|
env["HGPYTHONPATH"] = pythonpath
|
|
# run run-tests.py for a single test
|
|
p = subprocess.Popen(
|
|
args + [name], env=env, stderr=subprocess.PIPE, stdout=subprocess.PIPE
|
|
)
|
|
out, err = p.communicate("")
|
|
returncode = p.returncode
|
|
if returncode == 80:
|
|
# Extract skipped reason from output
|
|
match = re.search("Skipped [^:]*: (.*)", err + out)
|
|
if match:
|
|
reason = match.group(1)
|
|
else:
|
|
reason = "skipped by run-tests.py"
|
|
raise unittest.SkipTest(reason)
|
|
elif returncode != 0:
|
|
raise self.failureException(err + out)
|
|
|
|
return runsingletest
|
|
|
|
|
|
class hgtests(unittest.TestCase):
|
|
@classmethod
|
|
def collecttests(cls, path):
|
|
"""scan tests in path and add them as test methods"""
|
|
if os.environ.get("HGTEST_IGNORE_BLACKLIST") == "1":
|
|
blacklist = None
|
|
else:
|
|
blacklist = re.compile("\A%s\Z" % os.environ.get("HGTEST_BLACKLIST", ""))
|
|
# Randomize the port so a stress run of a single test would be fine
|
|
port = random.randint(10000, 60000)
|
|
with chdir(path):
|
|
cls._runtests_dir = os.getcwd()
|
|
for name in glob.glob("test-*.t") + glob.glob("test-*.py"):
|
|
method_name = name.replace(".", "_").replace("-", "_")
|
|
if blacklist and blacklist.match(method_name):
|
|
continue
|
|
# Running a lot run-tests.py in parallel will trigger race
|
|
# condition of the original port detection logic. So allocate
|
|
# ports here. run-tests.py could do adjustments.
|
|
# A test needs 3 ports at most. See portneeded in run-tests.py
|
|
port += 3
|
|
setattr(cls, method_name, gettestmethod(name, port))
|
|
|
|
|
|
hgtests.collecttests(os.environ.get("HGTEST_DIR", "."))
|