Fixed Test-Errors

Author: normanre

ariadne_codegen/client_generators/custom_arguments.py

@@ -1,16 +1,15 @@
 import ast
-import enum
 from typing import Any, Dict, List, Optional, Tuple, Union, cast
 
 from graphql import (
     GraphQLEnumType,
     GraphQLInputObjectType,
     GraphQLInterfaceType,
+    GraphQLList,
     GraphQLNonNull,
     GraphQLObjectType,
     GraphQLScalarType,
     GraphQLUnionType,
-    GraphQLList,
 )
 
 from ..codegen import (
@@ -25,10 +24,10 @@
     generate_dict,
     generate_import_from,
     generate_keyword,
+    generate_list_annotation,
     generate_name,
     generate_subscript,
     generate_tuple,
-    generate_list_annotation,
 )
 from ..exceptions import ParsingError
 from ..plugins.manager import PluginManager
@@ -38,6 +37,8 @@
     BASE_MODEL_FILE_PATH,
     DICT,
     INPUT_SCALARS_MAP,
+    LIST,
+    TYPING_MODULE,
     UPLOAD_CLASS_NAME,
 )
 from .custom_generator_utils import get_final_type
@@ -128,7 +129,13 @@ def _accumulate_return_arguments(
         return_arguments_values: List[ast.expr],
         arg_name: str,
         name: str,
-        complete_type: Union[GraphQLObjectType, GraphQLInterfaceType, GraphQLUnionType, GraphQLNonNull, GraphQLList],
+        complete_type: Union[
+            GraphQLObjectType,
+            GraphQLInterfaceType,
+            GraphQLUnionType,
+            GraphQLNonNull,
+            GraphQLList,
+        ],
         used_custom_scalar: Optional[str],
     ) -> None:
         """Accumulates return arguments."""
@@ -148,7 +155,13 @@ def _accumulate_return_arguments(
 
     def _generate_complete_type_name(
         self,
-        complete_type: Union[GraphQLObjectType, GraphQLInterfaceType, GraphQLUnionType, GraphQLNonNull, GraphQLList]
+        complete_type: Union[
+            GraphQLObjectType,
+            GraphQLInterfaceType,
+            GraphQLUnionType,
+            GraphQLNonNull,
+            GraphQLList,
+        ],
     ) -> str:
         if isinstance(complete_type, GraphQLNonNull):
             if hasattr(complete_type, "of_type"):
@@ -225,8 +238,18 @@ def _parse_graphql_type_name(
                 self._used_custom_scalars.append(used_custom_scalar)
         else:
             raise ParsingError(f"Incorrect argument type {name}")
-        return generate_annotation_name(name, nullable) if not is_list else generate_list_annotation(
-            generate_annotation_name(name, nullable=False), nullable), used_custom_scalar
+
+        if is_list:
+            self._add_import(generate_import_from(names=[LIST], from_=TYPING_MODULE))
+
+        return (
+            generate_annotation_name(name, nullable)
+            if not is_list
+            else generate_list_annotation(
+                generate_annotation_name(name, nullable=False), nullable
+            ),
+            used_custom_scalar,
+        )
 
     def add_custom_scalar_imports(self) -> None:
         """Adds imports for custom scalars used in the schema."""

ariadne_codegen/client_generators/custom_fields.py

@@ -42,7 +42,6 @@
     OPTIONAL,
     TYPING_MODULE,
     UNION,
-    LIST,
 )
 from .custom_generator_utils import TypeCollector, get_final_type
 from .scalars import ScalarData
@@ -71,7 +70,7 @@ def __init__(
         ]
         self._add_import(
             generate_import_from(
-                [OPTIONAL, UNION, ANY, DICT, LIST],
+                [OPTIONAL, UNION, ANY, DICT],
                 TYPING_MODULE,
             )
         )

ariadne_codegen/client_generators/custom_operation.py

@@ -35,7 +35,6 @@
     GRAPHQL_UNION_SUFFIX,
     OPTIONAL,
     TYPING_MODULE,
-    LIST,
 )
 from .custom_generator_utils import get_final_type
 from .scalars import ScalarData
@@ -68,7 +67,7 @@ def __init__(
 
         self._imports: List[ast.ImportFrom] = []
         self._type_imports: List[ast.ImportFrom] = []
-        self._add_import(generate_import_from([OPTIONAL, ANY, DICT, LIST], TYPING_MODULE))
+        self._add_import(generate_import_from([OPTIONAL, ANY, DICT], TYPING_MODULE))
         self.argument_generator = ArgumentGenerator(
             self.custom_scalars,
             self.convert_to_snake_case,