def test_get_execution_tasks(self, sample_task, sample_display_task):
mock_api = MagicMock()
mock_api.task_by_id.return_value = Task(sample_task, mock_api)
display_task = Task(sample_display_task, mock_api)
execution_tasks = display_task.get_execution_tasks()
assert len(execution_tasks) == len(
sample_display_task["execution_tasks"])
def test_get_execution_tasks_with_filters(self, sample_task,
sample_display_task):
mock_api = MagicMock()
mock_api.task_by_id.return_value = Task(sample_task, mock_api)
max_return = 2
seen = 0
def cap_seen(t):
nonlocal seen
if seen >= max_return:
return False
seen += 1
return True
display_task = Task(sample_display_task, mock_api)
execution_tasks = display_task.get_execution_tasks(filter_fn=cap_seen)
assert len(execution_tasks) == max_return
