# HG changeset patch
# User Siddharth Agarwal <sid0@fb.com>
# Date 1444162411 25200
# Node ID e8564e04382d49cc2e50f1a2d0d6e62fd240ad1a
# Parent  ca778cbe94f38d96e074e803971c5ce583e75654
lock: add a way to prevent locks from being inherited

We want to prevent locks from being inherited sometimes (e.g. when there's a
currently running transaction, which will break a lot of assumptions we're
making in here.)

diff -r ca778cbe94f3 -r e8564e04382d mercurial/lock.py
--- a/mercurial/lock.py	Tue Oct 06 15:55:50 2015 -0700
+++ b/mercurial/lock.py	Tue Oct 06 13:13:31 2015 -0700
@@ -40,7 +40,7 @@
     _host = None
 
     def __init__(self, vfs, file, timeout=-1, releasefn=None, acquirefn=None,
-                 desc=None, parentlock=None):
+                 desc=None, inheritchecker=None, parentlock=None):
         self.vfs = vfs
         self.f = file
         self.held = 0
@@ -48,6 +48,7 @@
         self.releasefn = releasefn
         self.acquirefn = acquirefn
         self.desc = desc
+        self._inheritchecker = inheritchecker
         self.parentlock = parentlock
         self._parentheld = False
         self._inherited = False
@@ -186,6 +187,8 @@
         if self._inherited:
             raise error.LockInheritanceContractViolation(
                 'inherit cannot be called while lock is already inherited')
+        if self._inheritchecker is not None:
+            self._inheritchecker()
         if self.releasefn:
             self.releasefn()
         if self._parentheld:
diff -r ca778cbe94f3 -r e8564e04382d tests/test-lock.py
--- a/tests/test-lock.py	Tue Oct 06 15:55:50 2015 -0700
+++ b/tests/test-lock.py	Tue Oct 06 13:13:31 2015 -0700
@@ -8,6 +8,7 @@
 import unittest
 
 from mercurial import (
+    error,
     lock,
     scmutil,
 )
@@ -250,5 +251,21 @@
 
         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)
+
+        def tryinherit():
+            with lock.inherit() as lockname:
+                pass
+
+        self.assertRaises(error.LockInheritanceContractViolation, tryinherit)
+
+        lock.release()
+
 if __name__ == '__main__':
     silenttestrunner.main(__name__)