From 266af6149e730d7eb20a7661fecd56ccfdbe0ccd Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@gmail.com>
Date: Wed, 27 May 2015 09:13:41 -0700
Subject: [PATCH] Allow the sleep function to be provided

---
 ChangeLog                 |  4 ++++
 fasteners/process_lock.py | 15 ++++++---------
 2 files changed, 10 insertions(+), 9 deletions(-)

diff --git a/ChangeLog b/ChangeLog
index 6203884..5d5414e 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,7 @@
+0.6:
+  - Allow the sleep function to be provided (so that
+    various alternatives other than time.sleep can
+    be used), ie eventlet.sleep (or other).
 0.5:
   - Make it possible to provide a acquisition timeout
     to the interprocess lock (which when acquisition
diff --git a/fasteners/process_lock.py b/fasteners/process_lock.py
index 97de9ec..fce7187 100644
--- a/fasteners/process_lock.py
+++ b/fasteners/process_lock.py
@@ -85,10 +85,11 @@ class _InterProcessLock(object):
     acquire the lock (and repeat).
     """
 
-    def __init__(self, path):
+    def __init__(self, path, sleep_func=time.sleep):
         self.lockfile = None
         self.path = path
         self.acquired = False
+        self.sleep_func = sleep_func
 
     def _do_acquire(self, delay_func, blocking, watch):
         attempts_iter = itertools.count(1)
@@ -124,19 +125,14 @@ class _InterProcessLock(object):
         if self.lockfile is None or self.lockfile.closed:
             self.lockfile = open(self.path, 'a')
 
-    @staticmethod
-    def _backoff_multiplier_delay(attempts, delay, max_delay):
+    def _backoff_multiplier_delay(self, attempts, delay, max_delay):
         maybe_delay = attempts * delay
         if maybe_delay < max_delay:
             actual_delay = maybe_delay
         else:
             actual_delay = max_delay
         actual_delay = max(0.0, actual_delay)
-        time.sleep(actual_delay)
-
-    def _fetch_delay_functor(self, delay, max_delay, watch):
-        return functools.partial(self._backoff_multiplier_delay,
-                                 delay=delay, max_delay=max_delay)
+        self.sleep_func(actual_delay)
 
     def acquire(self, blocking=True,
                 delay=DELAY_INCREMENT, max_delay=MAX_DELAY,
@@ -150,7 +146,8 @@ class _InterProcessLock(object):
         self._do_open()
         watch = timeutils.StopWatch(duration=timeout)
         if blocking:
-            delay_func = self._fetch_delay_functor(delay, max_delay, watch)
+            delay_func = functools.partial(self._backoff_multiplier_delay,
+                                           delay=delay, max_delay=max_delay)
         else:
             delay_func = _noop_delay
         with watch: