def test_get_task_group_summary(self, mock_resp, mock_objects): """ Test get task_group_summary with multiple tasks """ class MockQuerySet(object): def __init__(self, list_of_objects): self.items = list_of_objects self.i = 0 self.n = len(self.items) def count(self): return len(self.items) def filter(self, state=None): filtered_list = [] for item in self.items: if item['state'] == state: filtered_list.append(item) return MockQuerySet(filtered_list) mock_request = mock.MagicMock() mock_objects.return_value = MockQuerySet([{ 'id': 'mock_task', 'worker_name': 'mock', 'state': 'running' }, { 'id': 'mock_task', 'worker_name': 'mock', 'state': 'finished' }, { 'id': 'mock_task', 'worker_name': 'mock', 'state': 'waiting' }]) task_group_summary = TaskGroupSummaryView() response = task_group_summary.get(mock_request, 'mock_task') expected_content = { 'accepted': 0, 'finished': 1, 'running': 1, 'canceled': 0, 'waiting': 1, 'skipped': 0, 'suspended': 0, 'error': 0, 'total': 3 } mock_resp.assert_called_with(expected_content) self.assertTrue(response is mock_resp.return_value)
def test_get_task_group_summary_nonexistant(self, mock_resp, mock_objects): """ Test get task_group_summary with no tasks """ mock_request = mock.MagicMock() mock_objects.return_value = MockQuerySet([]) task_group_summary = TaskGroupSummaryView() response = task_group_summary.get(mock_request, 'mock_task') expected_content = {'accepted': 0, 'finished': 0, 'running': 0, 'canceled': 0, 'waiting': 0, 'skipped': 0, 'suspended': 0, 'error': 0, 'total': 0} mock_resp.assert_called_with(expected_content) self.assertTrue(response is mock_resp.return_value)
def test_get_task_group_summary(self, mock_resp, mock_objects): """ Test get task_group_summary with multiple tasks """ mock_request = mock.MagicMock() mock_objects.return_value = MockQuerySet([{'id': 'mock_task', 'worker_name': 'mock', 'state': 'running'}, {'id': 'mock_task', 'worker_name': 'mock', 'state': 'finished'}, {'id': 'mock_task', 'worker_name': 'mock', 'state': 'waiting'}]) task_group_summary = TaskGroupSummaryView() response = task_group_summary.get(mock_request, 'mock_task') expected_content = {'accepted': 0, 'finished': 1, 'running': 1, 'canceled': 0, 'waiting': 1, 'skipped': 0, 'suspended': 0, 'error': 0, 'total': 3} mock_resp.assert_called_with(expected_content) self.assertTrue(response is mock_resp.return_value)
def test_get_task_group_summary_nonexistant(self, mock_objects): """ Test get task_group_summary with no tasks """ mock_request = mock.MagicMock() mock_objects.return_value.count.return_value = 0 task_group_summary = TaskGroupSummaryView() self.assertRaises(MissingResource, task_group_summary.get, mock_request, 'mock_task')
def test_get_task_group_summary(self, mock_resp, mock_objects): """ Test get task_group_summary with multiple tasks """ class MockQuerySet(object): def __init__(self, list_of_objects): self.items = list_of_objects self.i = 0 self.n = len(self.items) def count(self): return len(self.items) def filter(self, state=None): filtered_list = [] for item in self.items: if item['state'] == state: filtered_list.append(item) return MockQuerySet(filtered_list) mock_request = mock.MagicMock() mock_objects.return_value = MockQuerySet([{'id': 'mock_task', 'worker_name': 'mock', 'state': 'running'}, {'id': 'mock_task', 'worker_name': 'mock', 'state': 'finished'}, {'id': 'mock_task', 'worker_name': 'mock', 'state': 'waiting'}]) task_group_summary = TaskGroupSummaryView() response = task_group_summary.get(mock_request, 'mock_task') expected_content = {'accepted': 0, 'finished': 1, 'running': 1, 'canceled': 0, 'waiting': 1, 'skipped': 0, 'suspended': 0, 'error': 0, 'total': 3} mock_resp.assert_called_with(expected_content) self.assertTrue(response is mock_resp.return_value)