require_relative "../../../test_helper"

module Unit
  module GemExt
    module ActiveRecord
      class TestRelation < MiniTest::Test

        describe ::ActiveRecord::Relation do
          before do
            @relation = Office.where(:id => 1)
          end

          describe "#qry_options" do
            it "initiates a DirectiveRecord::Relation instance and returns the query options" do
              relation = mock
              relation.expects(:qry_options).returns(:where => "id = 1")
              DirectiveRecord::Relation.expects(:new).with(@relation).returns(relation)
              assert_equal({:where => "id = 1"}, @relation.qry_options)
            end
          end

          describe "#to_qry" do
            it "delegates to its klass with qry_options" do
              @relation.expects(:qry_options).with("city").returns(:select => "city", :where => ["id = 1"])
              Office.expects(:to_qry).with(:select => "city", :where => ["id = 1"]).returns("SELECT city FROM offices WHERE id = 1")
              assert_equal "SELECT city FROM offices WHERE id = 1", @relation.to_qry("city")
            end
          end

          describe "#qry" do
            it "delegates to its klass with qry_options" do
              @relation.expects(:qry_options).with("city").returns(:select => "city", :where => ["id = 1"])
              Office.expects(:qry).with(:select => "city", :where => ["id = 1"]).returns(%w(NYC))
              assert_equal %w(NYC), @relation.qry("city")
            end
          end

          describe "#count" do
            describe "when only passing :all" do
              describe "when loaded" do
                it "invokes the original count method" do
                  @relation.expects(:loaded?).returns(true)
                  @relation.expects(:original_count)
                  @relation.count(:all)
                end
              end
              describe "when not loaded" do
                describe "when not having potential path query options" do
                  it "invokes the original count method" do
                    @relation.expects(:loaded?).returns(false)
                    @relation.expects(:qry_options).returns({:where => ["foo = 12"]})
                    @relation.expects(:original_count)
                    @relation.count(:all)

                    @relation.expects(:loaded?).returns(false)
                    @relation.expects(:qry_options).returns({:where => ["foo.bar = 12"]})
                    @relation.expects(:original_count)
                    @relation.count(:all)
                  end
                end
                describe "when having potential path query options" do
                  it "uses #qry to count the records" do
                    @relation.expects(:loaded?).returns(false)
                    @relation.expects(:qry_options).returns({:where => ["employees.foo = 12"]})
                    @relation.expects(:qry).with("COUNT(DISTINCT id)").returns([[1982]])
                    assert_equal 1982, @relation.count(:all)
                  end
                end
              end
            end
            describe "when otherwise" do
              it "invokes the original count method" do
                @relation.expects(:loaded?).returns(false)
                @relation.expects(:original_count)
                @relation.count(:foo)

                @relation.expects(:loaded?).returns(false)
                @relation.expects(:original_count)
                @relation.count(:all, {:foo => :bar})
              end
            end
          end
        end

      end
    end
  end
end