Provides a correct OperationType without name in GraphQLPrinter

Related GraphQL-js commit: graphql/graphql-js@6741c31

Author: syrusakbary

graphql/core/language/printer.py

@@ -24,14 +24,14 @@ def leave_Document(self, node, *args):
     def leave_OperationDefinition(self, node, *args):
         name = node.name
         selection_set = node.selection_set
-        if not name:
-            return selection_set
-
         op = node.operation
-        defs = wrap('(', join(node.variable_definitions, ', '), ')')
+        var_defs = wrap('(', join(node.variable_definitions, ', '), ')')
         directives = join(node.directives, ' ')
 
-        return join([op, join([name, defs]), directives, selection_set], ' ')
+        if not name and not directives and not var_defs and op == 'query':
+            return selection_set
+
+        return join([op, join([name, var_defs]), directives, selection_set], ' ')
 
     def leave_VariableDefinition(self, node, *args):
         return node.variable + ': ' + node.type + wrap(' = ', node.default_value)

tests/core_language/test_printer.py

@@ -28,6 +28,46 @@ def test_produces_helpful_error_messages():
     assert 'Invalid AST Node' in str(excinfo.value)
 
 
+def test_correctly_prints_query_operation_without_name():
+    query_ast_shorthanded = parse('query { id, name }')
+    assert print_ast(query_ast_shorthanded) == '''{
+  id
+  name
+}
+'''
+
+
+def test_correctly_prints_mutation_operation_without_name():
+    mutation_ast = parse('mutation { id, name }')
+    assert print_ast(mutation_ast) == '''mutation {
+  id
+  name
+}
+'''
+
+
+def test_correctly_prints_query_with_artifacts():
+    query_ast_shorthanded = parse(
+      'query ($foo: TestType) @testDirective { id, name }'
+    )
+    assert print_ast(query_ast_shorthanded) == '''query ($foo: TestType) @testDirective {
+  id
+  name
+}
+'''
+
+
+def test_correctly_prints_mutation_with_artifacts():
+    query_ast_shorthanded = parse(
+      'mutation ($foo: TestType) @testDirective { id, name }'
+    )
+    assert print_ast(query_ast_shorthanded) == '''mutation ($foo: TestType) @testDirective {
+  id
+  name
+}
+'''
+
+
 def test_prints_kitchen_sink():
     ast = parse(KITCHEN_SINK)
     printed = print_ast(ast)