69 lines
1.5 KiB
Ruby
69 lines
1.5 KiB
Ruby
# 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
|