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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::Database::Partitioning::TimePartition do
describe '.from_sql' do
subject { described_class.from_sql(table, partition_name, definition) }
let(:table) { 'foo' }
let(:partition_name) { 'foo_bar' }
let(:definition) { 'FOR VALUES FROM (\'2020-04-01 00:00:00\') TO (\'2020-05-01 00:00:00\')' }
it 'uses specified table name' do
expect(subject.table).to eq(table)
end
it 'uses specified partition name' do
expect(subject.partition_name).to eq(partition_name)
end
it 'parses start date' do
expect(subject.from).to eq(Date.parse('2020-04-01'))
end
it 'parses end date' do
expect(subject.to).to eq(Date.parse('2020-05-01'))
end
context 'with MINVALUE as a start date' do
let(:definition) { 'FOR VALUES FROM (MINVALUE) TO (\'2020-05-01\')' }
it 'sets from to nil' do
expect(subject.from).to be_nil
end
end
context 'with MAXVALUE as an end date' do
let(:definition) { 'FOR VALUES FROM (\'2020-04-01\') TO (MAXVALUE)' }
it 'raises a NotImplementedError' do
expect { subject }.to raise_error(NotImplementedError)
end
end
end
describe '#partition_name' do
subject { described_class.new(table, from, to, partition_name: partition_name).partition_name }
let(:table) { 'foo' }
let(:from) { '2020-04-01 00:00:00' }
let(:to) { '2020-05-01 00:00:00' }
let(:partition_name) { nil }
it 'uses table as prefix' do
expect(subject).to start_with(table)
end
it 'uses Year-Month (from) as suffix' do
expect(subject).to end_with("_202004")
end
context 'without from date' do
let(:from) { nil }
it 'uses 000000 as suffix for first partition' do
expect(subject).to end_with("_000000")
end
end
context 'with partition name explicitly given' do
let(:partition_name) { "foo_bar" }
it 'uses given partition name' do
expect(subject).to eq(partition_name)
end
end
end
describe '#to_sql' do
subject { described_class.new(table, from, to).to_sql }
let(:table) { 'foo' }
let(:from) { '2020-04-01 00:00:00' }
let(:to) { '2020-05-01 00:00:00' }
it 'transforms to a CREATE TABLE statement' do
expect(subject).to eq(<<~SQL)
CREATE TABLE IF NOT EXISTS "#{Gitlab::Database::DYNAMIC_PARTITIONS_SCHEMA}"."foo_202004"
PARTITION OF "foo"
FOR VALUES FROM ('2020-04-01') TO ('2020-05-01')
SQL
end
context 'without from date' do
let(:from) { nil }
it 'uses MINVALUE instead' do
expect(subject).to eq(<<~SQL)
CREATE TABLE IF NOT EXISTS "#{Gitlab::Database::DYNAMIC_PARTITIONS_SCHEMA}"."foo_000000"
PARTITION OF "foo"
FOR VALUES FROM (MINVALUE) TO ('2020-05-01')
SQL
end
end
end
describe 'object equality - #eql' do
def expect_inequality(actual, other)
expect(actual.eql?(other)).to be_falsey
expect(actual).not_to eq(other)
end
def expect_equality(actual, other)
expect(actual).to eq(other)
expect(actual.eql?(other)).to be_truthy
expect(actual.hash).to eq(other.hash)
end
def make_new(table: 'foo', from: '2020-04-01 00:00:00', to: '2020-05-01 00:00:00', partition_name: 'foo_202004')
described_class.new(table, from, to, partition_name: partition_name)
end
it 'treats objects identical with identical attributes' do
expect_equality(make_new, make_new)
end
it 'different table leads to in-equality' do
expect_inequality(make_new, make_new(table: 'bar'))
end
it 'different from leads to in-equality' do
expect_inequality(make_new, make_new(from: '2020-05-01 00:00:00'))
end
it 'different to leads to in-equality' do
expect_inequality(make_new, make_new(to: '2020-06-01 00:00:00'))
end
it 'different partition_name leads to in-equality' do
expect_inequality(make_new, make_new(partition_name: 'different'))
end
it 'nil partition_name is ignored if auto-generated matches' do
expect_equality(make_new, make_new(partition_name: nil))
end
end
describe 'Comparable, #<=>' do
let(:table) { 'foo' }
it 'sorts by partition name, i.e. by month - MINVALUE partition first' do
partitions = [
described_class.new(table, '2020-04-01', '2020-05-01'),
described_class.new(table, '2020-02-01', '2020-03-01'),
described_class.new(table, nil, '2020-02-01'),
described_class.new(table, '2020-03-01', '2020-04-01')
]
expect(partitions.sort).to eq([
described_class.new(table, nil, '2020-02-01'),
described_class.new(table, '2020-02-01', '2020-03-01'),
described_class.new(table, '2020-03-01', '2020-04-01'),
described_class.new(table, '2020-04-01', '2020-05-01')
])
end
it 'returns nil for partitions of different tables' do
one = described_class.new('foo', '2020-02-01', '2020-03-01')
two = described_class.new('bar', '2020-02-01', '2020-03-01')
expect(one.<=>(two)).to be_nil
end
end
end
|
