require "spec_helper"

describe PgSearch::Multisearch::Rebuilder do
  with_table "pg_search_documents", {}, &DOCUMENTS_SCHEMA

  describe "#rebuild" do
    context "when the model defines .rebuild_pg_search_documents" do
      context "and multisearchable is not conditional" do
        with_model :Model do
          model do
            include PgSearch
            multisearchable

            def rebuild_pg_search_documents
            end
          end
        end

        it "should call .rebuild_pg_search_documents" do
          rebuilder = PgSearch::Multisearch::Rebuilder.new(Model)
          Model.should_receive(:rebuild_pg_search_documents)
          rebuilder.rebuild
        end
      end

      context "when multisearch is conditional" do
        [:if, :unless].each do |conditional_key|
          context "via :#{conditional_key}" do
            with_model :Model do
              table do |t|
                t.boolean :active
              end

              model do
                include PgSearch
                multisearchable conditional_key => :active?

                def rebuild_pg_search_documents
                end
              end
            end

            it "should call .rebuild_pg_search_documents" do
              rebuilder = PgSearch::Multisearch::Rebuilder.new(Model)
              Model.should_receive(:rebuild_pg_search_documents)
              rebuilder.rebuild
            end
          end
        end
      end
    end

    context "when the model does not define .rebuild_pg_search_documents" do
      context "when multisearchable is not conditional" do
        with_model :Model do
          table do |t|
            t.string :name
          end

          model do
            include PgSearch
            multisearchable :against => :name
          end
        end

        it "should not call :rebuild_pg_search_documents" do
          rebuilder = PgSearch::Multisearch::Rebuilder.new(Model)

          # stub respond_to? to return false since should_not_receive defines the method
          original_respond_to = Model.method(:respond_to?)
          Model.stub(:respond_to?) do |method_name, *args|
            if method_name == :rebuild_pg_search_documents
              false
            else
              original_respond_to.call(method_name, *args)
            end
          end

          Model.should_not_receive(:rebuild_pg_search_documents)
          rebuilder.rebuild
        end

        it "should execute the default SQL" do
          time = DateTime.parse("2001-01-01")
          rebuilder = PgSearch::Multisearch::Rebuilder.new(Model, lambda { time } )

          expected_sql = <<-SQL
INSERT INTO "pg_search_documents" (searchable_type, searchable_id, content, created_at, updated_at)
  SELECT 'Model' AS searchable_type,
         #{Model.quoted_table_name}.id AS searchable_id,
         (
           coalesce(#{Model.quoted_table_name}.name::text, '')
         ) AS content,
         '2001-01-01 00:00:00' AS created_at,
         '2001-01-01 00:00:00' AS updated_at
  FROM #{Model.quoted_table_name}
          SQL

          executed_sql = []

          notifier = ActiveSupport::Notifications.subscribe("sql.active_record") do |name, start, finish, id, payload|
            executed_sql << payload[:sql]
          end

          rebuilder.rebuild
          ActiveSupport::Notifications.unsubscribe(notifier)

          executed_sql.length.should == 1
          executed_sql.first.should == expected_sql
        end
      end

      context "when multisearchable is conditional" do
        context "via :if" do
          with_model :Model do
            table do |t|
              t.boolean :active
            end

            model do
              include PgSearch
              multisearchable :if => :active?
            end
          end

          it "calls update_pg_search_document on each record" do
            record1 = Model.create!(:active => true)
            record2 = Model.create!(:active => false)

            rebuilder = PgSearch::Multisearch::Rebuilder.new(Model)

            # stub respond_to? to return false since should_not_receive defines the method
            original_respond_to = Model.method(:respond_to?)
            Model.stub(:respond_to?) do |method_name, *args|
              if method_name == :rebuild_pg_search_documents
                false
              else
                original_respond_to.call(method_name, *args)
              end
            end
            Model.should_not_receive(:rebuild_pg_search_documents)

            rebuilder.rebuild

            record1.pg_search_document.should be_present
            record2.pg_search_document.should_not be_present
          end
        end

        context "via :unless" do
          with_model :Model do
            table do |t|
              t.boolean :inactive
            end

            model do
              include PgSearch
              multisearchable :unless => :inactive?
            end
          end

          it "calls update_pg_search_document on each record" do
            record1 = Model.create!(:inactive => true)
            record2 = Model.create!(:inactive => false)

            rebuilder = PgSearch::Multisearch::Rebuilder.new(Model)

            # stub respond_to? to return false since should_not_receive defines the method
            original_respond_to = Model.method(:respond_to?)
            Model.stub(:respond_to?) do |method_name, *args|
              if method_name == :rebuild_pg_search_documents
                false
              else
                original_respond_to.call(method_name, *args)
              end
            end
            Model.should_not_receive(:rebuild_pg_search_documents)

            rebuilder.rebuild

            record1.pg_search_document.should_not be_present
            record2.pg_search_document.should be_present
          end
        end
      end
    end
  end
end