def test_task_delete_with_results(self): task_id = self._create_task()["uuid"] db.task_result_create(task_id, {task_id: task_id}, {task_id: task_id}) res = db.task_result_get_all_by_uuid(task_id) self.assertEqual(len(res), 1) db.task_delete(task_id) res = db.task_result_get_all_by_uuid(task_id) self.assertEqual(len(res), 0)
def test_task_get_detailed(self): validation_result = { "etype": "FooError", "msg": "foo message", "trace": "foo t/b", } task1 = self._create_task({ "validation_result": validation_result, "tag": "bar" }) key = { "name": "atata", "pos": 0, "kw": { "args": { "a": "A" }, "context": { "c": "C" }, "sla": { "s": "S" }, "runner": { "r": "R", "type": "T" } } } data = { "raw": [], "sla": [{ "s": "S", "success": True }, { "1": "2", "success": True }, { "a": "A", "success": True }], "load_duration": 13, "full_duration": 42, "hooks": [], } db.task_result_create(task1["uuid"], key, data) task1_full = db.task_get_detailed(task1["uuid"]) self.assertEqual(validation_result, json.loads(task1_full["verification_log"])) self.assertEqual("bar", task1_full["tag"]) results = task1_full["results"] self.assertEqual(1, len(results)) self.assertEqual(key, results[0]["key"]) self.assertEqual(data, results[0]["data"])
def test_task_get_detailed_last(self): task1 = self._create_task() key = {"name": "atata"} data = {"a": "b", "c": "d"} db.task_result_create(task1["uuid"], key, data) task1_full = db.task_get_detailed_last() results = task1_full["results"] self.assertEqual(len(results), 1) self.assertEqual(results[0]["key"], key) self.assertEqual(results[0]["data"], data)
def test_task_result_get_all_by_uuid(self): task1 = self._create_task()["uuid"] task2 = self._create_task()["uuid"] for task_id in (task1, task2): db.task_result_create(task_id, {task_id: task_id}, {task_id: task_id}) for task_id in (task1, task2): res = db.task_result_get_all_by_uuid(task_id) data = {task_id: task_id} self.assertEqual(len(res), 1) self.assertEqual(res[0]["key"], data) self.assertEqual(res[0]["data"], data)
def test_task_result_create(self): task_id = self._create_task()["uuid"] key = { "name": "atata", "pos": 0, "kw": { "args": { "a": "A" }, "context": { "c": "C" }, "sla": { "s": "S" }, "runner": { "r": "R", "type": "T" } } } data = { "raw": [{ "error": "anError" }, { "duration": 1 }, { "duration": 2 }], "sla": [{ "s": "S", "success": True }, { "1": "2", "success": True }, { "a": "A", "success": True }], "load_duration": 13, "full_duration": 42, "hooks": [], } db.task_result_create(task_id, key, data) res = db.task_result_get_all_by_uuid(task_id) self.assertEqual(1, len(res)) self.assertEqual(data["raw"], res[0]["data"]["raw"])
def test_task_get_detailed_last(self): task1 = self._create_task() key = { "name": "atata", "pos": 0, "kw": { "args": { "a": "A" }, "context": { "c": "C" }, "sla": { "s": "S" }, "runner": { "r": "R", "type": "T" } } } data = { "raw": [], "sla": [{ "s": "S", "success": True }, { "1": "2", "success": True }, { "a": "A", "success": True }], "load_duration": 13, "full_duration": 42, "hooks": [], } db.task_result_create(task1["uuid"], key, data) task1_full = db.task_get_detailed_last() results = task1_full["results"] self.assertEqual(1, len(results)) self.assertEqual(key, results[0]["key"]) self.assertEqual(data, results[0]["data"])
def test_task_result_get_all_by_uuid(self): task1 = self._create_task()["uuid"] task2 = self._create_task()["uuid"] key = { "name": "atata", "pos": 0, "kw": { "args": { "task_id": "task_id" }, "context": { "c": "C" }, "sla": { "s": "S" }, "runner": { "r": "R", "type": "T" } } } data = { "raw": [], "sla": [{ "success": True }], "load_duration": 13, "full_duration": 42, "hooks": [], } for task_id in (task1, task2): key["kw"]["args"]["task_id"] = task_id data["sla"][0] = {"success": True} db.task_result_create(task_id, key, data) for task_id in (task1, task2): res = db.task_result_get_all_by_uuid(task_id) key["kw"]["args"]["task_id"] = task_id data["sla"][0] = {"success": True} self.assertEqual(len(res), 1) self.assertEqual(res[0]["key"], key) self.assertEqual(res[0]["data"], data)
def test_task_delete_with_results(self): task_id = self._create_task()["uuid"] key = { "name": "atata", "pos": 0, "kw": { "args": { "a": "A" }, "context": { "c": "C" }, "sla": { "s": "S" }, "runner": { "r": "R", "type": "T" } } } data = { "raw": [], "sla": [{ "s": "S", "success": True }, { "1": "2", "success": True }, { "a": "A", "success": True }], "load_duration": 13, "full_duration": 42 } db.task_result_create(task_id, key, data) res = db.task_result_get_all_by_uuid(task_id) self.assertEqual(len(res), 1) db.task_delete(task_id) res = db.task_result_get_all_by_uuid(task_id) self.assertEqual(len(res), 0)
def append_results(self, key, value): db.task_result_create(self.task["uuid"], key, value)