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
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe FromSetOperator do
let_it_be(:from_set_operator) do
Class.new do
extend FromSetOperator
define_set_operator Gitlab::SQL::Union
def table_name
'groups'
end
def from(*args)
''
end
end
end
context 'when set operator method already exists' do
let(:redefine_method) do
Class.new do
def self.from_union
# This method intentionally left blank.
end
extend FromSetOperator
define_set_operator Gitlab::SQL::Union
end
end
it { expect { redefine_method }.to raise_exception(RuntimeError) }
end
context 'with members' do
let_it_be(:group1) { create :group }
let_it_be(:group2) { create :group }
let_it_be(:groups) do
[
Group.where(id: group1),
Group.where(id: group2)
]
end
shared_examples 'set operator called with correct members' do
it do
expect(Gitlab::SQL::Union).to receive(:new).with(groups, anything).and_call_original
subject
end
end
context 'as array' do
subject { from_set_operator.new.from_union(groups) }
it_behaves_like 'set operator called with correct members'
it { expect { subject }.not_to make_queries }
end
context 'as multiple parameters' do
subject { from_set_operator.new.from_union(*groups) }
it_behaves_like 'set operator called with correct members'
it { expect { subject }.not_to make_queries }
end
end
end
|
