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
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.shared_examples 'validate schema data' do |tables_and_views|
it 'all tables and views have assigned a known gitlab_schema' do
expect(tables_and_views).to all(
match([be_a(String), be_in(Gitlab::Database.schemas_to_base_models.keys.map(&:to_sym))])
)
end
end
RSpec.describe Gitlab::Database::GitlabSchema, feature_category: :database do
shared_examples 'maps table name to table schema' do
using RSpec::Parameterized::TableSyntax
where(:name, :classification) do
'ci_builds' | :gitlab_ci
'my_schema.ci_builds' | :gitlab_ci
'my_schema.ci_runner_machine_builds_100' | :gitlab_ci
'my_schema._test_gitlab_main_table' | :gitlab_main
'information_schema.columns' | :gitlab_internal
'audit_events_part_5fc467ac26' | :gitlab_main
'_test_gitlab_main_table' | :gitlab_main
'_test_gitlab_ci_table' | :gitlab_ci
'_test_gitlab_main_clusterwide_table' | :gitlab_main_clusterwide
'_test_gitlab_main_cell_table' | :gitlab_main_cell
'_test_gitlab_pm_table' | :gitlab_pm
'_test_my_table' | :gitlab_shared
'pg_attribute' | :gitlab_internal
end
with_them do
it { is_expected.to eq(classification) }
end
end
describe '.deleted_views_and_tables_to_schema' do
include_examples 'validate schema data', described_class.deleted_views_and_tables_to_schema
end
describe '.views_and_tables_to_schema' do
include_examples 'validate schema data', described_class.views_and_tables_to_schema
# group configurations by db_docs_dir, since then we expect all sharing this
# to contain exactly those tables
Gitlab::Database.all_database_connections.values.group_by(&:db_docs_dir).each do |db_docs_dir, db_infos|
context "for #{db_docs_dir}" do
let(:all_gitlab_schemas) { db_infos.flat_map(&:gitlab_schemas).to_set }
let(:tables_for_gitlab_schemas) do
described_class.views_and_tables_to_schema.select do |_, gitlab_schema|
all_gitlab_schemas.include?(gitlab_schema)
end
end
db_infos.to_h { |db_info| [db_info.name, db_info.connection_class] }
.compact.each do |db_config_name, connection_class|
context "validates '#{db_config_name}' using '#{connection_class}'" do
let(:data_sources) { connection_class.connection.data_sources }
it 'new data sources are added' do
missing_data_sources = data_sources.to_set - tables_for_gitlab_schemas.keys
expect(missing_data_sources).to be_empty, \
"Missing table/view(s) #{missing_data_sources.to_a} not found in " \
"#{described_class}.views_and_tables_to_schema. " \
"Any new tables or views must be added to the database dictionary. " \
"More info: https://docs.gitlab.com/ee/development/database/database_dictionary.html"
end
it 'non-existing data sources are removed' do
extra_data_sources = tables_for_gitlab_schemas.keys.to_set - data_sources
expect(extra_data_sources).to be_empty, \
"Extra table/view(s) #{extra_data_sources.to_a} found in " \
"#{described_class}.views_and_tables_to_schema. " \
"Any removed or renamed tables or views must be removed from the database dictionary. " \
"More info: https://docs.gitlab.com/ee/development/database/database_dictionary.html"
end
end
end
end
end
end
describe '.tables_to_schema' do
let(:database_models) { Gitlab::Database.database_base_models.except(:geo) }
let(:views) { database_models.flat_map { |_, m| m.connection.views }.sort.uniq }
subject { described_class.tables_to_schema }
it 'returns only tables' do
tables = subject.keys
expect(tables).not_to include(views.to_set)
end
end
describe '.views_to_schema' do
let(:database_models) { Gitlab::Database.database_base_models.except(:geo) }
let(:tables) { database_models.flat_map { |_, m| m.connection.tables }.sort.uniq }
subject { described_class.views_to_schema }
it 'returns only views' do
views = subject.keys
expect(views).not_to include(tables.to_set)
end
end
describe '.table_schemas!' do
let(:tables) { %w[projects issues ci_builds] }
subject { described_class.table_schemas!(tables) }
it 'returns the matched schemas' do
expect(subject).to match_array %i[gitlab_main_cell gitlab_ci].to_set
end
context 'when one of the tables does not have a matching table schema' do
let(:tables) { %w[namespaces projects unknown ci_builds] }
it 'raises error' do
expect { subject }.to raise_error(/Could not find gitlab schema for table unknown/)
end
end
end
describe '.table_schema' do
subject { described_class.table_schema(name) }
it_behaves_like 'maps table name to table schema'
context 'when mapping fails' do
let(:name) { 'unknown_table' }
it { is_expected.to be_nil }
end
context 'when an index name is used as the table name' do
before do
ApplicationRecord.connection.execute(<<~SQL)
CREATE INDEX index_on_projects ON public.projects USING gin (name gin_trgm_ops)
SQL
end
let(:name) { 'index_on_projects' }
it { is_expected.to be_nil }
end
end
describe '.table_schema!' do
subject { described_class.table_schema!(name) }
it_behaves_like 'maps table name to table schema'
context 'when mapping fails' do
let(:name) { 'non_existing_table' }
it "raises error" do
expect { subject }.to raise_error(
Gitlab::Database::GitlabSchema::UnknownSchemaError,
"Could not find gitlab schema for table #{name}: " \
"Any new or deleted tables must be added to the database dictionary " \
"See https://docs.gitlab.com/ee/development/database/database_dictionary.html"
)
end
end
end
context 'when testing cross schema access' do
using RSpec::Parameterized::TableSyntax
before do
allow(Gitlab::Database).to receive(:all_gitlab_schemas).and_return(
[
Gitlab::Database::GitlabSchemaInfo.new(
name: "gitlab_main_clusterwide",
allow_cross_joins: %i[gitlab_shared gitlab_main],
allow_cross_transactions: %i[gitlab_internal gitlab_shared gitlab_main],
allow_cross_foreign_keys: %i[gitlab_main]
),
Gitlab::Database::GitlabSchemaInfo.new(
name: "gitlab_main",
allow_cross_joins: %i[gitlab_shared],
allow_cross_transactions: %i[gitlab_internal gitlab_shared],
allow_cross_foreign_keys: %i[]
),
Gitlab::Database::GitlabSchemaInfo.new(
name: "gitlab_ci",
allow_cross_joins: %i[gitlab_shared],
allow_cross_transactions: %i[gitlab_internal gitlab_shared],
allow_cross_foreign_keys: %i[]
),
Gitlab::Database::GitlabSchemaInfo.new(
name: "gitlab_main_cell",
allow_cross_joins: [
:gitlab_shared,
:gitlab_main,
{ gitlab_main_clusterwide: { specific_tables: %w[plans] } }
],
allow_cross_transactions: [
:gitlab_internal,
:gitlab_shared,
:gitlab_main,
{ gitlab_main_clusterwide: { specific_tables: %w[plans] } }
],
allow_cross_foreign_keys: [
{ gitlab_main_clusterwide: { specific_tables: %w[plans] } }
]
)
].index_by(&:name)
)
end
describe '.cross_joins_allowed?' do
where(:schemas, :tables, :result) do
%i[] | %i[] | true
%i[gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_main gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_internal] | %i[] | false
%i[gitlab_main gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_main gitlab_shared] | %i[] | true
%i[gitlab_main_clusterwide gitlab_shared] | %i[] | true
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[users namespaces] | false
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[plans namespaces] | true
end
with_them do
it { expect(described_class.cross_joins_allowed?(schemas, tables)).to eq(result) }
end
end
describe '.cross_transactions_allowed?' do
where(:schemas, :tables, :result) do
%i[] | %i[] | true
%i[gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_main gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_internal] | %i[] | true
%i[gitlab_main gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_main gitlab_shared] | %i[] | true
%i[gitlab_main_clusterwide gitlab_shared] | %i[] | true
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[users namespaces] | false
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[plans namespaces] | true
end
with_them do
it { expect(described_class.cross_transactions_allowed?(schemas, tables)).to eq(result) }
end
end
describe '.cross_foreign_key_allowed?' do
where(:schemas, :tables, :result) do
%i[] | %i[] | false
%i[gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_main] | %i[] | true
%i[gitlab_main_clusterwide gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_internal] | %i[] | false
%i[gitlab_main gitlab_ci] | %i[] | false
%i[gitlab_main_clusterwide gitlab_shared] | %i[] | false
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[users namespaces] | false
%i[gitlab_main_clusterwide gitlab_main_cell] | %w[plans namespaces] | true
end
with_them do
it { expect(described_class.cross_foreign_key_allowed?(schemas, tables)).to eq(result) }
end
end
end
end
|