gitlab-org--gitlab-foss/spec/migrations/20210430134202_copy_adoptio...

45 lines
1.4 KiB
Ruby

# frozen_string_literal: true
#
require 'spec_helper'
require_migration!
RSpec.describe CopyAdoptionSnapshotNamespace, :migration, schema: 20210430124630 do
let(:namespaces_table) { table(:namespaces) }
let(:segments_table) { table(:analytics_devops_adoption_segments) }
let(:snapshots_table) { table(:analytics_devops_adoption_snapshots) }
it 'updates all snapshots without namespace set' do
namespaces_table.create!(id: 123, name: 'group1', path: 'group1')
namespaces_table.create!(id: 124, name: 'group2', path: 'group2')
segments_table.create!(id: 1, namespace_id: 123)
segments_table.create!(id: 2, namespace_id: 124)
create_snapshot(id: 1, segment_id: 1)
create_snapshot(id: 2, segment_id: 2)
create_snapshot(id: 3, segment_id: 2, namespace_id: 123)
migrate!
expect(snapshots_table.find(1).namespace_id).to eq 123
expect(snapshots_table.find(2).namespace_id).to eq 124
expect(snapshots_table.find(3).namespace_id).to eq 123
end
def create_snapshot(**additional_params)
defaults = {
recorded_at: Time.zone.now,
issue_opened: true,
merge_request_opened: true,
merge_request_approved: true,
runner_configured: true,
pipeline_succeeded: true,
deploy_succeeded: true,
end_time: Time.zone.now.end_of_month
}
snapshots_table.create!(defaults.merge(additional_params))
end
end