def test_display_with_keyboard_interrupt(self, mock_display): # Setup task_list = [] for i in range(0, 3): task = Task(TASK_TEMPLATE) task.task_id = 'task_%s' % i task_list.append(task) # Side effect to simulate keyboard interrupt def interrupt(context, renderer, task_id, quiet_waiting=True): if task_id == 'task_1': raise KeyboardInterrupt() else: return task_id mock_display.side_effect = interrupt # Test status._display_status(self.context, self.renderer, task_list) # Verify self.assertEqual( 2, mock_display.call_count) # not called for the third task for i, call_args in enumerate(mock_display.call_args_list): self.assertEqual(call_args[0][0], self.context) self.assertEqual(call_args[0][1], self.renderer) self.assertEqual(call_args[0][2], 'task_%s' % i) expected_quiet = i > 0 self.assertEqual(call_args[1]['quiet_waiting'], expected_quiet)
def test_display_with_keyboard_interrupt(self, mock_display): # Setup task_list = [] for i in range(0, 3): task = Task(TASK_TEMPLATE) task.task_id = 'task_%s' % i task_list.append(task) # Side effect to simulate keyboard interrupt def interrupt(context, renderer, task_id, quiet_waiting=True): if task_id == 'task_1': raise KeyboardInterrupt() else: return task_id mock_display.side_effect = interrupt # Test status._display_status(self.context, self.renderer, task_list) # Verify self.assertEqual(2, mock_display.call_count) # not called for the third task for i, call_args in enumerate(mock_display.call_args_list): self.assertEqual(call_args[0][0], self.context) self.assertEqual(call_args[0][1], self.renderer) self.assertEqual(call_args[0][2], 'task_%s' % i) expected_quiet = i > 0 self.assertEqual(call_args[1]['quiet_waiting'], expected_quiet)
def test_display_status_rejected(self): # Setup rejected_task = Task(TASK_TEMPLATE) rejected_task.response = RESPONSE_REJECTED # Test status._display_status(self.context, self.renderer, [rejected_task]) # Verify expected_tags = ['ctrl-c', 'rejected-msg', 'rejected-desc'] self.assertEqual(expected_tags, self.prompt.get_write_tags())
def test_display_status_postponed(self): # Setup postponed_task = Task(TASK_TEMPLATE) postponed_task.response = RESPONSE_POSTPONED postponed_task.state = STATE_WAITING # Test status._display_status(self.context, self.renderer, [postponed_task]) # Verify expected_tags = ['ctrl-c', 'postponed'] self.assertEqual(expected_tags, self.prompt.get_write_tags())
def test_display_status(self, mock_display): # Setup task_list = [] for i in range(0, 2): task = Task(TASK_TEMPLATE) task.task_id = 'task_%s' % i task_list.append(task) # Test status._display_status(self.context, self.renderer, task_list) # Verify self.assertEqual(2, mock_display.call_count) for i, call_args in enumerate(mock_display.call_args_list): self.assertEqual(call_args[0][0], self.context) self.assertEqual(call_args[0][1], self.renderer) self.assertEqual(call_args[0][2], 'task_%s' % i) expected_quiet = i > 0 self.assertEqual(call_args[1]['quiet_waiting'], expected_quiet)