Fix mypy issues

Author: Cito

graphql/validation/rules/known_argument_names.py

@@ -1,8 +1,8 @@
-from typing import Dict, List, Union
+from typing import cast, Dict, List, Union
 
 from ...error import GraphQLError
 from ...language import (
-    ArgumentNode, FieldNode, DirectiveDefinitionNode, DirectiveNode, SKIP)
+    ArgumentNode, DirectiveDefinitionNode, DirectiveNode, SKIP)
 from ...pyutils import quoted_or_list, suggestion_list
 from ...type import specified_directives
 from . import ASTValidationRule, SDLValidationContext, ValidationContext
@@ -48,7 +48,7 @@ def __init__(self, context: Union[
         schema = context.schema
         defined_directives = (
             schema.directives if schema else specified_directives)
-        for directive in defined_directives:
+        for directive in cast(List, defined_directives):
             directive_args[directive.name] = list(directive.args)
 
         ast_definitions = context.document.definitions
@@ -100,4 +100,3 @@ def enter_argument(
                 unknown_arg_message(
                     arg_name, field_name, parent_type.name,
                     suggestion_list(arg_name, known_args_names)), arg_node))
-

graphql/validation/rules/provided_required_arguments.py

@@ -1,9 +1,9 @@
-from typing import Dict, Union
+from typing import cast, Dict, List, Union
 
 from ...error import GraphQLError
 from ...language import (
     DirectiveDefinitionNode, DirectiveNode, FieldNode,
-    InputValueDefinitionNode, NonNullTypeNode, print_ast)
+    InputValueDefinitionNode, NonNullTypeNode, TypeNode, print_ast)
 from ...type import (
     GraphQLArgument, is_required_argument, is_type, specified_directives)
 from . import ASTValidationRule, SDLValidationContext, ValidationContext
@@ -38,12 +38,13 @@ class ProvidedRequiredArgumentsOnDirectivesRule(ASTValidationRule):
     def __init__(self, context: Union[
             ValidationContext, SDLValidationContext]) -> None:
         super().__init__(context)
-        required_args_map: Dict[str, Dict[str, GraphQLArgument]] = {}
+        required_args_map: Dict[str, Dict[str, Union[
+            GraphQLArgument, InputValueDefinitionNode]]] = {}
 
         schema = context.schema
         defined_directives = (
             schema.directives if schema else specified_directives)
-        for directive in defined_directives:
+        for directive in cast(List, defined_directives):
             required_args_map[directive.name] = {
                 name: arg for name, arg in directive.args.items()
                 if is_required_argument(arg)}
@@ -72,7 +73,8 @@ def leave_directive(self, directive_node: DirectiveNode, *_args):
                     self.report_error(GraphQLError(
                         missing_directive_arg_message(
                             directive_name, arg_name, str(arg_type)
-                            if is_type(arg_type) else print_ast(arg_type)),
+                            if is_type(arg_type)
+                            else print_ast(cast(TypeNode, arg_type))),
                         [directive_node]))
 
 

tests/validation/test_known_argument_names.py

@@ -153,7 +153,6 @@ def unknown_args_deeply():
             unknown_arg('unknown', 'doesKnowCommand', 'Dog', [], 9, 37)
         ])
 
-
     def describe_within_sdl():
 
         def known_arg_on_directive_inside_sdl():
@@ -180,7 +179,7 @@ def misspelled_arg_name_is_reported_on_directive_defined_inside_sdl():
                 type Query {
                   foo: String @test(agr: "")
                 }
-        
+
                 directive @test(arg: String) on FIELD_DEFINITION
                 """) == [
                 unknown_directive_arg('agr', 'test', ['arg'], 3, 37)]