def test_blocking_start(self): config = DotDict() config.logger = self.logger config.idle_delay = 1 config.quit_on_empty_queue = False class MyTaskManager(TaskManager): def _responsive_sleep( self, seconds, wait_log_interval=0, wait_reason='' ): try: if self.count >= 2: self.quit = True self.count += 1 except AttributeError: self.count = 0 tm = MyTaskManager( config, task_func=Mock() ) waiting_func = Mock() tm.blocking_start(waiting_func=waiting_func) eq_( tm.task_func.call_count, 10 ) eq_(waiting_func.call_count, 0)
def test_blocking_start(self): config = DotDict() config.logger = self.logger config.idle_delay = 1 config.quit_on_empty_queue = False class MyTaskManager(TaskManager): def _responsive_sleep(self, seconds, wait_log_interval=0, wait_reason=''): try: if self.count >= 2: self.quit = True self.count += 1 except AttributeError: self.count = 0 tm = MyTaskManager(config, task_func=Mock()) waiting_func = Mock() tm.blocking_start(waiting_func=waiting_func) eq_(tm.task_func.call_count, 10) eq_(waiting_func.call_count, 0)
def test_blocking_start_with_quit_on_empty(self): config = DotDict() config.logger = self.logger config.idle_delay = 1 config.quit_on_empty_queue = True tm = TaskManager(config, task_func=Mock()) waiting_func = Mock() tm.blocking_start(waiting_func=waiting_func) eq_(tm.task_func.call_count, 10) eq_(waiting_func.call_count, 0)