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(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)