task: Fix flaky TestInfrahubNode::test_task_query test

[ogenstad]

Component

No response

Task Description

The TestInfrahubNode::test_task_query test occasionally fails with this error:

    async def test_task_query(
        self, client: InfrahubClient, base_dataset: None, set_pagination_size3: AsyncGenerator[None, None]
    ) -> None:
        nbr_tasks = await client.task.count()
        assert nbr_tasks
    
        tasks = await client.task.filter(filter=TaskFilter(state=[TaskState.COMPLETED]))
        assert tasks
        task_ids = [task.id for task in tasks]
    
        # Query Tasks using Parallel mode
        tasks_parallel = await client.task.filter(filter=TaskFilter(state=[TaskState.COMPLETED]), parallel=True)
        assert tasks_parallel
        task_parallel_ids = [task.id for task in tasks_parallel]
    
        # Additional tasks might have been completed between the two queries
        # validate that we get at least as many tasks as in the first query
        # and that all task IDs from the first query are present in the second one
        assert len(tasks_parallel) >= len(tasks)
>       assert set(task_ids).issubset(set(task_parallel_ids))
E       AssertionError: assert False
E        +  where False = <built-in method issubset of set object at 0x7c864cd32260>({'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', '21f155ea-b6c4-40da-8a27-14c5ceb46b6c', ...})
E        +    where <built-in method issubset of set object at 0x7c864cd32260> = {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1a701c02-dfee-41ed-9f84-cd1c4cb001f3', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', ...}.issubset
E        +      where {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1a701c02-dfee-41ed-9f84-cd1c4cb001f3', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', ...} = set(['9a3c5c58-3f38-42a4-b6ab-a08a73c0dfbd', 'd802f5c7-040c-4857-8d96-4c25dfd64346', '77b22e89-8b64-4e89-9586-c1d8df165b47', '68aa0b2c-73de-4053-9fda-1f278b98c017', '9fa47ebe-ef31-4125-a483-9b0f1e317a0b', 'dea0eceb-bfda-4996-9be5-40cf33e91a1f', ...])
E        +    and   {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', '21f155ea-b6c4-40da-8a27-14c5ceb46b6c', ...} = set(['68aa0b2c-73de-4053-9fda-1f278b98c017', '9fa47ebe-ef31-4125-a483-9b0f1e317a0b', 'dea0eceb-bfda-4996-9be5-40cf33e91a1f', '9a3c5c58-3f38-42a4-b6ab-a08a73c0dfbd', 'd802f5c7-040c-4857-8d96-4c25dfd64346', '77b22e89-8b64-4e89-9586-c1d8df165b47', ...])

FAILED tests/integration/test_infrahub_client.py::TestInfrahubNode::test_task_query - AssertionError: assert False
 +  where False = <built-in method issubset of set object at 0x7c864cd32260>({'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', '21f155ea-b6c4-40da-8a27-14c5ceb46b6c', ...})
 +    where <built-in method issubset of set object at 0x7c864cd32260> = {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1a701c02-dfee-41ed-9f84-cd1c4cb001f3', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', ...}.issubset
 +      where {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1a701c02-dfee-41ed-9f84-cd1c4cb001f3', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', ...} = set(['9a3c5c58-3f38-42a4-b6ab-a08a73c0dfbd', 'd802f5c7-040c-4857-8d96-4c25dfd64346', '77b22e89-8b64-4e89-9586-c1d8df165b47', '68aa0b2c-73de-4053-9fda-1f278b98c017', '9fa47ebe-ef31-4125-a483-9b0f1e317a0b', 'dea0eceb-bfda-4996-9be5-40cf33e91a1f', ...])
 +    and   {'01babefa-8e14-41e0-8727-f2096d5b6cb6', '07187252-4e41-4c4c-8092-826276b26f12', '0a5ff2ba-a05c-43a4-9fac-53ed09dd834c', '1d3a5dee-3a52-4d6b-8c5c-ed04b66749d5', '20caefea-5869-48cc-ac08-be1ce1302ac6', '21f155ea-b6c4-40da-8a27-14c5ceb46b6c', ...} = set(['68aa0b2c-73de-4053-9fda-1f278b98c017', '9fa47ebe-ef31-4125-a483-9b0f1e317a0b', 'dea0eceb-bfda-4996-9be5-40cf33e91a1f', '9a3c5c58-3f38-42a4-b6ab-a08a73c0dfbd', 'd802f5c7-040c-4857-8d96-4c25dfd64346', '77b22e89-8b64-4e89-9586-c1d8df165b47', ...])
======= 1 failed, 87 passed, 2 xfailed, 6 warnings in 1174.01s (0:19:34) =======

Investigate why this is happening, possibly a timing issue.