minor fixes

Author: LucasG0

backend/infrahub/api/query.py

@@ -26,7 +26,7 @@
 from infrahub.graphql.utils import extract_data
 from infrahub.groups.models import RequestGraphQLQueryGroupUpdate
 from infrahub.log import get_logger
-from infrahub.workflows.catalogue import REQUEST_GRAPHQL_QUERY_GROUP_UPDATE
+from infrahub.workflows.catalogue import UPDATE_GRAPHQL_QUERY_GROUP
 
 if TYPE_CHECKING:
     from infrahub.auth import AccountSession
@@ -115,7 +115,7 @@ async def execute_query(
             subscribers=sorted(subscribers),
             params=params,
         )
-        await service.workflow.submit_workflow(workflow=REQUEST_GRAPHQL_QUERY_GROUP_UPDATE, parameters={"model": model})
+        await service.workflow.submit_workflow(workflow=UPDATE_GRAPHQL_QUERY_GROUP, parameters={"model": model})
 
     return response_payload
 

backend/infrahub/graphql/tasks.py

@@ -1,13 +1,14 @@
+from infrahub_sdk.groups import group_add_subscriber
 from infrahub_sdk.utils import dict_hash
+from prefect import flow
 
 from infrahub.core.constants import InfrahubKind
-from infrahub.graphql.tasks import _group_add_subscriber
 from infrahub.groups.models import RequestGraphQLQueryGroupUpdate
 from infrahub.services import services
 from infrahub.workflows.utils import add_branch_tag
 
 
-# @flow(name="update_graphql_query_group")
+@flow(name="update_graphql_query_group")
 async def update_graphql_query_group(model: RequestGraphQLQueryGroupUpdate) -> None:
     """Create or Update a GraphQLQueryGroup."""
 
@@ -30,6 +31,6 @@ async def update_graphql_query_group(model: RequestGraphQLQueryGroupUpdate) -> N
     await group.save(allow_upsert=True)
 
     if model.subscribers:
-        await _group_add_subscriber(
+        await group_add_subscriber(
             client=service.client, group=group, subscribers=model.subscribers, branch=model.branch
         )

backend/infrahub/groups/tasks.py

@@ -187,7 +187,7 @@
     function="cancel_proposed_changes_branch",
 )
 
-REQUEST_GRAPHQL_QUERY_GROUP_UPDATE = WorkflowDefinition(
+UPDATE_GRAPHQL_QUERY_GROUP = WorkflowDefinition(
     name="update_graphql_query_group",
     type=WorkflowType.INTERNAL,
     module="infrahub.graphql.tasks",
@@ -221,5 +221,5 @@
     TRIGGER_GENERATOR_DEFINITION_RUN,
     BRANCH_CANCEL_PROPOSED_CHANGES,
     REQUEST_GENERATOR_DEFINITION_RUN,
-    REQUEST_GRAPHQL_QUERY_GROUP_UPDATE,
+    UPDATE_GRAPHQL_QUERY_GROUP,
 ]

backend/infrahub/workflows/catalogue.py

@@ -7,7 +7,7 @@
 
 from infrahub.core.initialization import create_branch
 from infrahub.groups.models import RequestGraphQLQueryGroupUpdate
-from infrahub.workflows.catalogue import REQUEST_GRAPHQL_QUERY_GROUP_UPDATE
+from infrahub.workflows.catalogue import UPDATE_GRAPHQL_QUERY_GROUP
 
 if TYPE_CHECKING:
     from fastapi.testclient import TestClient
@@ -72,7 +72,7 @@ async def test_query_endpoint_group_no_params(
 
         expected_calls = [
             call(
-                workflow=REQUEST_GRAPHQL_QUERY_GROUP_UPDATE,
+                workflow=UPDATE_GRAPHQL_QUERY_GROUP,
                 parameters={"model": model},
             ),
         ]
@@ -113,7 +113,7 @@ async def test_query_endpoint_group_params(
 
         expected_calls = [
             call(
-                workflow=REQUEST_GRAPHQL_QUERY_GROUP_UPDATE,
+                workflow=UPDATE_GRAPHQL_QUERY_GROUP,
                 parameters={"model": model},
             ),
         ]