Welcome to mirror list, hosted at ThFree Co, Russian Federation.

fix_batched_migrations_old_format_job_arguments_spec.rb « migrations « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: e15011d0dab284b7d2f16e268e766f21bc0fa31e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
# frozen_string_literal: true

require 'spec_helper'
require_migration!

# rubocop:disable Style/WordArray
RSpec.describe FixBatchedMigrationsOldFormatJobArguments do
  let(:batched_background_migrations) { table(:batched_background_migrations) }

  context 'when migrations with legacy job arguments exists' do
    it 'updates job arguments to current format' do
      legacy_events_migration = create_batched_migration('events', 'id', ['id', 'id_convert_to_bigint'])
      legacy_push_event_payloads_migration = create_batched_migration('push_event_payloads', 'event_id', ['event_id', 'event_id_convert_to_bigint'])

      migrate!

      expect(legacy_events_migration.reload.job_arguments).to eq([['id'], ['id_convert_to_bigint']])
      expect(legacy_push_event_payloads_migration.reload.job_arguments).to eq([['event_id'], ['event_id_convert_to_bigint']])
    end
  end

  context 'when only migrations with current job arguments exists' do
    it 'updates nothing' do
      events_migration = create_batched_migration('events', 'id', [['id'], ['id_convert_to_bigint']])
      push_event_payloads_migration = create_batched_migration('push_event_payloads', 'event_id', [['event_id'], ['event_id_convert_to_bigint']])

      migrate!

      expect(events_migration.reload.job_arguments).to eq([['id'], ['id_convert_to_bigint']])
      expect(push_event_payloads_migration.reload.job_arguments).to eq([['event_id'], ['event_id_convert_to_bigint']])
    end
  end

  context 'when migrations with both legacy and current job arguments exist' do
    it 'updates nothing' do
      legacy_events_migration = create_batched_migration('events', 'id', ['id', 'id_convert_to_bigint'])
      events_migration = create_batched_migration('events', 'id', [['id'], ['id_convert_to_bigint']])
      legacy_push_event_payloads_migration = create_batched_migration('push_event_payloads', 'event_id', ['event_id', 'event_id_convert_to_bigint'])
      push_event_payloads_migration = create_batched_migration('push_event_payloads', 'event_id', [['event_id'], ['event_id_convert_to_bigint']])

      migrate!

      expect(legacy_events_migration.reload.job_arguments).to eq(['id', 'id_convert_to_bigint'])
      expect(events_migration.reload.job_arguments).to eq([['id'], ['id_convert_to_bigint']])
      expect(legacy_push_event_payloads_migration.reload.job_arguments).to eq(['event_id', 'event_id_convert_to_bigint'])
      expect(push_event_payloads_migration.reload.job_arguments).to eq([['event_id'], ['event_id_convert_to_bigint']])
    end
  end

  def create_batched_migration(table_name, column_name, job_arguments)
    batched_background_migrations.create!(
      max_value: 10,
      batch_size: 10,
      sub_batch_size: 10,
      interval: 1,
      job_class_name: 'CopyColumnUsingBackgroundMigrationJob',
      table_name: table_name,
      column_name: column_name,
      job_arguments: job_arguments
    )
  end
end
# rubocop:enable Style/WordArray