def setup_application(self):
yield super(BulkMessageApplication, self).setup_application()
wm_redis = self.redis.sub_manager('%s:window_manager' %
(self.worker_name, ))
self.window_manager = WindowManager(wm_redis,
window_size=self.max_ack_window,
flight_lifetime=self.max_ack_wait)
self.window_manager.monitor(self.on_window_key_ready,
interval=self.monitor_interval,
cleanup=self.monitor_window_cleanup,
cleanup_callback=self.on_window_cleanup)
def setUp(self):
self.persistence_helper = self.add_helper(PersistenceHelper())
redis = yield self.persistence_helper.get_redis_manager()
self.window_id = 'window_id'
# Patch the count_waiting so we can fake the race condition
self.clock = Clock()
self.patch(WindowManager, 'count_waiting', lambda _, window_id: 100)
self.wm = WindowManager(redis, window_size=10, flight_lifetime=10)
self.add_cleanup(self.wm.stop)
yield self.wm.create_window(self.window_id)
self.redis = self.wm.redis
def setUp(self):
self.persistence_helper = self.add_helper(PersistenceHelper())
redis = yield self.persistence_helper.get_redis_manager()
self.window_id = 'window_id'
# Patch the clock so we can control time
self.clock = Clock()
self.patch(WindowManager, 'get_clock', lambda _: self.clock)
self.wm = WindowManager(redis, window_size=10, flight_lifetime=10)
self.add_cleanup(self.wm.stop)
yield self.wm.create_window(self.window_id)
self.redis = self.wm.redis
