1 files changed, 57 insertions, 0 deletions
diff --git a/spec/lib/gitlab/background_migration/backfill_prepared_at_merge_requests_spec.rb b/spec/lib/gitlab/background_migration/backfill_prepared_at_merge_requests_spec.rb
new file mode 100644
index 00000000000..b33a1a31c40
--- /dev/null
+++ b/spec/lib/gitlab/background_migration/backfill_prepared_at_merge_requests_spec.rb
@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::BackgroundMigration::BackfillPreparedAtMergeRequests, :migration,
+  feature_category: :code_review_workflow, schema: 20230202135758 do
+  let(:namespaces) { table(:namespaces) }
+  let(:projects) { table(:projects) }
+  let(:mr_table) { table(:merge_requests) }
+
+  let(:namespace) { namespaces.create!(name: 'batchtest1', type: 'Group', path: 'space1') }
+  let(:proj_namespace) { namespaces.create!(name: 'proj1', path: 'proj1', type: 'Project', parent_id: namespace.id) }
+  let(:project) do
+    projects.create!(name: 'proj1', path: 'proj1', namespace_id: namespace.id, project_namespace_id: proj_namespace.id)
+  end
+
+  let(:test_worker) do
+    described_class.new(
+      start_id: 1,
+      end_id: 100,
+      batch_table: :merge_requests,
+      batch_column: :id,
+      sub_batch_size: 10,
+      pause_ms: 0,
+      connection: ApplicationRecord.connection
+    )
+  end
+
+  it 'updates merge requests with prepared_at nil' do
+    time = Time.current
+
+    mr_1 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
+      prepared_at: nil, merge_status: 'checking')
+    mr_2 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
+      prepared_at: nil, merge_status: 'preparing')
+    mr_3 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
+      prepared_at: time)
+    mr_4 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
+      prepared_at: time, merge_status: 'checking')
+    mr_5 = mr_table.create!(target_project_id: project.id, source_branch: 'master', target_branch: 'feature',
+      prepared_at: time, merge_status: 'preparing')
+
+    expect(mr_1.prepared_at).to be_nil
+    expect(mr_2.prepared_at).to be_nil
+    expect(mr_3.prepared_at.to_i).to eq(time.to_i)
+    expect(mr_4.prepared_at.to_i).to eq(time.to_i)
+    expect(mr_5.prepared_at.to_i).to eq(time.to_i)
+
+    test_worker.perform
+
+    expect(mr_1.reload.prepared_at.to_i).to eq(mr_1.created_at.to_i)
+    expect(mr_2.reload.prepared_at).to be_nil
+    expect(mr_3.reload.prepared_at.to_i).to eq(time.to_i)
+    expect(mr_4.reload.prepared_at.to_i).to eq(time.to_i)
+    expect(mr_5.reload.prepared_at.to_i).to eq(time.to_i)
+  end
+end