Add wait_until_completion to InfrahubBranchManager.create

Author: LucasG0

infrahub_sdk/branch.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import warnings
 from typing import TYPE_CHECKING, Any, Optional, Union
 from urllib.parse import urlencode
 
@@ -77,10 +78,20 @@ async def create(
         branch_name: str,
         sync_with_git: bool = True,
         description: str = "",
-        background_execution: bool = False,
+        wait_until_completion: bool = False,
+        background_execution: Optional[bool] = False,
     ) -> BranchData:
+
+        if background_execution is not None:
+            warnings.warn(
+                "`background_execution` is deprecated, please use `wait_until_completion` instead.",
+                DeprecationWarning,
+                stacklevel=2
+            )
+
+        wait_until_completion = background_execution or wait_until_completion
         input_data = {
-            "background_execution": background_execution,
+            "wait_until_completion": wait_until_completion,
             "data": {
                 "name": branch_name,
                 "description": description,
@@ -90,7 +101,8 @@ async def create(
 
         query = Mutation(mutation="BranchCreate", input_data=input_data, query=MUTATION_QUERY_DATA)
         response = await self.client.execute_graphql(query=query.render(), tracker="mutation-branch-create")
-
+        if not wait_until_completion:
+            return BranchData(**response["BranchCreate"]["task"])
         return BranchData(**response["BranchCreate"]["object"])
 
     async def delete(self, branch_name: str) -> bool: