spec/models/concerns/from_set_operator_spec.rb


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