visitor: Remove visitor_keys argument

Replicates graphql/graphql-js@b67ed17

Author: Cito

src/graphql/language/visitor.py

@@ -213,11 +213,7 @@ class Stack(NamedTuple):
     prev: Any  # 'Stack' (python/mypy/issues/731)
 
 
-def visit(
-    root: Node,
-    visitor: Visitor,
-    visitor_keys: Optional[Dict[str, Tuple[str, ...]]] = None,
-) -> Any:
+def visit(root: Node, visitor: Visitor) -> Any:
     """Visit each node in an AST.
 
     :func:`~.visit` will walk through an AST using a depth-first traversal, calling the
@@ -232,16 +228,11 @@ def visit(
     When using :func:`~.visit` to edit an AST, the original AST will not be modified,
     and a new version of the AST with the changes applied will be returned from the
     visit function.
-
-    To customize the node attributes to be used for traversal, you can provide a
-    dictionary visitor_keys mapping node kinds to node attributes.
     """
     if not isinstance(root, Node):
         raise TypeError(f"Not an AST Node: {inspect(root)}.")
     if not isinstance(visitor, Visitor):
         raise TypeError(f"Not an AST Visitor: {inspect(visitor)}.")
-    if visitor_keys is None:
-        visitor_keys = QUERY_DOCUMENT_KEYS
     stack: Any = None
     in_array = isinstance(root, list)
     keys: Tuple[Node, ...] = (root,)
@@ -340,7 +331,7 @@ def visit(
         else:
             stack = Stack(in_array, idx, keys, edits, stack)
             in_array = isinstance(node, list)
-            keys = node if in_array else visitor_keys.get(node.kind, ())
+            keys = node if in_array else QUERY_DOCUMENT_KEYS.get(node.kind, ())
             idx = -1
             edits = []
             if parent:

tests/language/test_visitor.py

@@ -1,6 +1,6 @@
 from copy import copy
 from functools import partial
-from typing import cast, Dict, List, Optional, Tuple
+from typing import cast, List, Optional
 
 from pytest import mark, raises
 
@@ -18,7 +18,6 @@
     ParallelVisitor,
     Visitor,
 )
-from graphql.language.visitor import QUERY_DOCUMENT_KEYS
 from graphql.pyutils import FrozenList
 
 from ..fixtures import kitchen_sink_query  # noqa: F401
@@ -1019,46 +1018,6 @@ def leave(node, *_args):
             ["leave", "document", None],
         ]
 
-    def does_traverse_unknown_node_with_visitor_keys():
-        custom_query_document_keys: Dict[str, Tuple[str, ...]] = {
-            **QUERY_DOCUMENT_KEYS,
-            "custom_field": ("name", "selection_set"),
-        }
-        visited = []
-
-        class TestVisitor(Visitor):
-            @staticmethod
-            def enter(node, *_args):
-                visited.append(["enter", node.kind, get_value(node)])
-
-            @staticmethod
-            def leave(node, *_args):
-                visited.append(["leave", node.kind, get_value(node)])
-
-        visit(custom_ast, TestVisitor(), custom_query_document_keys)
-        assert visited == [
-            ["enter", "document", None],
-            ["enter", "operation_definition", None],
-            ["enter", "selection_set", None],
-            ["enter", "field", None],
-            ["enter", "name", "a"],
-            ["leave", "name", "a"],
-            ["leave", "field", None],
-            ["enter", "custom_field", None],
-            ["enter", "name", "b"],
-            ["leave", "name", "b"],
-            ["enter", "selection_set", None],
-            ["enter", "custom_field", None],
-            ["enter", "name", "c"],
-            ["leave", "name", "c"],
-            ["leave", "custom_field", None],
-            ["leave", "selection_set", None],
-            ["leave", "custom_field", None],
-            ["leave", "selection_set", None],
-            ["leave", "operation_definition", None],
-            ["leave", "document", None],
-        ]
-
     def cannot_define_visitor_with_unknown_ast_nodes():
         with raises(TypeError) as exc_info: