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
                it "uses qry to count the records" do
                  @relation.expects(:loaded?).returns(false)
                  @relation.expects(:qry).with("COUNT(*)").returns([[1982]])
                  assert_equal 1982, @relation.count(:all)
                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