Allow tests to ignore recursion

1 files changed, 5 insertions, 1 deletions

diff --git a/lib/gitlab/graphql/query_analyzers/recursion_analyzer.rb b/lib/gitlab/graphql/query_analyzers/recursion_analyzer.rb
index 70d4672d079..ccf9e597307 100644
--- a/lib/gitlab/graphql/query_analyzers/recursion_analyzer.rb
+++ b/lib/gitlab/graphql/query_analyzers/recursion_analyzer.rb
@@ -45,13 +45,17 @@ module Gitlab
         def recursion_too_deep?(node_name, times_encountered)
           return if IGNORED_FIELDS.include?(node_name)
 
-          times_encountered > RECURSION_THRESHOLD
+          times_encountered > recursion_threshold
         end
 
         def skip_node?(irep_node)
           ast_node = irep_node.ast_node
           !ast_node.is_a?(GraphQL::Language::Nodes::Field) || ast_node.selections.empty?
         end
+
+        def recursion_threshold
+          RECURSION_THRESHOLD
+        end
       end
     end
   end