41 lines
961 B
Ruby
41 lines
961 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
RSpec.describe AsCte do
|
||
|
|
let(:klass) do
|
||
|
|
Class.new(ApplicationRecord) do
|
||
|
|
include AsCte
|
||
|
|
|
||
|
|
self.table_name = 'users'
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
let(:query) { klass.where(id: [1, 2, 3]) }
|
||
|
|
let(:name) { :klass_cte }
|
||
|
|
|
||
|
|
describe '.as_cte' do
|
||
|
|
subject { query.as_cte(name) }
|
||
|
|
|
||
|
|
it { expect(subject).to be_a(Gitlab::SQL::CTE) }
|
||
|
|
it { expect(subject.query).to eq(query) }
|
||
|
|
it { expect(subject.table.name).to eq(name.to_s) }
|
||
|
|
|
||
|
|
context 'with materialized parameter', if: Gitlab::Database::AsWithMaterialized.materialized_supported? do
|
||
|
|
subject { query.as_cte(name, materialized: materialized).to_arel.to_sql }
|
||
|
|
|
||
|
|
context 'as true' do
|
||
|
|
let(:materialized) { true }
|
||
|
|
|
||
|
|
it { expect(subject).to match /MATERIALIZE/ }
|
||
|
|
end
|
||
|
|
|
||
|
|
context 'as false' do
|
||
|
|
let(:materialized) { false }
|
||
|
|
|
||
|
|
it { expect(subject).not_to match /MATERIALIZE/ }
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|