Add input type

example/mutations/__init__.py

@@ -1,9 +1,13 @@
 from typing import Any
 
-from . import compare_roles
-from . import dates_delta
+from . import (
+    calc,
+    compare_roles,
+    dates_delta,
+)
 
 mutations: list[Any] = [
+    calc.Mutation,
     compare_roles.Mutation,
     dates_delta.Mutation,
 ]

example/mutations/calc.py

@@ -0,0 +1,32 @@
+from enum import StrEnum
+
+from ariadne_graphql_modules import GraphQLInput, GraphQLObject
+from graphql import GraphQLResolveInfo
+
+from ..scalars.date import DateScalar
+
+
+class CalcOperation(StrEnum):
+    ADD = "add"
+    SUB = "sub"
+    MUL = "MUL"
+
+
+class CalcInput(GraphQLInput):
+    a: int
+    b: int
+    op: CalcOperation
+
+
+class Mutation(GraphQLObject):
+    @GraphQLObject.field(name="calc")
+    @staticmethod
+    def resolve_calc(obj, info: GraphQLResolveInfo, *, input: CalcInput) -> int:
+        if input.op == CalcOperation.ADD:
+            return input.a + input.b
+        if input.op == CalcOperation.SUB:
+            return input.a - input.b
+        if input.op == CalcOperation.MUL:
+            return input.a * input.b
+
+        return 0

tests/test_mutation.py

@@ -1,6 +1,18 @@
 import pytest
 
 
+@pytest.mark.asyncio
+async def test_query_calc_mutation(exec_query):
+    result = await exec_query(
+        """
+        mutation {
+            calc(input: {a: 4, b: 3, op: MUL})
+        }
+        """
+    )
+    assert result.data == {"calc": 12}
+
+
 @pytest.mark.asyncio
 async def test_query_compare_roles_mutation(exec_query):
     result = await exec_query(