fix: crash for count users query (#1649)

kili-technology · Feb 7, 2024 · ad34fe0 · ad34fe0
1 parent 64dec2f
commit ad34fe0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/kili/adapters/kili_api_gateway/user/operation_mixin.py b/src/kili/adapters/kili_api_gateway/user/operation_mixin.py
@@ -39,7 +39,8 @@ def list_users(
 
     def count_users(self, user_filters: UserFilter) -> int:
         """Return the number of users that match the filter."""
-        result = self.graphql_client.execute(GQL_COUNT_USERS, user_where_mapper(user_filters))
+        variables = {"where": user_where_mapper(user_filters)}
+        result = self.graphql_client.execute(GQL_COUNT_USERS, variables)
         return result["data"]
 
     def get_current_user(self, fields: ListOrTuple[str]) -> Dict:

diff --git a/tests/integration/adapters/kili_api_gateway/test_user.py b/tests/integration/adapters/kili_api_gateway/test_user.py
@@ -70,12 +70,14 @@ def test_given_kili_gateway_when_querying_count_users_it_calls_proper_resolver(
     graphql_client.execute.assert_called_once_with(
         GQL_COUNT_USERS,
         {
-            "activated": None,
-            "apiKey": None,
-            "email": None,
-            "id": "fake_user_id",
-            "idIn": None,
-            "organization": {"id": None},
+            "where": {
+                "activated": None,
+                "apiKey": None,
+                "email": None,
+                "id": "fake_user_id",
+                "idIn": None,
+                "organization": {"id": None},
+            }
         },
     )