Add scalars to extract operation plugin tests

mirumee · Jan 7, 2024 · badc2da · badc2da
1 parent 184f9f2
commit badc2da
Show file tree

Hide file tree

Showing 7 changed files with 49 additions and 0 deletions.
diff --git a/tests/main/clients/operations/expected_client/__init__.py b/tests/main/clients/operations/expected_client/__init__.py
@@ -10,6 +10,7 @@
     C_SUBSCRIPTION_GQL,
     GET_A_GQL,
     GET_A_WITH_FRAGMENT_GQL,
+    GET_S_GQL,
     GET_XYZ_GQL,
 )
 from .exceptions import (
@@ -26,6 +27,7 @@
     GetAWithFragmentA,
     GetAWithFragmentAValueB,
 )
+from .get_s import GetS, GetSS
 from .get_xyz import GetXYZ, GetXYZXyzTypeX, GetXYZXyzTypeY, GetXYZXyzTypeZ
 
 __all__ = [
@@ -42,13 +44,16 @@
     "FragmentY",
     "GET_A_GQL",
     "GET_A_WITH_FRAGMENT_GQL",
+    "GET_S_GQL",
     "GET_XYZ_GQL",
     "GetA",
     "GetAA",
     "GetAAValueB",
     "GetAWithFragment",
     "GetAWithFragmentA",
     "GetAWithFragmentAValueB",
+    "GetS",
+    "GetSS",
     "GetXYZ",
     "GetXYZXyzTypeX",
     "GetXYZXyzTypeY",

diff --git a/tests/main/clients/operations/expected_client/client.py b/tests/main/clients/operations/expected_client/client.py
@@ -10,10 +10,12 @@
     C_SUBSCRIPTION_GQL,
     GET_A_GQL,
     GET_A_WITH_FRAGMENT_GQL,
+    GET_S_GQL,
     GET_XYZ_GQL,
 )
 from .get_a import GetA
 from .get_a_with_fragment import GetAWithFragment
+from .get_s import GetS
 from .get_xyz import GetXYZ
 
 
@@ -77,3 +79,11 @@ async def get_xyz(self, **kwargs: Any) -> GetXYZ:
         )
         data = self.get_data(response)
         return GetXYZ.model_validate(data)
+
+    async def get_s(self, **kwargs: Any) -> GetS:
+        variables: Dict[str, object] = {}
+        response = await self.execute(
+            query=GET_S_GQL, operation_name="getS", variables=variables, **kwargs
+        )
+        data = self.get_data(response)
+        return GetS.model_validate(data)
diff --git a/tests/main/clients/operations/expected_client/custom_operations.py b/tests/main/clients/operations/expected_client/custom_operations.py
@@ -4,6 +4,7 @@
     "C_SUBSCRIPTION_GQL",
     "GET_A_GQL",
     "GET_A_WITH_FRAGMENT_GQL",
+    "GET_S_GQL",
     "GET_XYZ_GQL",
 ]
 
@@ -68,3 +69,11 @@
   valueY
 }
 """
+
+GET_S_GQL = """
+query getS {
+  s {
+    id
+  }
+}
+"""
diff --git a/tests/main/clients/operations/expected_client/get_s.py b/tests/main/clients/operations/expected_client/get_s.py
@@ -0,0 +1,9 @@
+from .base_model import BaseModel
+
+
+class GetS(BaseModel):
+    s: "GetSS"
+
+
+class GetSS(BaseModel):
+    id: int
diff --git a/tests/main/clients/operations/pyproject.toml b/tests/main/clients/operations/pyproject.toml
@@ -5,5 +5,8 @@ target_package_name = "client_with_operations"
 include_comments = "none"
 plugins = ["ariadne_codegen.contrib.extract_operations.ExtractOperationsPlugin"]
 
+[tool.ariadne-codegen.scalars.CUSTOMID]
+type = "int"
+
 [tool.ariadne-codegen.extract-operations]
 operations_module_name = "custom_operations"
diff --git a/tests/main/clients/operations/queries.graphql b/tests/main/clients/operations/queries.graphql
@@ -46,3 +46,9 @@ query getXYZ {
     }
   }
 }
+
+query getS {
+  s {
+    id
+  }
+}
diff --git a/tests/main/clients/operations/schema.graphql b/tests/main/clients/operations/schema.graphql
@@ -1,6 +1,9 @@
+scalar CUSTOMID
+
 type Query {
     constQuery: String!
     a: TypeA!
+    s: TypeS!
     xyz: TypeXYZ!
 }
 
@@ -21,6 +24,10 @@ type TypeB {
     value: String!
 }
 
+type TypeS {
+  id: CUSTOMID!
+}
+
 type TypeX {
     valueX: String!
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -46,3 +46,9 @@ query getXYZ { @@
         }
       }
     }
+    query getS {
+      s {
+        id
+      }
+    }