def test__extend_results(self, mock_task_extend_results):
mock_task_extend_results.side_effect = iter([["extended_foo"],
["extended_bar"],
["extended_spam"]])
tasks_results = [{
"key": "%s_key" % k,
"sla": "%s_sla" % k,
"full_duration": "%s_full_duration" % k,
"load_duration": "%s_load_duration" % k,
"result": "%s_result" % k
} for k in ("foo", "bar", "spam")]
generic_results = [{
"id": None,
"created_at": None,
"updated_at": None,
"task_uuid": None,
"key": "%s_key" % k,
"data": {
"raw": "%s_result" % k,
"full_duration": "%s_full_duration" % k,
"load_duration": "%s_load_duration" % k,
"sla": "%s_sla" % k
}
} for k in ("foo", "bar", "spam")]
results = plot._extend_results(tasks_results)
self.assertEqual([mock.call([r]) for r in generic_results],
mock_task_extend_results.mock_calls)
self.assertEqual(["extended_foo", "extended_bar", "extended_spam"],
results)
def test__extend_results(self, mock_task_extend_results):
mock_task_extend_results.side_effect = iter(
[["extended_foo"], ["extended_bar"], ["extended_spam"]])
tasks_results = [
{"key": "%s_key" % k, "sla": "%s_sla" % k,
"full_duration": "%s_full_duration" % k,
"load_duration": "%s_load_duration" % k,
"result": "%s_result" % k} for k in ("foo", "bar", "spam")]
generic_results = [
{"id": None, "created_at": None, "updated_at": None,
"task_uuid": None, "key": "%s_key" % k,
"data": {"raw": "%s_result" % k,
"full_duration": "%s_full_duration" % k,
"load_duration": "%s_load_duration" % k,
"sla": "%s_sla" % k}} for k in ("foo", "bar", "spam")]
results = plot._extend_results(tasks_results)
self.assertEqual([mock.call([r]) for r in generic_results],
mock_task_extend_results.mock_calls)
self.assertEqual(["extended_foo", "extended_bar", "extended_spam"],
results)
def test__extend_results_empty(self):
self.assertEqual([], plot._extend_results([]))
def test__extend_results_empty(self):
self.assertEqual([], plot._extend_results([]))