From ee0d789365b089baa72a2fcdcd237014a58b3d2b Mon Sep 17 00:00:00 2001 From: marc_shaw Date: Wed, 17 Dec 2025 13:09:24 +0100 Subject: [PATCH] Change shared example to show extra queries --- ee/spec/graphql/types/vulnerability_type_spec.rb | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ee/spec/graphql/types/vulnerability_type_spec.rb b/ee/spec/graphql/types/vulnerability_type_spec.rb index 75992ecbf80f13..2d9bd9c2dd29f3 100644 --- a/ee/spec/graphql/types/vulnerability_type_spec.rb +++ b/ee/spec/graphql/types/vulnerability_type_spec.rb @@ -91,8 +91,8 @@ stub_env('GITALY_DISABLE_REQUEST_LIMITS', 'true') GitlabSchema.execute(query, context: { current_user: user }) - control_count = ActiveRecord::QueryRecorder.new { GitlabSchema.execute(query, context: { current_user: user }) }.count - expect(control_count <= single_query_count).to be_truthy + control = ActiveRecord::QueryRecorder.new { GitlabSchema.execute(query, context: { current_user: user }) } + expect(control.count <= single_query_count).to be_truthy create_three_vulnerabilities = create_list(:vulnerability, 3, :with_finding, :with_state_transition, :with_remediation, project: project) create_three_vulnerabilities.each do |v| @@ -104,7 +104,8 @@ create(:policy_dismissal, :preserved, project: project, security_findings_uuids: [v.finding.uuid]) end - expect { GitlabSchema.execute(query, context: { current_user: user }) }.not_to exceed_query_limit(single_query_count) + expect { GitlabSchema.execute(query, context: { current_user: user }) }.not_to exceed_query_limit(control) + .with_threshold(single_query_count - control.count) end end -- GitLab