mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +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
345 lines
10 KiB
Python
345 lines
10 KiB
Python
from __future__ import absolute_import
|
|
|
|
import copy
|
|
import errno
|
|
import os
|
|
import tempfile
|
|
import types
|
|
import unittest
|
|
|
|
import silenttestrunner
|
|
from edenscm.mercurial import (
|
|
encoding,
|
|
error,
|
|
extensions,
|
|
lock,
|
|
pycompat,
|
|
util,
|
|
vfs as vfsmod,
|
|
)
|
|
from hghave import require
|
|
|
|
|
|
require(["py2"])
|
|
|
|
|
|
testlockname = "testlock"
|
|
|
|
# work around http://bugs.python.org/issue1515
|
|
if types.MethodType not in copy._deepcopy_dispatch:
|
|
|
|
def _deepcopy_method(x, memo):
|
|
return type(x)(x.__func__, copy.deepcopy(x.__self__, memo), x.im_class)
|
|
|
|
copy._deepcopy_dispatch[types.MethodType] = _deepcopy_method
|
|
|
|
|
|
class lockwrapper(lock.lock):
|
|
def __init__(self, pidoffset, *args, **kwargs):
|
|
# lock.lock.__init__() calls lock(), so the pidoffset assignment needs
|
|
# to be earlier
|
|
self._pidoffset = pidoffset
|
|
super(lockwrapper, self).__init__(*args, **kwargs)
|
|
|
|
def _getpid(self):
|
|
pid = super(lockwrapper, self)._getpid()
|
|
return "%s/%s" % (pid, self._pidoffset)
|
|
|
|
|
|
class teststate(object):
|
|
def __init__(self, testcase, dir, pidoffset=0):
|
|
self._testcase = testcase
|
|
self._acquirecalled = False
|
|
self._releasecalled = False
|
|
self._postreleasecalled = False
|
|
self.vfs = vfsmod.vfs(dir, audit=False)
|
|
self._pidoffset = pidoffset
|
|
|
|
def makelock(self, *args, **kwargs):
|
|
l = lockwrapper(
|
|
self._pidoffset,
|
|
self.vfs,
|
|
testlockname,
|
|
releasefn=self.releasefn,
|
|
acquirefn=self.acquirefn,
|
|
*args,
|
|
**kwargs
|
|
)
|
|
l.postrelease.append(self.postreleasefn)
|
|
return l
|
|
|
|
def acquirefn(self):
|
|
self._acquirecalled = True
|
|
|
|
def releasefn(self):
|
|
self._releasecalled = True
|
|
|
|
def postreleasefn(self):
|
|
self._postreleasecalled = True
|
|
|
|
def assertacquirecalled(self, called):
|
|
self._testcase.assertEqual(
|
|
self._acquirecalled,
|
|
called,
|
|
"expected acquire to be %s but was actually %s"
|
|
% (self._tocalled(called), self._tocalled(self._acquirecalled)),
|
|
)
|
|
|
|
def resetacquirefn(self):
|
|
self._acquirecalled = False
|
|
|
|
def assertreleasecalled(self, called):
|
|
self._testcase.assertEqual(
|
|
self._releasecalled,
|
|
called,
|
|
"expected release to be %s but was actually %s"
|
|
% (self._tocalled(called), self._tocalled(self._releasecalled)),
|
|
)
|
|
|
|
def assertpostreleasecalled(self, called):
|
|
self._testcase.assertEqual(
|
|
self._postreleasecalled,
|
|
called,
|
|
"expected postrelease to be %s but was actually %s"
|
|
% (self._tocalled(called), self._tocalled(self._postreleasecalled)),
|
|
)
|
|
|
|
def assertlockexists(self, exists):
|
|
actual = self.vfs.lexists(testlockname)
|
|
self._testcase.assertEqual(
|
|
actual,
|
|
exists,
|
|
"expected lock to %s but actually did %s"
|
|
% (self._toexists(exists), self._toexists(actual)),
|
|
)
|
|
|
|
def _tocalled(self, called):
|
|
if called:
|
|
return "called"
|
|
else:
|
|
return "not called"
|
|
|
|
def _toexists(self, exists):
|
|
if exists:
|
|
return "exist"
|
|
else:
|
|
return "not exist"
|
|
|
|
|
|
class testlock(unittest.TestCase):
|
|
def testlock(self):
|
|
state = teststate(self, tempfile.mkdtemp(dir=os.getcwd()))
|
|
lock = state.makelock()
|
|
state.assertacquirecalled(True)
|
|
lock.release()
|
|
state.assertreleasecalled(True)
|
|
state.assertpostreleasecalled(True)
|
|
state.assertlockexists(False)
|
|
|
|
def testrecursivelock(self):
|
|
state = teststate(self, tempfile.mkdtemp(dir=os.getcwd()))
|
|
lock = state.makelock()
|
|
state.assertacquirecalled(True)
|
|
|
|
state.resetacquirefn()
|
|
lock.lock()
|
|
# recursive lock should not call acquirefn again
|
|
state.assertacquirecalled(False)
|
|
|
|
lock.release() # brings lock refcount down from 2 to 1
|
|
state.assertreleasecalled(False)
|
|
state.assertpostreleasecalled(False)
|
|
state.assertlockexists(True)
|
|
|
|
lock.release() # releases the lock
|
|
state.assertreleasecalled(True)
|
|
state.assertpostreleasecalled(True)
|
|
state.assertlockexists(False)
|
|
|
|
def testlockfork(self):
|
|
state = teststate(self, tempfile.mkdtemp(dir=os.getcwd()))
|
|
lock = state.makelock()
|
|
state.assertacquirecalled(True)
|
|
|
|
# fake a fork
|
|
forklock = copy.deepcopy(lock)
|
|
forklock._pidoffset = 1
|
|
forklock.release()
|
|
state.assertreleasecalled(False)
|
|
state.assertpostreleasecalled(False)
|
|
state.assertlockexists(True)
|
|
|
|
# release the actual lock
|
|
lock.release()
|
|
state.assertreleasecalled(True)
|
|
state.assertpostreleasecalled(True)
|
|
state.assertlockexists(False)
|
|
|
|
def testinheritlock(self):
|
|
d = tempfile.mkdtemp(dir=os.getcwd())
|
|
parentstate = teststate(self, d)
|
|
parentlock = parentstate.makelock()
|
|
parentstate.assertacquirecalled(True)
|
|
|
|
# set up lock inheritance
|
|
with parentlock.inherit() as lockname:
|
|
parentstate.assertreleasecalled(True)
|
|
parentstate.assertpostreleasecalled(False)
|
|
parentstate.assertlockexists(True)
|
|
|
|
childstate = teststate(self, d, pidoffset=1)
|
|
childlock = childstate.makelock(parentlock=lockname)
|
|
childstate.assertacquirecalled(True)
|
|
|
|
childlock.release()
|
|
childstate.assertreleasecalled(True)
|
|
childstate.assertpostreleasecalled(False)
|
|
childstate.assertlockexists(True)
|
|
|
|
parentstate.resetacquirefn()
|
|
|
|
parentstate.assertacquirecalled(True)
|
|
|
|
parentlock.release()
|
|
parentstate.assertreleasecalled(True)
|
|
parentstate.assertpostreleasecalled(True)
|
|
parentstate.assertlockexists(False)
|
|
|
|
def testmultilock(self):
|
|
d = tempfile.mkdtemp(dir=os.getcwd())
|
|
state0 = teststate(self, d)
|
|
lock0 = state0.makelock()
|
|
state0.assertacquirecalled(True)
|
|
|
|
with lock0.inherit() as lock0name:
|
|
state0.assertreleasecalled(True)
|
|
state0.assertpostreleasecalled(False)
|
|
state0.assertlockexists(True)
|
|
|
|
state1 = teststate(self, d, pidoffset=1)
|
|
lock1 = state1.makelock(parentlock=lock0name)
|
|
state1.assertacquirecalled(True)
|
|
|
|
# from within lock1, acquire another lock
|
|
with lock1.inherit() as lock1name:
|
|
# since the file on disk is lock0's this should have the same
|
|
# name
|
|
self.assertEqual(lock0name, lock1name)
|
|
|
|
state2 = teststate(self, d, pidoffset=2)
|
|
lock2 = state2.makelock(parentlock=lock1name)
|
|
state2.assertacquirecalled(True)
|
|
|
|
lock2.release()
|
|
state2.assertreleasecalled(True)
|
|
state2.assertpostreleasecalled(False)
|
|
state2.assertlockexists(True)
|
|
|
|
state1.resetacquirefn()
|
|
|
|
state1.assertacquirecalled(True)
|
|
|
|
lock1.release()
|
|
state1.assertreleasecalled(True)
|
|
state1.assertpostreleasecalled(False)
|
|
state1.assertlockexists(True)
|
|
|
|
lock0.release()
|
|
|
|
def testinheritlockfork(self):
|
|
d = tempfile.mkdtemp(dir=os.getcwd())
|
|
parentstate = teststate(self, d)
|
|
parentlock = parentstate.makelock()
|
|
parentstate.assertacquirecalled(True)
|
|
|
|
# set up lock inheritance
|
|
with parentlock.inherit() as lockname:
|
|
childstate = teststate(self, d, pidoffset=1)
|
|
childlock = childstate.makelock(parentlock=lockname)
|
|
childstate.assertacquirecalled(True)
|
|
|
|
# fork the child lock
|
|
forkchildlock = copy.deepcopy(childlock)
|
|
forkchildlock._pidoffset += 1
|
|
forkchildlock.release()
|
|
childstate.assertreleasecalled(False)
|
|
childstate.assertpostreleasecalled(False)
|
|
childstate.assertlockexists(True)
|
|
|
|
# release the child lock
|
|
childlock.release()
|
|
childstate.assertreleasecalled(True)
|
|
childstate.assertpostreleasecalled(False)
|
|
childstate.assertlockexists(True)
|
|
|
|
parentlock.release()
|
|
|
|
def testinheritcheck(self):
|
|
d = tempfile.mkdtemp(dir=os.getcwd())
|
|
state = teststate(self, d)
|
|
|
|
def check():
|
|
raise error.LockInheritanceContractViolation("check failed")
|
|
|
|
lock = state.makelock(inheritchecker=check)
|
|
state.assertacquirecalled(True)
|
|
|
|
with self.assertRaises(error.LockInheritanceContractViolation):
|
|
with lock.inherit():
|
|
pass
|
|
|
|
lock.release()
|
|
|
|
def testfrequentlockunlock(self):
|
|
"""This tests whether lock acquisition fails as expected, even if
|
|
(1) lock can't be acquired (makelock fails by EEXIST), and
|
|
(2) lockinfo can't be read in (readlock fails by ENOENT) while
|
|
retrying 5 times.
|
|
"""
|
|
|
|
d = tempfile.mkdtemp(dir=os.getcwd())
|
|
state = teststate(self, d)
|
|
|
|
def emulatefrequentlock(*args, **kwargs):
|
|
raise OSError(errno.EEXIST, "File exists")
|
|
|
|
def emulatefrequentunlock(*args, **kwargs):
|
|
raise OSError(errno.ENOENT, "No such file or directory")
|
|
|
|
state.vfs.makelock = emulatefrequentlock
|
|
state.vfs.readlock = emulatefrequentunlock
|
|
|
|
try:
|
|
state.makelock(timeout=0)
|
|
self.fail("unexpected lock acquisition")
|
|
except error.LockHeld as why:
|
|
self.assertTrue(why.errno == errno.ETIMEDOUT)
|
|
self.assertTrue(why.lockinfo == lock.emptylockinfo)
|
|
state.assertlockexists(False)
|
|
|
|
|
|
if not pycompat.iswindows:
|
|
|
|
class testposixmakelock(unittest.TestCase):
|
|
def testremovesymlinkplaceholder(self):
|
|
class SpecificError(Exception):
|
|
pass
|
|
|
|
# Rename is the last step of makelock. Make it fail.
|
|
def _failrename(orig, src, dst):
|
|
raise SpecificError()
|
|
|
|
testtmp = encoding.environ.get("TESTTMP")
|
|
lockpath = os.path.join(testtmp, "testlock")
|
|
with extensions.wrappedfunction(
|
|
os, "rename", _failrename
|
|
), self.assertRaises(SpecificError):
|
|
util.makelock("foo:%s" % os.getpid(), lockpath)
|
|
|
|
# The placeholder lock should be removed.
|
|
self.assertFalse(os.path.lexists(lockpath))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
silenttestrunner.main(__name__)
|