# HG changeset patch
# User Pulkit Goyal <7895pulkit@gmail.com>
# Date 2020-09-10 11:21:40
# Node ID f6c67bb4ca03630ec08160e2f9995daf024cb3fa
# Parent  037e88d453fa99665988dbc20bfb014149afb379

tests: update lockdelay.py to handle the `wait` argument

Spotted by a future change.

diff --git a/tests/lockdelay.py b/tests/lockdelay.py
--- a/tests/lockdelay.py
+++ b/tests/lockdelay.py
@@ -10,11 +10,11 @@ import time
 
 def reposetup(ui, repo):
     class delayedlockrepo(repo.__class__):
-        def lock(self):
+        def lock(self, wait=True):
             delay = float(os.environ.get('HGPRELOCKDELAY', '0.0'))
             if delay:
                 time.sleep(delay)
-            res = super(delayedlockrepo, self).lock()
+            res = super(delayedlockrepo, self).lock(wait=wait)
             delay = float(os.environ.get('HGPOSTLOCKDELAY', '0.0'))
             if delay:
                 time.sleep(delay)