@@ -106,5 +106,39 @@ class NodeVisitorStubTest extends Specification {
106106 control == TraversalControl . QUIT
107107 }
108108
109+ def " definitions call visitDefinition by default" () {
110+ given :
111+ NodeVisitorStub nodeVisitorStub = Spy (NodeVisitorStub , constructorArgs : [])
112+ OperationDefinition operationDefinition = new OperationDefinition ()
113+ FragmentDefinition fragmentDefinition = new FragmentDefinition ()
114+ DirectiveDefinition directiveDefinition = new DirectiveDefinition (" " )
115+ SchemaDefinition schemaDefinition = new SchemaDefinition ()
116+ TraverserContext context = Mock (TraverserContext )
117+
118+ when :
119+ def control = nodeVisitorStub. visitOperationDefinition(operationDefinition, context)
120+ then :
121+ 1 * nodeVisitorStub. visitDefinition(operationDefinition, context) >> TraversalControl . QUIT
122+ control == TraversalControl . QUIT
123+
124+ when :
125+ control = nodeVisitorStub. visitFragmentDefinition(fragmentDefinition, context)
126+ then :
127+ 1 * nodeVisitorStub. visitDefinition(fragmentDefinition, context) >> TraversalControl . QUIT
128+ control == TraversalControl . QUIT
129+
130+ when :
131+ control = nodeVisitorStub. visitDirectiveDefinition(directiveDefinition, context)
132+ then :
133+ 1 * nodeVisitorStub. visitDefinition(directiveDefinition, context) >> TraversalControl . QUIT
134+ control == TraversalControl . QUIT
135+
136+ when :
137+ control = nodeVisitorStub. visitSchemaDefinition(schemaDefinition, context)
138+ then :
139+ 1 * nodeVisitorStub. visitDefinition(schemaDefinition, context) >> TraversalControl . QUIT
140+ control == TraversalControl . QUIT
141+ }
142+
109143
110144}
0 commit comments