def testInstance(self):
t = Timer()
# Mock the timestamp to some fixed value
t._time = self.mock.Mock()
t._time.return_value = 100.0
# Start the timer. Elapsed should remain 0.0 before and after start
self.assertEqual(t.elapsed, 0.0)
t.start()
self.assertEqual(t.elapsed, 0.0)
# Make sure running time is 5s
t._time.return_value = 105.0
self.assertEqual(t.elapsed, 5.0)
# Stop and make sure elapsed time is 5s as well
t.stop()
self.assertEqual(t.elapsed, 5.0)
# Bump time by 5s, but make sure elapsed stays the same
t._time.return_value = 110.0
self.assertEqual(t.elapsed, 5.0)
# Re-stop the timer, which should update stop_time and elapsed to 10s
t.stop()
self.assertEqual(t.elapsed, 10.0)
class ExecuteContext(object):
"""An abstraction used internally by various tasks to track work
Encapsulates common metrics for work that can be retried later, hooks for
signalling completion, etc"""
def __init__(self, attempt=1, item=None, deferred=None, future=None):
self.attempt = attempt
self.item = item
self.deferred = deferred
self.future = future
self.running = threading.Event()
self.timer = Timer()
def start(self):
"""Indicate that execution has started"""
if not self.running.is_set():
if self.future is not None:
self.future.set_running_or_notify_cancel()
self.timer.start()
self.running.set()
def set_result(self, result):
"""Indicate that execution has completed"""
self.timer.stop()
if self.future is not None:
self.future.set_result(result)
if self.deferred is not None:
self.deferred.callback(result)
def set_exception(self, exception):
"""Indicate that execution has failed"""
handled = False
self.timer.stop()
if self.future is not None:
self.future.set_exception(exception)
if self.deferred is not None:
unhandled = []
self.deferred.addErrback(self._unhandledErrback, unhandled)
self.deferred.errback(exception)
if not unhandled:
handled = True
return handled
@property
def elapsed(self):
"""Convenience property. Returns timer duration."""
return self.timer.elapsed
@staticmethod
def _unhandledErrback(error, unhandled):
"""Fallback errback for deferred processing"""
unhandled.append(error)
return None
def __cmp__(self, obj):
"""Custom comparators for comparing contexts' work `item`s"""
lhs, rhs = id(self), obj
if isinstance(obj, ExecuteContext):
lhs, rhs = self.item, obj.item
return cmp(lhs, rhs)
def __lt__(self, obj):
"""Override __lt__ explicitly for priority queue implementations"""
assert isinstance(obj, ExecuteContext)
return self.item < obj.item
def __eq__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item == obj.item
def __ne__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item != obj.item
def __gt__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item > obj.item
class ExecuteContext(object):
"""An abstraction used internally by various tasks to track work
Encapsulates common metrics for work that can be retried later, hooks for
signalling completion, etc"""
def __init__(self, attempt=1, item=None, deferred=None, future=None):
self.attempt = attempt
self.item = item
self.deferred = deferred
self.future = future
self.running = threading.Event()
self.timer = Timer()
def start(self):
"""Indicate that execution has started"""
if not self.running.is_set():
if self.future is not None:
self.future.set_running_or_notify_cancel()
self.timer.start()
self.running.set()
def set_result(self, result):
"""Indicate that execution has completed"""
self.timer.stop()
if self.future is not None:
self.future.set_result(result)
if self.deferred is not None:
self.deferred.callback(result)
def set_exception(self, exception):
"""Indicate that execution has failed"""
handled = False
self.timer.stop()
if self.future is not None:
self.future.set_exception(exception)
if self.deferred is not None:
unhandled = []
self.deferred.addErrback(self._unhandledErrback, unhandled)
self.deferred.errback(exception)
if not unhandled:
handled = True
return handled
@property
def elapsed(self):
"""Convenience property. Returns timer duration."""
return self.timer.elapsed
@staticmethod
def _unhandledErrback(error, unhandled):
"""Fallback errback for deferred processing"""
unhandled.append(error)
return None
def __cmp__(self, obj):
"""Custom comparators for comparing contexts' work `item`s"""
lhs, rhs = id(self), obj
if isinstance(obj, ExecuteContext):
lhs, rhs = self.item, obj.item
return cmp(lhs, rhs)
def __lt__(self, obj):
"""Override __lt__ explicitly for priority queue implementations"""
assert isinstance(obj, ExecuteContext)
return self.item < obj.item
def __eq__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item == obj.item
def __ne__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item != obj.item
def __gt__(self, obj):
assert isinstance(obj, ExecuteContext)
return self.item > obj.item
