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

20230613192703_swap_ci_build_needs_to_big_int_for_self_hosts_spec.rb « migrations « spec - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 3db5d3b3c1612f1f1cf5b0bc311a792b976726d5 (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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
# frozen_string_literal: true

require 'spec_helper'
require_migration!

RSpec.describe SwapCiBuildNeedsToBigIntForSelfHosts, feature_category: :continuous_integration do
  after do
    connection = described_class.new.connection
    connection.execute('ALTER TABLE ci_build_needs DROP COLUMN IF EXISTS id_convert_to_bigint')
  end

  describe '#up' do
    context 'when on GitLab.com, dev, or test' do
      before do
        connection = described_class.new.connection
        connection.execute('ALTER TABLE ci_build_needs ALTER COLUMN id TYPE bigint')
        connection.execute('ALTER TABLE ci_build_needs DROP COLUMN IF EXISTS id_convert_to_bigint')
      end

      it 'does not swap the columns' do
        # rubocop: disable RSpec/AnyInstanceOf
        allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(true)
        # rubocop: enable RSpec/AnyInstanceOf

        ci_build_needs = table(:ci_build_needs)

        disable_migrations_output do
          reversible_migration do |migration|
            migration.before -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
              expect(ci_build_needs.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
            }

            migration.after -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
              expect(ci_build_needs.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
            }
          end
        end
      end
    end

    context 'when a self-hosted installation has already completed the swap' do
      before do
        connection = described_class.new.connection
        connection.execute('ALTER TABLE ci_build_needs ALTER COLUMN id TYPE bigint')
        connection.execute('ALTER TABLE ci_build_needs ADD COLUMN IF NOT EXISTS id_convert_to_bigint integer')
      end

      it 'does not swap the columns' do
        # rubocop: disable RSpec/AnyInstanceOf
        allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(false)
        # rubocop: enable RSpec/AnyInstanceOf

        ci_build_needs = table(:ci_build_needs)

        migrate!

        expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
        expect(ci_build_needs.columns.find do |c|
          c.name == 'id_convert_to_bigint'
        end.sql_type).to eq('integer')
      end
    end

    context 'when a self-hosted installation has the `id_convert_to_bigint` column already dropped' do
      before do
        connection = described_class.new.connection
        connection.execute('ALTER TABLE ci_build_needs ALTER COLUMN id TYPE bigint')
        connection.execute('ALTER TABLE ci_build_needs DROP COLUMN IF EXISTS id_convert_to_bigint')
      end

      it 'does not swap the columns' do
        # rubocop: disable RSpec/AnyInstanceOf
        allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(false)
        # rubocop: enable RSpec/AnyInstanceOf

        ci_build_needs = table(:ci_build_needs)

        disable_migrations_output do
          reversible_migration do |migration|
            migration.before -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
              expect(ci_build_needs.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
            }

            migration.after -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
              expect(ci_build_needs.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
            }
          end
        end
      end
    end

    context 'when an installation is self-hosted' do
      before do
        connection = described_class.new.connection
        connection.execute('ALTER TABLE ci_build_needs ALTER COLUMN id TYPE integer')
        connection.execute('ALTER TABLE ci_build_needs ADD COLUMN IF NOT EXISTS id_convert_to_bigint bigint')
        connection.execute('ALTER TABLE ci_build_needs ALTER COLUMN id_convert_to_bigint TYPE bigint')
        connection.execute('DROP INDEX IF EXISTS index_ci_build_needs_on_id_convert_to_bigint')
        connection.execute('CREATE OR REPLACE FUNCTION trigger_3207b8d0d6f3() RETURNS trigger LANGUAGE plpgsql AS $$
          BEGIN NEW."id_convert_to_bigint" := NEW."id"; RETURN NEW; END; $$;')
      end

      it 'swaps the columns' do
        # rubocop: disable RSpec/AnyInstanceOf
        allow_any_instance_of(described_class).to receive(:com_or_dev_or_test_but_not_jh?).and_return(false)
        # rubocop: enable RSpec/AnyInstanceOf

        ci_build_needs = table(:ci_build_needs)

        disable_migrations_output do
          reversible_migration do |migration|
            migration.before -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('integer')
              expect(ci_build_needs.columns.find do |c|
                       c.name == 'id_convert_to_bigint'
                     end.sql_type).to eq('bigint')
            }

            migration.after -> {
              ci_build_needs.reset_column_information

              expect(ci_build_needs.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
              expect(ci_build_needs.columns.find do |c|
                       c.name == 'id_convert_to_bigint'
                     end.sql_type).to eq('integer')
            }
          end
        end
      end
    end
  end
end