def test_done_with_processed_tasks_empty_except(self): task_queue = Mock() task_queue.qsize = Mock(return_value=10) task_queue.get_nowait = Mock(side_effect=gevent_queue.Empty()) with patch('notification_pusher.logger', Mock()) as logger: notification_pusher.done_with_processed_tasks(task_queue) self.assertTrue(logger.debug.called)
def test_done_with_processed_tasks_empty_queue(self): q = Mock(None) q.qsize = Mock(return_value=42) q.get_nowait = Mock(side_effect=gevent_queue.Empty) np.done_with_processed_tasks(q) self.assertRaises(gevent_queue.Empty)
def test_done_with_processed_tasks_success(self): task_queue = Mock() task_queue.qsize = Mock(return_value=10) task = Mock() task_queue.get_nowait = Mock(return_value=(task, 'action_name')) with patch('notification_pusher.logger', Mock()) as logger: notification_pusher.done_with_processed_tasks(task_queue) self.assertTrue(logger.debug.called)
def test_done_with_processed_tasks_execution(self): task = Mock() task_queue = Mock() task_queue.qsize = Mock(return_value=3) task_queue.get_nowait = Mock(side_effect=( (task, 'action1'), (task, 'action2'), gevent_queue.Empty )) notification_pusher.done_with_processed_tasks(task_queue) assert task.action1.called assert task.action2.called assert not task.action3.called
def test_done_with_processed_tasks_execution_exception(self): task = Mock() task.action1 = Mock(side_effect=tarantool.DatabaseError) task_queue = Mock() task_queue.qsize = Mock(return_value=1) task_queue.get_nowait = Mock(side_effect=( (task, 'action1'), )) mocked_logger = Mock() original_logger = notification_pusher.logger notification_pusher.logger = mocked_logger notification_pusher.done_with_processed_tasks(task_queue) notification_pusher.logger = original_logger assert task.action1.called assert mocked_logger.exception.called