def test_results_with_tasks_loaded_missing_result(self, get_multi_async): """Ensure results uses the cached tasks and yields them out when tasks are cached and there's no results. """ marker1 = FuriousAsyncMarker() context = Context(_task_ids=["1", "2", "3"]) context_result = ContextResult(context) context_result._task_cache = {"1": marker1, "2": None, "3": None} results = list(context_result.items()) results = sorted(results) self.assertEqual(results, [("1", None), ("2", None), ("3", None)]) self.assertFalse(get_multi_async.called)
def test_results_with_tasks_loaded(self, get_multi_async): """Ensure results uses the cached tasks and yields them out when tasks are cached. """ marker1 = _build_marker(payload="1", status=1) marker2 = _build_marker(payload="2", status=1) marker3 = _build_marker(payload="3", status=1) context = Context(_task_ids=["1", "2", "3"]) context_result = ContextResult(context) context_result._task_cache = {"1": marker1, "2": marker2, "3": marker3} results = list(context_result.items()) results = sorted(results) self.assertEqual(results, [("1", "1"), ("2", "2"), ("3", "3")]) self.assertFalse(get_multi_async.called)
def test_results_with_tasks_loaded_missing_result(self, get_multi_async): """Ensure results uses the cached tasks and yields them out when tasks are cached and there's no results. """ marker1 = FuriousAsyncMarker() context = Context(_task_ids=["1", "2", "3"]) context_result = ContextResult(context) context_result._task_cache = { "1": marker1, "2": None, "3": None } results = list(context_result.items()) results = sorted(results) self.assertEqual(results, [("1", None), ("2", None), ("3", None)]) self.assertFalse(get_multi_async.called)
def test_results_with_tasks_loaded(self, get_multi_async): """Ensure results uses the cached tasks and yields them out when tasks are cached. """ marker1 = _build_marker(payload="1", status=1) marker2 = _build_marker(payload="2", status=1) marker3 = _build_marker(payload="3", status=1) context = Context(_task_ids=["1", "2", "3"]) context_result = ContextResult(context) context_result._task_cache = { "1": marker1, "2": marker2, "3": marker3 } results = list(context_result.items()) results = sorted(results) self.assertEqual(results, [("1", "1"), ("2", "2"), ("3", "3")]) self.assertFalse(get_multi_async.called)