torque-postgresql 1.1.1 → 2.0.1

data/spec/tests/auxiliary_statement_spec.rb

@@ -0,0 +1,593 @@
+require 'spec_helper'
+
+RSpec.describe 'AuxiliaryStatement' do
+  before :each do
+    User.auxiliary_statements_list = {}
+  end
+
+  context 'on relation' do
+    let(:klass) { User }
+    let(:true_value) { 'TRUE' }
+    subject { klass.unscoped }
+
+    it 'has its method' do
+      expect(subject).to respond_to(:with)
+    end
+
+    it 'can perform simple queries' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'can perform more complex queries' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.distinct_on(:user_id).order(:user_id, id: :desc)
+        cte.attributes content: :last_comment
+      end
+
+      result = 'WITH "comments" AS (SELECT DISTINCT ON ( "comments"."user_id" )'
+      result << ' "comments"."user_id", "comments"."content" AS last_comment'
+      result << ' FROM "comments" ORDER BY "comments"."user_id" ASC,'
+      result << ' "comments"."id" DESC) SELECT "users".*,'
+      result << ' "comments"."last_comment" FROM "users" INNER JOIN "comments"'
+      result << ' ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts extra select columns' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content, "comments"."slug" AS comment_slug FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_content", "comments"."comment_slug" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments, select: {slug: :comment_slug}).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts extra join columns' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."active", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id" AND "comments"."active" = "users"."active"'
+      expect(subject.with(:comments, join: {active: :active}).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts extra conditions' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content'
+      result << ' FROM "comments" WHERE "comments"."active" = $1)'
+      result << ' SELECT "users".*, "comments"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments, where: {active: true}).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts scopes from both sides' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.where(id: 1).all
+        cte.attributes content: :comment_content
+      end
+
+      query = subject.where(id: 2).with(:comments)
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments"'
+      result << ' WHERE "comments"."id" = $1)'
+      result << ' SELECT "users".*, "comments"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      result << ' WHERE "users"."id" = $2'
+
+      expect(query.arel.to_sql).to eql(result)
+      expect(query.send(:bound_attributes).map(&:value_before_type_cast)).to eql([1, 2])
+    end
+
+    it 'accepts string as attributes' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes sql('MAX(id)') => :comment_id
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", MAX(id) AS comment_id FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_id" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts complex string as attributes' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes sql('ROW_NUMBER() OVER (PARTITION BY ORDER BY "comments"."id")') => :comment_id
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", ROW_NUMBER() OVER (PARTITION BY ORDER BY "comments"."id") AS comment_id FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_id" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts arel attribute as attributes' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes col(:id).minimum => :comment_id
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", MIN("comments"."id") AS comment_id FROM "comments")'
+      result << ' SELECT "users".*, "comments"."comment_id" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts custom join properties' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+        cte.join name: :id, 'a.col' => :col
+      end
+
+      result = 'WITH "comments" AS (SELECT "comments"."id", "comments"."col",'
+      result << ' "comments"."content" AS comment_content FROM "comments") SELECT "users".*,'
+      result << ' "comments"."comment_content" FROM "users" INNER JOIN "comments"'
+      result << ' ON "comments"."id" = "users"."name" AND "comments"."col" = "a"."col"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'can perform other types of joins' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+        cte.join_type :left
+      end
+
+      result = 'WITH "comments" AS (SELECT "comments"."user_id",'
+      result << ' "comments"."content" AS comment_content FROM "comments") SELECT "users".*,'
+      result << ' "comments"."comment_content" FROM "users" LEFT OUTER JOIN "comments"'
+      result << ' ON "comments"."user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'can manually define the association' do
+      klass.has_many :sample_comment, class_name: 'Comment', foreign_key: :a_user_id
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.through :sample_comment
+        cte.attributes content: :sample_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."a_user_id", "comments"."content" AS sample_content FROM "comments")'
+      result << ' SELECT "users".*, "comments"."sample_content" FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."a_user_id" = "users"."id"'
+      expect(subject.with(:comments).arel.to_sql).to eql(result)
+    end
+
+    it 'accepts complex scopes from dependencies' do
+      klass.send(:auxiliary_statement, :comments1) do |cte|
+        cte.query Comment.where(id: 1).all
+        cte.attributes content: :comment_content1
+      end
+
+      klass.send(:auxiliary_statement, :comments2) do |cte|
+        cte.requires :comments1
+        cte.query Comment.where(id: 2).all
+        cte.attributes content: :comment_content2
+      end
+
+      query = subject.where(id: 3).with(:comments2)
+
+      result = 'WITH '
+      result << '"comments1" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content1 FROM "comments" WHERE "comments"."id" = $1), '
+      result << '"comments2" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content2 FROM "comments" WHERE "comments"."id" = $2)'
+      result << ' SELECT "users".*, "comments1"."comment_content1", "comments2"."comment_content2" FROM "users"'
+      result << ' INNER JOIN "comments1" ON "comments1"."user_id" = "users"."id"'
+      result << ' INNER JOIN "comments2" ON "comments2"."user_id" = "users"."id"'
+      result << ' WHERE "users"."id" = $3'
+
+      expect(query.arel.to_sql).to eql(result)
+      expect(query.send(:bound_attributes).map(&:value_before_type_cast)).to eql([1, 2, 3])
+    end
+
+    context 'with dependency' do
+      before :each do
+        klass.send(:auxiliary_statement, :comments1) do |cte|
+          cte.query Comment.all
+          cte.attributes content: :comment_content1
+        end
+
+        klass.send(:auxiliary_statement, :comments2) do |cte|
+          cte.requires :comments1
+          cte.query Comment.all
+          cte.attributes content: :comment_content2
+        end
+      end
+
+      it 'can requires another statement as dependency' do
+        result = 'WITH '
+        result << '"comments1" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content1 FROM "comments"), '
+        result << '"comments2" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content2 FROM "comments")'
+        result << ' SELECT "users".*, "comments1"."comment_content1", "comments2"."comment_content2" FROM "users"'
+        result << ' INNER JOIN "comments1" ON "comments1"."user_id" = "users"."id"'
+        result << ' INNER JOIN "comments2" ON "comments2"."user_id" = "users"."id"'
+        expect(subject.with(:comments2).arel.to_sql).to eql(result)
+      end
+
+      it 'can uses already already set dependent' do
+        result = 'WITH '
+        result << '"comments1" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content1 FROM "comments"), '
+        result << '"comments2" AS (SELECT "comments"."user_id", "comments"."content" AS comment_content2 FROM "comments")'
+        result << ' SELECT "users".*, "comments1"."comment_content1", "comments2"."comment_content2" FROM "users"'
+        result << ' INNER JOIN "comments1" ON "comments1"."user_id" = "users"."id"'
+        result << ' INNER JOIN "comments2" ON "comments2"."user_id" = "users"."id"'
+        expect(subject.with(:comments1, :comments2).arel.to_sql).to eql(result)
+      end
+
+      it 'raises an error if the dependent does not exist' do
+        klass.send(:auxiliary_statement, :comments2) do |cte|
+          cte.requires :comments3
+          cte.query Comment.all
+          cte.attributes content: :comment_content2
+        end
+        expect{ subject.with(:comments2).arel.to_sql }.to raise_error(ArgumentError)
+      end
+    end
+
+    context 'query as string' do
+      it 'performs correctly' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, 'SELECT * FROM comments'
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        result = 'WITH "comments" AS (SELECT * FROM comments)'
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+        expect(subject.with(:comments).arel.to_sql).to eql(result)
+      end
+
+      it 'accepts arguments to format the query' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, 'SELECT * FROM comments WHERE active = %{active}'
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        result = "WITH \"comments\" AS (SELECT * FROM comments WHERE active = #{true_value})"
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+        expect(subject.with(:comments, args: {active: true}).arel.to_sql).to eql(result)
+      end
+
+      it 'raises an error when join columns are not given' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, 'SELECT * FROM comments'
+          cte.attributes content: :comment
+        end
+
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /join columns/)
+      end
+
+      it 'raises an error when not given the table name as first argument' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query 'SELECT * FROM comments'
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /table name/)
+      end
+    end
+
+    context 'query as proc' do
+      it 'performs correctly for result as relation' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, -> { Comment.all }
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        result = 'WITH "comments" AS'
+        result << ' (SELECT "comments"."user_id", "comments"."content" AS comment FROM "comments")'
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+        expect(subject.with(:comments).arel.to_sql).to eql(result)
+      end
+
+      it 'performs correctly for anything that has a call method' do
+        obj = Struct.new(:call, :arity).new('SELECT * FROM comments', 0)
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, obj
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        result = 'WITH "comments" AS (SELECT * FROM comments)'
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+        expect(subject.with(:comments).arel.to_sql).to eql(result)
+      end
+
+      it 'performs correctly for result as string' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, -> { 'SELECT * FROM comments' }
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        result = 'WITH "comments" AS (SELECT * FROM comments)'
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+        expect(subject.with(:comments).arel.to_sql).to eql(result)
+      end
+
+      it 'performs correctly when the proc requires arguments' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, -> (args) { Comment.where(id: args.id) }
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        query = subject.with(:comments, args: {id: 1})
+
+        result = 'WITH "comments" AS'
+        result << ' (SELECT "comments"."user_id", "comments"."content" AS comment'
+        result << ' FROM "comments" WHERE "comments"."id" = $1)'
+        result << ' SELECT "users".*, "comments"."comment" FROM "users"'
+        result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+
+        expect(query.arel.to_sql).to eql(result)
+        expect(query.send(:bound_attributes).map(&:value_before_type_cast)).to eql([1])
+      end
+
+      it 'raises an error when join columns are not given' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, -> { Author.all }
+          cte.attributes content: :comment
+        end
+
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /join columns/)
+      end
+
+      it 'raises an error when not given the table name as first argument' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query -> { Comment.all }
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /table name/)
+      end
+
+      it 'raises an error when the result of the proc is an invalid type' do
+        klass.send(:auxiliary_statement, :comments) do |cte|
+          cte.query :comments, -> { false }
+          cte.attributes content: :comment
+          cte.join id: :user_id
+        end
+
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /query objects/)
+      end
+    end
+
+    context 'with inheritance' do
+      let(:base) { Activity }
+      let(:klass) { ActivityBook }
+
+      it 'accepts ancestors auxiliary statements' do
+        base.send(:auxiliary_statement, :authors) do |cte|
+          cte.query Author.all
+          cte.attributes name: :author_name
+          cte.join author_id: :id
+        end
+
+        result = 'WITH "authors" AS'
+        result << ' (SELECT "authors"."id", "authors"."name" AS author_name FROM "authors")'
+        result << ' SELECT "activity_books".*, "authors"."author_name" FROM "activity_books"'
+        result << ' INNER JOIN "authors" ON "authors"."id" = "activity_books"."author_id"'
+        expect(subject.with(:authors).arel.to_sql).to eql(result)
+      end
+
+      it 'can replace ancestors auxiliary statements' do
+        base.send(:auxiliary_statement, :authors) do |cte|
+          cte.query Author.all
+          cte.attributes name: :author_name
+          cte.join author_id: :id
+        end
+
+        klass.send(:auxiliary_statement, :authors) do |cte|
+          cte.query Author.all
+          cte.attributes type: :author_type
+          cte.join author_id: :id
+        end
+
+        result = 'WITH "authors" AS'
+        result << ' (SELECT "authors"."id", "authors"."type" AS author_type FROM "authors")'
+        result << ' SELECT "activity_books".*, "authors"."author_type" FROM "activity_books"'
+        result << ' INNER JOIN "authors" ON "authors"."id" = "activity_books"."author_id"'
+        expect(subject.with(:authors).arel.to_sql).to eql(result)
+      end
+
+      it 'raises an error when no class has the auxiliary statement' do
+        expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError)
+      end
+    end
+
+    it 'works with count and does not add extra columns' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT COUNT(*) FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+
+      query = get_last_executed_query{ subject.with(:comments).count }
+      expect(query).to eql(result)
+    end
+
+    it 'works with sum and does not add extra columns' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes id: :value
+      end
+
+      result = 'WITH "comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."id" AS value FROM "comments")'
+      result << ' SELECT SUM("comments"."value") FROM "users"'
+      result << ' INNER JOIN "comments" ON "comments"."user_id" = "users"."id"'
+
+      query = get_last_executed_query{ subject.with(:comments).sum(comments: :value) }
+      expect(query).to eql(result)
+    end
+
+    it 'raises an error when using an invalid type of object as query' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query :string, String
+      end
+
+      expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError, /object types/)
+    end
+
+    it 'raises an error when traying to use a statement that is not defined' do
+      expect{ subject.with(:does_not_exist).arel.to_sql }.to raise_error(ArgumentError)
+    end
+
+    it 'raises an error when using an invalid type of join' do
+      klass.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+        cte.join_type :invalid
+      end
+
+      expect{ subject.with(:comments).arel.to_sql }.to raise_error(ArgumentError)
+    end
+  end
+
+  context 'on model' do
+    subject { User }
+
+    it 'has its configurator' do
+      expect(subject.protected_methods).to include(:cte)
+      expect(subject.protected_methods).to include(:auxiliary_statement)
+    end
+
+    it 'allows configurate new auxiliary statements' do
+      subject.send(:auxiliary_statement, :cte1)
+      expect(subject.auxiliary_statements_list).to include(:cte1)
+      expect(subject.const_defined?('Cte1_AuxiliaryStatement')).to be_truthy
+    end
+
+    it 'has its query method' do
+      expect(subject).to respond_to(:with)
+    end
+
+    it 'returns a relation when using the method' do
+      subject.send(:auxiliary_statement, :comments) do |cte|
+        cte.query Comment.all
+        cte.attributes content: :comment_content
+      end
+      expect(subject.with(:comments)).to be_a(ActiveRecord::Relation)
+    end
+  end
+
+  context 'on external' do
+    let(:klass) { Torque::PostgreSQL::AuxiliaryStatement }
+    subject { User }
+
+    it 'has the external method available' do
+      expect(klass).to respond_to(:create)
+    end
+
+    it 'accepts simple auxiliary statement definition' do
+      sample = klass.create(Comment.all)
+      query = subject.with(sample, select: {content: :comment_content}).arel.to_sql
+
+      result = 'WITH "comment" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT "users".*, "comment"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comment" ON "comment"."user_id" = "users"."id"'
+      expect(query).to eql(result)
+    end
+
+    it 'accepts a hash auxiliary statement definition' do
+      sample = klass.create(query: Comment.all, select: {content: :comment_content})
+      query = subject.with(sample).arel.to_sql
+
+      result = 'WITH "comment" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT "users".*, "comment"."comment_content" FROM "users"'
+      result << ' INNER JOIN "comment" ON "comment"."user_id" = "users"."id"'
+      expect(query).to eql(result)
+    end
+
+    it 'accepts a block when creating the auxiliary statement' do
+      sample = klass.create(:all_comments) do |cte|
+        cte.query Comment.all
+        cte.select content: :comment_content
+      end
+
+      result = 'WITH "all_comments" AS'
+      result << ' (SELECT "comments"."user_id", "comments"."content" AS comment_content FROM "comments")'
+      result << ' SELECT "users".*, "all_comments"."comment_content" FROM "users"'
+      result << ' INNER JOIN "all_comments" ON "all_comments"."user_id" = "users"."id"'
+
+      query = subject.with(sample).arel.to_sql
+      expect(query).to eql(result)
+    end
+  end
+
+  context 'on settings' do
+    let(:base) { User }
+    let(:statement_klass) do
+      base.send(:auxiliary_statement, :statement)
+      base::Statement_AuxiliaryStatement
+    end
+
+    subject do
+      Torque::PostgreSQL::AuxiliaryStatement::Settings.new(base, statement_klass)
+    end
+
+    it 'has access to base' do
+      expect(subject.base).to eql(User)
+      expect(subject.base_table).to be_a(Arel::Table)
+    end
+
+    it 'has access to statement table' do
+      expect(subject.table_name).to eql('statement')
+      expect(subject.table).to be_a(Arel::Table)
+    end
+
+    it 'has access to the query arel table' do
+      subject.query Comment.all
+      expect(subject.query_table).to be_a(Arel::Table)
+    end
+
+    it 'raises an error when trying to access query table before defining the query' do
+      expect{ subject.with(:comments).arel.to_sql }.to raise_error(StandardError)
+    end
+  end
+end