tests

andimarek · andimarek · commit 5acb6f1152d8 · 2018-03-17T21:14:04.000+11:00
diff --git a/src/test/groovy/graphql/language/NodeVisitorStubTest.groovy b/src/test/groovy/graphql/language/NodeVisitorStubTest.groovy
@@ -3,6 +3,9 @@ package graphql.language
 import graphql.util.TraversalControl
 import graphql.util.TraverserContext
 import spock.lang.Specification
+import spock.lang.Unroll
+
+import static java.util.Collections.emptyList
 
 class NodeVisitorStubTest extends Specification {
 
@@ -221,4 +224,39 @@ class NodeVisitorStubTest extends Specification {
         control == TraversalControl.QUIT
     }
 
+    @Unroll
+    def "default for #visitMethod is to call visitNode"() {
+        given:
+        NodeVisitorStub nodeVisitorStub = Spy(NodeVisitorStub, constructorArgs: [])
+        TraverserContext context = Mock(TraverserContext)
+
+        when:
+        def control = nodeVisitorStub."$visitMethod"(node, context)
+        then:
+        1 * nodeVisitorStub.visitNode(node, context) >> TraversalControl.QUIT
+        control == TraversalControl.QUIT
+
+        where:
+        node                           | visitMethod
+        new Argument("", null)         | 'visitArgument'
+        new Directive("", emptyList()) | 'visitDirective'
+        new DirectiveLocation("")      | 'visitDirectiveLocation'
+        new Document()                 | 'visitDocument'
+        new EnumValueDefinition("")    | 'visitEnumValueDefinition'
+        new FieldDefinition("")        | 'visitFieldDefinition'
+        new InputValueDefinition("")   | 'visitInputValueDefinition'
+        new InputValueDefinition("")   | 'visitInputValueDefinition'
+        new ObjectField("", null)      | 'visitObjectField'
+        new OperationTypeDefinition()  | 'visitOperationTypeDefinition'
+        new OperationTypeDefinition()  | 'visitOperationTypeDefinition'
+        new SelectionSet()             | 'visitSelectionSet'
+        new VariableDefinition()       | 'visitVariableDefinition'
+        new StringValue("")            | 'visitValue'
+        new OperationDefinition()      | 'visitDefinition'
+        new UnionTypeDefinition("")    | 'visitTypeDefinition'
+        new Field()                    | 'visitSelection'
+        new NonNullType()              | 'visitType'
+
+    }
+
 }
