tests

Author: wvandeun

tests/unit/sdk/test_client.py

@@ -128,18 +128,17 @@ async def test_method_get_user_permissions(clients, mock_query_infrahub_user, cl
 @pytest.mark.parametrize("client_type", client_types)
 async def test_method_all_with_limit(clients, mock_query_repository_page1_2, client_type):
     if client_type == "standard":
-        repos = await clients.standard.all(kind="CoreRepository", limit=3)
-        assert len(clients.standard.store._store["CoreRepository"]) == 3
-
         repos = await clients.standard.all(kind="CoreRepository", populate_store=False, limit=3)
         assert not clients.standard.store._store["CoreRepository"]
-    else:
-        repos = clients.sync.all(kind="CoreRepository", limit=3)
-        assert len(clients.sync.store._store["CoreRepository"]) == 3
 
+        repos = await clients.standard.all(kind="CoreRepository", limit=3)
+        assert len(clients.standard.store._store["CoreRepository"]) == 3
+    else:
         repos = clients.sync.all(kind="CoreRepository", populate_store=False, limit=3)
         assert not clients.sync.store._store["CoreRepository"]
 
+        repos = clients.sync.all(kind="CoreRepository", limit=3)
+        assert len(clients.sync.store._store["CoreRepository"]) == 3
     assert len(repos) == 3
 
 
@@ -148,18 +147,18 @@ async def test_method_all_multiple_pages(
     clients, mock_query_repository_page1_2, mock_query_repository_page2_2, client_type
 ):
     if client_type == "standard":
-        repos = await clients.standard.all(kind="CoreRepository")
-        assert len(clients.standard.store._store["CoreRepository"]) == 5
-
         repos = await clients.standard.all(kind="CoreRepository", populate_store=False)
         assert not clients.standard.store._store["CoreRepository"]
-    else:
-        repos = clients.sync.all(kind="CoreRepository")
-        assert len(clients.sync.store._store["CoreRepository"]) == 5
 
+        repos = await clients.standard.all(kind="CoreRepository")
+        assert len(clients.standard.store._store["CoreRepository"]) == 5
+    else:
         repos = clients.sync.all(kind="CoreRepository", populate_store=False)
         assert not clients.sync.store._store["CoreRepository"]
 
+        repos = clients.sync.all(kind="CoreRepository")
+        assert len(clients.sync.store._store["CoreRepository"]) == 5
+
     assert len(repos) == 5
 
 
@@ -168,36 +167,36 @@ async def test_method_all_batching(
     clients, mock_query_location_batch_count, mock_query_location_batch, client_type, use_parallel
 ):
     if client_type == "standard":
-        locations = await clients.standard.all(kind="BuiltinLocation", parallel=use_parallel)
-        assert len(clients.standard.store._store["BuiltinLocation"]) == 30
-
         locations = await clients.standard.all(kind="BuiltinLocation", populate_store=False, parallel=use_parallel)
         assert not clients.standard.store._store["BuiltinLocation"]
-    else:
-        locations = clients.sync.all(kind="BuiltinLocation", parallel=use_parallel)
-        assert len(clients.sync.store._store["BuiltinLocation"]) == 30
 
+        locations = await clients.standard.all(kind="BuiltinLocation", parallel=use_parallel)
+        assert len(clients.standard.store._store["BuiltinLocation"]) == 30
+    else:
         locations = clients.sync.all(kind="BuiltinLocation", populate_store=False, parallel=use_parallel)
         assert not clients.sync.store._store["BuiltinLocation"]
 
+        locations = clients.sync.all(kind="BuiltinLocation", parallel=use_parallel)
+        assert len(clients.sync.store._store["BuiltinLocation"]) == 30
+
     assert len(locations) == 30
 
 
 @pytest.mark.parametrize("client_type", client_types)
 async def test_method_all_single_page(clients, mock_query_repository_page1_1, client_type):
     if client_type == "standard":
-        repos = await clients.standard.all(kind="CoreRepository")
-        assert len(clients.standard.store._store["CoreRepository"]) == 2
-
         repos = await clients.standard.all(kind="CoreRepository", populate_store=False)
         assert not clients.standard.store._store["CoreRepository"]
-    else:
-        repos = clients.sync.all(kind="CoreRepository")
-        assert len(clients.sync.store._store["CoreRepository"]) == 2
 
+        repos = await clients.standard.all(kind="CoreRepository")
+        assert len(clients.standard.store._store["CoreRepository"]) == 2
+    else:
         repos = clients.sync.all(kind="CoreRepository", populate_store=False)
         assert not clients.sync.store._store["CoreRepository"]
 
+        repos = clients.sync.all(kind="CoreRepository")
+        assert len(clients.sync.store._store["CoreRepository"]) == 2
+
     assert len(repos) == 2