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