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