1 files changed, 18 insertions, 17 deletions
diff --git a/spec/migrations/schedule_recalculate_vulnerability_finding_signatures_for_findings_spec.rb b/spec/migrations/schedule_recalculate_vulnerability_finding_signatures_for_findings_spec.rb
index 9b62dd79e08..8903a32285e 100644
--- a/spec/migrations/schedule_recalculate_vulnerability_finding_signatures_for_findings_spec.rb
+++ b/spec/migrations/schedule_recalculate_vulnerability_finding_signatures_for_findings_spec.rb
@@ -3,7 +3,8 @@
 require 'spec_helper'
 require_migration!
 
-RSpec.describe ScheduleRecalculateVulnerabilityFindingSignaturesForFindings, :migration do
+RSpec.describe ScheduleRecalculateVulnerabilityFindingSignaturesForFindings, :migration,
+feature_category: :vulnerability_management do
   before do
     allow(Gitlab).to receive(:ee?).and_return(ee?)
     stub_const("#{described_class.name}::BATCH_SIZE", 2)
@@ -20,21 +21,21 @@ RSpec.describe ScheduleRecalculateVulnerabilityFindingSignaturesForFindings, :mi
   context 'when the Gitlab instance is EE' do
     let(:ee?) { true }
 
-    let_it_be(:namespaces) { table(:namespaces) }
-    let_it_be(:projects) { table(:projects) }
-    let_it_be(:findings) { table(:vulnerability_occurrences) }
-    let_it_be(:scanners) { table(:vulnerability_scanners) }
-    let_it_be(:identifiers) { table(:vulnerability_identifiers) }
-    let_it_be(:vulnerability_finding_signatures) { table(:vulnerability_finding_signatures) }
+    let!(:namespaces) { table(:namespaces) }
+    let!(:projects) { table(:projects) }
+    let!(:findings) { table(:vulnerability_occurrences) }
+    let!(:scanners) { table(:vulnerability_scanners) }
+    let!(:identifiers) { table(:vulnerability_identifiers) }
+    let!(:vulnerability_finding_signatures) { table(:vulnerability_finding_signatures) }
 
-    let_it_be(:namespace) { namespaces.create!(name: 'test', path: 'test') }
-    let_it_be(:project) { projects.create!(namespace_id: namespace.id, name: 'gitlab', path: 'gitlab') }
+    let!(:namespace) { namespaces.create!(name: 'test', path: 'test') }
+    let!(:project) { projects.create!(namespace_id: namespace.id, name: 'gitlab', path: 'gitlab') }
 
-    let_it_be(:scanner) do
+    let!(:scanner) do
       scanners.create!(project_id: project.id, external_id: 'trivy', name: 'Security Scanner')
     end
 
-    let_it_be(:identifier) do
+    let!(:identifier) do
       identifiers.create!(project_id: project.id,
                           fingerprint: 'd432c2ad2953e8bd587a3a43b3ce309b5b0154c123',
                           external_type: 'SECURITY_ID',
@@ -42,14 +43,14 @@ RSpec.describe ScheduleRecalculateVulnerabilityFindingSignaturesForFindings, :mi
                           name: 'SECURITY_IDENTIFIER 0')
     end
 
-    let_it_be(:finding1) { findings.create!(finding_params) }
-    let_it_be(:signature1) { vulnerability_finding_signatures.create!(finding_id: finding1.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
+    let!(:finding1) { findings.create!(finding_params) }
+    let!(:signature1) { vulnerability_finding_signatures.create!(finding_id: finding1.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
 
-    let_it_be(:finding2) { findings.create!(finding_params) }
-    let_it_be(:signature2) { vulnerability_finding_signatures.create!(finding_id: finding2.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
+    let!(:finding2) { findings.create!(finding_params) }
+    let!(:signature2) { vulnerability_finding_signatures.create!(finding_id: finding2.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
 
-    let_it_be(:finding3) { findings.create!(finding_params) }
-    let_it_be(:signature3) { vulnerability_finding_signatures.create!(finding_id: finding3.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
+    let!(:finding3) { findings.create!(finding_params) }
+    let!(:signature3) { vulnerability_finding_signatures.create!(finding_id: finding3.id, algorithm_type: 0, signature_sha: ::Digest::SHA1.digest(SecureRandom.hex(50))) }
 
     # this migration is now a no-op
     it 'does not schedule the background jobs', :aggregate_failure do