mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 00:45:18 +03:00
98d9269874
Summary: Create a fork of the Mercurial code that we can use to build server rpms. The hg servers will continue to exist for a few more months while we move the darkstorm and ediscovery use cases off them. In the mean time, we want to start making breaking changes to the client, so let's create a stable copy of the hg code to produce rpms for the hg servers. The fork is based off c7770c78d, the latest hg release. This copies the files as is, then adds some minor tweaks to get it to build: - Disables some lint checks that appear to be bypassed by path - sed replace eden/scm with eden/hg-server - Removed a dependency on scm/telemetry from the edenfs-client tests since scm/telemetry pulls in the original eden/scm/lib/configparser which conflicts with the hg-server conflict parser. allow-large-files Reviewed By: quark-zju Differential Revision: D27632557 fbshipit-source-id: b2f442f4ec000ea08e4d62de068750832198e1f4
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)
|