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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::UsageDataQueries do
describe '#add_metric' do
let(:metric) { 'CountBoardsMetric' }
it 'builds the query for given metric' do
expect(described_class.add_metric(metric)).to eq('SELECT COUNT("boards"."id") FROM "boards"')
end
end
describe '.with_metadata' do
it 'yields passed block' do
expect { |block| described_class.with_metadata(&block) }.to yield_with_no_args
end
end
describe '.count' do
it 'returns the raw SQL' do
expect(described_class.count(User)).to start_with('SELECT COUNT("users"."id") FROM "users"')
end
it 'does not mix a nil column with keyword arguments' do
expect(described_class.count(User, nil)).to eq('SELECT COUNT("users"."id") FROM "users"')
end
end
describe '.distinct_count' do
it 'returns the raw SQL' do
expect(described_class.distinct_count(Issue, :author_id)).to eq('SELECT COUNT(DISTINCT "issues"."author_id") FROM "issues"')
end
it 'does not mix a nil column with keyword arguments' do
expect(described_class.distinct_count(Issue, nil, start: 1, finish: 2)).to eq('SELECT COUNT(DISTINCT "issues"."id") FROM "issues"')
end
end
describe '.redis_usage_data' do
subject(:redis_usage_data) { described_class.redis_usage_data { 42 } }
it 'returns a stringified class for redis_usage_data with a counter call' do
expect(described_class.redis_usage_data(Gitlab::UsageDataCounters::WikiPageCounter))
.to eq(redis_usage_data_counter: "Gitlab::UsageDataCounters::WikiPageCounter")
end
it 'returns a placeholder string for redis_usage_data with a block' do
is_expected.to include(:redis_usage_data_block)
expect(redis_usage_data[:redis_usage_data_block]).to eq('non-SQL usage data block')
end
end
describe '.alt_usage_data' do
subject(:alt_usage_data) { described_class.alt_usage_data { 42 } }
it 'returns value when used with value' do
expect(described_class.alt_usage_data(1))
.to eq(alt_usage_data_value: 1)
end
it 'returns a placeholder string for alt_usage_data with a block' do
expect(alt_usage_data[:alt_usage_data_block]).to eq('non-SQL usage data block')
end
end
describe '.sum' do
it 'returns the raw SQL' do
expect(described_class.sum(Issue, :weight)).to eq('SELECT SUM("issues"."weight") FROM "issues"')
end
end
describe '.add' do
it 'returns the combined raw SQL with an inner query' do
expect(described_class.add(
'SELECT COUNT("users"."id") FROM "users"',
'SELECT COUNT("issues"."id") FROM "issues"'
)).to eq('SELECT (SELECT COUNT("users"."id") FROM "users") + (SELECT COUNT("issues"."id") FROM "issues")')
end
end
describe '.histogram' do
it 'returns the histogram sql' do
expect(described_class.histogram(
AlertManagement::HttpIntegration.active, :project_id, buckets: 1..2, bucket_size: 101
)).to match(/^WITH "count_cte" AS MATERIALIZED/)
end
end
describe 'min/max methods' do
it 'returns nil' do
# user min/max
expect(described_class.minimum_id(User)).to eq(nil)
expect(described_class.maximum_id(User)).to eq(nil)
# issue min/max
expect(described_class.minimum_id(Issue)).to eq(nil)
expect(described_class.maximum_id(Issue)).to eq(nil)
# deployment min/max
expect(described_class.minimum_id(Deployment)).to eq(nil)
expect(described_class.maximum_id(Deployment)).to eq(nil)
# project min/max
expect(described_class.minimum_id(Project)).to eq(nil)
expect(described_class.maximum_id(Project)).to eq(nil)
end
end
describe 'sent_in_product_marketing_email_count' do
it 'returns sql query that returns correct value' do
expect(described_class.sent_in_product_marketing_email_count(nil, 0, 0)).to eq(
'SELECT COUNT("in_product_marketing_emails"."id") ' \
'FROM "in_product_marketing_emails" ' \
'WHERE "in_product_marketing_emails"."track" = 0 AND "in_product_marketing_emails"."series" = 0'
)
end
end
describe 'clicked_in_product_marketing_email_count' do
it 'returns sql query that returns correct value' do
expect(described_class.clicked_in_product_marketing_email_count(nil, 0, 0)).to eq(
'SELECT COUNT("in_product_marketing_emails"."id") ' \
'FROM "in_product_marketing_emails" ' \
'WHERE "in_product_marketing_emails"."track" = 0 AND "in_product_marketing_emails"."series" = 0 ' \
'AND "in_product_marketing_emails"."cta_clicked_at" IS NOT NULL'
)
end
end
end
|
