Run tests on infrahub 1.1, fix issue with branch async

Author: dgarros

infrahub_sdk/branch.py

@@ -40,6 +40,7 @@ class BranchData(BaseModel):
 
 
 MUTATION_QUERY_DATA = {"ok": None, "object": BRANCH_DATA}
+MUTATION_QUERY_TASK = {"ok": None, "task": {"id": None}}
 
 QUERY_ALL_BRANCHES_DATA = {"Branch": BRANCH_DATA}
 
@@ -119,13 +120,14 @@ async def create(
             },
         }
 
-        query = Mutation(mutation="BranchCreate", input_data=input_data, query=MUTATION_QUERY_DATA)
+        mutation_query = MUTATION_QUERY_TASK if background_execution else MUTATION_QUERY_DATA
+        query = Mutation(mutation="BranchCreate", input_data=input_data, query=mutation_query)
         response = await self.client.execute_graphql(query=query.render(), tracker="mutation-branch-create")
 
         # Make sure server version is recent enough to support background execution, as previously
         # using background_execution=True had no effect.
         if background_execution and "task" in response["BranchCreate"]:
-            return BranchData(**response["BranchCreate"]["task"]["id"])
+            return response["BranchCreate"]["task"]["id"]
         return BranchData(**response["BranchCreate"]["object"])
 
     async def delete(self, branch_name: str) -> bool:

tests/integration/test_infrahub_client.py

@@ -18,7 +18,7 @@
 class TestInfrahubNode(TestInfrahubDockerClient, SchemaAnimal):
     @pytest.fixture(scope="class")
     def infrahub_version(self) -> str:
-        return "1.0.10"
+        return "1.1.0"
 
     @pytest.fixture(scope="class")
     async def base_dataset(

tests/integration/test_node.py

@@ -13,7 +13,7 @@
 class TestInfrahubNode(TestInfrahubDockerClient, SchemaCarPerson):
     @pytest.fixture(scope="class")
     def infrahub_version(self) -> str:
-        return "1.0.10"
+        return "1.1.0"
 
     @pytest.fixture(scope="class")
     async def initial_schema(self, default_branch: str, client: InfrahubClient, schema_base: SchemaRoot) -> None: