blob: a2d9887a6c0c7b1947695601163de4e3106368bf (
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
147
148
149
150
151
152
153
154
155
|
# frozen_string_literal: true
require 'spec_helper'
require_migration!
RSpec.describe SwapMergeRequestMetricsIdToBigintForSelfHosts, feature_category: :database do
after do
connection = described_class.new.connection
connection.execute('ALTER TABLE merge_request_metrics DROP COLUMN IF EXISTS id_convert_to_bigint')
end
describe '#up' do
context 'when is GitLab.com, dev, or test' do
before do
# As we call `schema_migrate_down!` before each example, and for this migration
# `#down` is same as `#up`, we need to ensure we start from the expected state.
connection = described_class.new.connection
connection.execute('ALTER TABLE merge_request_metrics ALTER COLUMN id TYPE bigint')
connection.execute('ALTER TABLE merge_request_metrics 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
merge_request_metrics = table(:merge_request_metrics)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
}
migration.after -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
}
end
end
end
end
context 'when is a self-host customer with the swapped already completed' do
before do
# As we call `schema_migrate_down!` before each example, and for this migration
# `#down` is same as `#up`, we need to ensure we start from the expected state.
connection = described_class.new.connection
connection.execute('ALTER TABLE merge_request_metrics ALTER COLUMN id TYPE bigint')
connection.execute('ALTER TABLE merge_request_metrics 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
merge_request_metrics = table(:merge_request_metrics)
migrate!
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find do |c|
c.name == 'id_convert_to_bigint'
end.sql_type).to eq('integer')
end
end
context 'when is a self-host customer with the `id_convert_to_bigint` column already dropped ' do
before do
# As we call `schema_migrate_down!` before each example, and for this migration
# `#down` is same as `#up`, we need to ensure we start from the expected state.
connection = described_class.new.connection
connection.execute('ALTER TABLE merge_request_metrics ALTER COLUMN id TYPE bigint')
connection.execute('ALTER TABLE merge_request_metrics 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
merge_request_metrics = table(:merge_request_metrics)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
}
migration.after -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find { |c| c.name == 'id_convert_to_bigint' }).to be nil
}
end
end
end
end
context 'when is a self-host customer' do
before do
# As we call `schema_migrate_down!` before each example, and for this migration
# `#down` is same as `#up`, we need to ensure we start from the expected state.
connection = described_class.new.connection
connection.execute('ALTER TABLE merge_request_metrics ALTER COLUMN id TYPE integer')
connection.execute('ALTER TABLE merge_request_metrics ADD COLUMN IF NOT EXISTS id_convert_to_bigint bigint')
connection.execute('ALTER TABLE merge_request_metrics ALTER COLUMN id_convert_to_bigint TYPE bigint')
connection.execute('DROP INDEX IF EXISTS index_merge_request_metrics_on_id_convert_to_bigint')
connection.execute('DROP INDEX IF EXISTS tmp_index_mr_metrics_on_target_project_id_merged_at_nulls_last')
connection.execute('CREATE OR REPLACE FUNCTION trigger_c7107f30d69d() 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
merge_request_metrics = table(:merge_request_metrics)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('integer')
expect(merge_request_metrics.columns.find do |c|
c.name == 'id_convert_to_bigint'
end.sql_type).to eq('bigint')
}
migration.after -> {
merge_request_metrics.reset_column_information
expect(merge_request_metrics.columns.find { |c| c.name == 'id' }.sql_type).to eq('bigint')
expect(merge_request_metrics.columns.find do |c|
c.name == 'id_convert_to_bigint'
end.sql_type).to eq('integer')
}
end
end
end
end
end
end
|