diff options
Diffstat (limited to 'lib/gitlab/ci/reports/security/reports.rb')
| -rw-r--r-- | lib/gitlab/ci/reports/security/reports.rb | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/gitlab/ci/reports/security/reports.rb b/lib/gitlab/ci/reports/security/reports.rb index a7a6e5b2593..b7a5e36b108 100644 --- a/lib/gitlab/ci/reports/security/reports.rb +++ b/lib/gitlab/ci/reports/security/reports.rb @@ -22,8 +22,8 @@ module Gitlab reports.values.flat_map(&:findings) end - def violates_default_policy_against?(target_reports, vulnerabilities_allowed) - unsafe_findings_count(target_reports) > vulnerabilities_allowed + def violates_default_policy_against?(target_reports, vulnerabilities_allowed, severity_levels) + unsafe_findings_count(target_reports, severity_levels) > vulnerabilities_allowed end private @@ -32,8 +32,8 @@ module Gitlab findings - target_reports&.findings.to_a end - def unsafe_findings_count(target_reports) - findings_diff(target_reports).count(&:unsafe?) + def unsafe_findings_count(target_reports, severity_levels) + findings_diff(target_reports).count {|finding| finding.unsafe?(severity_levels)} end end end |