Merge pull request #4860 from opsmill/dga-20241106-fix-task-id

Fix issues with InfrahubTask filter not working with ids

Author: dgarros

backend/infrahub/graphql/queries/task.py

@@ -74,6 +74,7 @@ async def query(
         prefect_tasks = await PrefectTask.query(
             db=context.db,
             fields=fields,
+            ids=ids,
             branch=branch,
             statuses=statuses,
             tags=tags,

backend/tests/unit/graphql/queries/test_task.py

@@ -3,6 +3,7 @@
 
 import pytest
 from graphql import ExecutionResult, graphql
+from infrahub_sdk.graphql import Query
 from prefect.artifacts import ArtifactRequest
 from prefect.client.orchestration import PrefectClient, get_client
 from prefect.states import State
@@ -389,6 +390,39 @@ async def test_task_query_prefect(
     assert result.data["InfrahubTask"]["count"] == len(task_names)
 
 
+async def test_task_query_filter_id(
+    db: InfrahubDatabase, default_branch: Branch, register_core_models_schema: None, flow_runs_data
+):
+    dummy_completed_br1_db = flow_runs_data["dummy-completed-br1-db"]
+    dummy_running_br1 = flow_runs_data["dummy-running-br1"]
+
+    query = Query(
+        query={
+            "InfrahubTask": {
+                "@filters": {"ids": [str(dummy_completed_br1_db.id), str(dummy_running_br1.id)]},
+                "count": None,
+                "edges": {"node": {"id": None, "title": None}},
+            }
+        }
+    )
+
+    result = await run_query(
+        db=db,
+        branch=default_branch,
+        query=query.render(),
+        variables={},
+    )
+    assert result.errors is None
+    assert result.data
+
+    task_names = sorted([task["node"]["title"] for task in result.data["InfrahubTask"]["edges"]])
+    assert task_names == [
+        "dummy-completed-br1-db",
+        "dummy-running-br1",
+    ]
+    assert result.data["InfrahubTask"]["count"] == len(task_names)
+
+
 async def test_task_query_filter_branch(
     db: InfrahubDatabase, default_branch: Branch, register_core_models_schema: None, flow_runs_data
 ):