mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +03:00
052e7c3877
Summary: Update `contrib/check-code.py` to Python 3. Mostly it was already compatible, however stricter regular expression parsing revealed a case where one of our tests wasn't working, and as a result lots of instances of `open(file).read()` existed that this test should have caught. I have fixed up most of the instances in the code, although there are many in the test suite that I have ignored for now. Reviewed By: quark-zju Differential Revision: D21427212 fbshipit-source-id: 7461a7c391e0ade947f779a2b476ca937fd24a8d
100 lines
3.2 KiB
Python
100 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# This stress test checks if the replace logic in Mercurial is really atomic
|
|
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
import os
|
|
import random
|
|
import shutil
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
|
|
|
|
CONTENT = "Aeneas was a lively fellow"
|
|
FILENAME = "hg.stresstest.atomicreplace.file"
|
|
CMD = """%s debugshell -c "from edenscm.mercurial.util import atomictempfile; f=atomictempfile('%s'); f.write('%s'); f.close()" """
|
|
|
|
|
|
def run_stress_test(n, binary, kill_median, kill_half_width):
|
|
content = CONTENT
|
|
tempdir = tempfile.mkdtemp(prefix="hg.stresstest.dir")
|
|
filename = os.path.join(tempdir, FILENAME)
|
|
py_filename = filename.replace("\\", "\\\\")
|
|
min_sleep = (kill_median - kill_half_width) / 1000.0
|
|
max_sleep = (kill_median + kill_half_width) / 1000.0
|
|
print("Will run (file content changes between iterations):")
|
|
print(" ", CMD % (binary, py_filename, content))
|
|
print(
|
|
"for %i times and kill each run after unform(%f..%f) s"
|
|
% (n, min_sleep, max_sleep)
|
|
)
|
|
print("Will check the existense of %s after each run\n" % filename)
|
|
try:
|
|
# let's create a file upfront
|
|
with open(filename, "w"):
|
|
pass
|
|
for p in range(n):
|
|
content += "!"
|
|
proc = subprocess.Popen(CMD % (binary, py_filename, content), shell=True)
|
|
tosleep = random.uniform(min_sleep, max_sleep)
|
|
time.sleep(tosleep)
|
|
proc.kill()
|
|
if not os.path.exists(filename):
|
|
print(
|
|
u"ALARM! Iteration %i failed. File not found. Slept: %fs"
|
|
% (p, tosleep)
|
|
)
|
|
finally:
|
|
shutil.rmtree(tempdir)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if os.name == "nt":
|
|
default_binary = os.path.realpath("../hg.exe")
|
|
else:
|
|
default_binary = os.path.realpath("../hg")
|
|
|
|
desc = (
|
|
"Stress-test the atomic replace logic in Mercurial.\n\n "
|
|
+ "This script runs N iterations of hg binary, asking it to "
|
|
+ "atomically create a file with some content. It kills each "
|
|
+ "process after a random period of time. After the process is"
|
|
+ "killed, the script checks the existense of the file and "
|
|
+ "reports if it is missing."
|
|
)
|
|
parser = argparse.ArgumentParser(
|
|
description=desc, formatter_class=argparse.ArgumentDefaultsHelpFormatter
|
|
)
|
|
parser.add_argument(
|
|
"-n", type=int, default=200, help="a number of iterations to run"
|
|
)
|
|
parser.add_argument(
|
|
"-k",
|
|
type=int,
|
|
default=450,
|
|
help="an average time after which to kill the hg "
|
|
"process in ms (kills will be in +/-D ms "
|
|
"interval from from this time)",
|
|
)
|
|
parser.add_argument(
|
|
"-w",
|
|
type=int,
|
|
default=50,
|
|
help="half a width of uniform distribution of " "kill times (ms)",
|
|
)
|
|
parser.add_argument(
|
|
"-b",
|
|
type=str,
|
|
default=default_binary,
|
|
help="a path to a mercurial binary to stress-test",
|
|
)
|
|
args = parser.parse_args()
|
|
run_stress_test(args.n, args.b, args.k, args.w)
|