blob: 79583fb0a21465a6f25414fac1b467e1480605ef (
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
|
# frozen_string_literal: true
require 'spec_helper'
require_migration!
RSpec.describe SwapSuggestionsNoteIdToBigintForSelfHosts, feature_category: :database do
describe '#up' do
context 'when 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 suggestions ALTER COLUMN note_id TYPE bigint')
connection.execute('ALTER TABLE suggestions DROP COLUMN IF EXISTS note_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
suggestions = table(:suggestions)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }).to be nil
}
migration.after -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }).to be nil
}
end
end
end
end
context 'when self-managed instance with the columns already swapped' 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 suggestions ALTER COLUMN note_id TYPE bigint')
connection.execute('ALTER TABLE suggestions ADD COLUMN IF NOT EXISTS note_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
suggestions = table(:suggestions)
migrate!
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find do |c|
c.name == 'note_id_convert_to_bigint'
end.sql_type).to eq('integer')
end
end
context 'when self-managed instance with the `note_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 suggestions ALTER COLUMN note_id TYPE bigint')
connection.execute('ALTER TABLE suggestions DROP COLUMN IF EXISTS note_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
suggestions = table(:suggestions)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }).to be nil
}
migration.after -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find { |c| c.name == 'note_id_convert_to_bigint' }).to be nil
}
end
end
end
end
context 'when self-managed instance' 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 suggestions ALTER COLUMN note_id TYPE integer')
connection.execute('ALTER TABLE suggestions ADD COLUMN IF NOT EXISTS note_id_convert_to_bigint bigint')
connection.execute('ALTER TABLE suggestions ALTER COLUMN note_id_convert_to_bigint TYPE bigint')
connection.execute('DROP INDEX IF EXISTS index_suggestions_on_note_id_convert_to_bigint')
connection.execute('CREATE OR REPLACE FUNCTION trigger_ee7956d805e6() RETURNS trigger LANGUAGE plpgsql AS $$
BEGIN NEW."note_id_convert_to_bigint" := NEW."note_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
suggestions = table(:suggestions)
disable_migrations_output do
reversible_migration do |migration|
migration.before -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('integer')
expect(suggestions.columns.find do |c|
c.name == 'note_id_convert_to_bigint'
end.sql_type).to eq('bigint')
}
migration.after -> {
suggestions.reset_column_information
expect(suggestions.columns.find { |c| c.name == 'note_id' }.sql_type).to eq('bigint')
expect(suggestions.columns.find do |c|
c.name == 'note_id_convert_to_bigint'
end.sql_type).to eq('integer')
}
end
end
end
end
end
end
|