require File.join(File.dirname(__FILE__), 'spec_helper')

describe 'highlighted fulltext queries', :type => :query do
  it 'should not send highlight parameter when highlight not requested' do
    session.search(Post) do
      keywords 'test'
    end
    connection.should_not have_last_search_with(:hl)
  end

  it 'should enable highlighting when highlighting requested as keywords argument' do
    session.search(Post) do
      keywords 'test', :highlight => true
    end
    connection.should have_last_search_with(:hl => 'on')
  end

  it 'should not set highlight fields parameter if highlight fields are not passed' do
    session.search(Post) do
      keywords 'test', :highlight => true, :fields => [:title]
    end
    connection.should_not have_last_search_with(:'hl.fl')
  end

  it 'should enable highlighting on multiple fields when highlighting requested as array of fields via keywords argument' do
    session.search(Post) do
      keywords 'test', :highlight => [:title, :body]
    end

    connection.should have_last_search_with(:hl => 'on', :'hl.fl' => %w(title_text body_texts))
  end

  it 'should raise UnrecognizedFieldError if try to highlight unexisting field via keywords argument' do
    lambda {
      session.search(Post) do
        keywords 'test', :highlight => [:unknown_field]
      end
    }.should raise_error(Sunspot::UnrecognizedFieldError)
  end

  it 'should enable highlighting on multiple fields when highlighting requested as list of fields via block call' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :body
      end
    end

    connection.should have_last_search_with(:hl => 'on', :'hl.fl' => %w(title_text body_texts))
  end

  it 'should enable highlighting on multiple fields for multiple search types' do
    session.search(Post, Namespaced::Comment) do
      keywords 'test' do
        highlight :body
      end
    end
    connection.searches.last[:'hl.fl'].to_set.should == Set['body_text', 'body_texts']
  end

  it 'should raise UnrecognizedFieldError if try to highlight unexisting field via block call' do
    lambda {
      session.search(Post) do
        keywords 'test' do
          highlight :unknown_field
        end
      end
    }.should raise_error(Sunspot::UnrecognizedFieldError)
  end

  it 'should set internal formatting' do
    session.search(Post) do
      keywords 'test', :highlight => true
    end
    connection.should have_last_search_with(
      :"hl.simple.pre" => '@@@hl@@@',
      :"hl.simple.post" => '@@@endhl@@@'
    )
  end

  it 'should set highlight fields from DSL' do
    session.search(Post) do
      keywords 'test' do
        highlight :title
      end
    end
    connection.should have_last_search_with(
      :"hl.fl" => %w(title_text)
    )
  end

  it 'should not set formatting params specific to fields if fields specified' do
    session.search(Post) do
      keywords 'test', :highlight => :body
    end
    connection.should have_last_search_with(
      :"hl.simple.pre" => '@@@hl@@@',
      :"hl.simple.post" => '@@@endhl@@@'
    )
  end

  it 'should set maximum highlights per field' do
    session.search(Post) do
      keywords 'test' do
        highlight :max_snippets => 3
      end
    end
    connection.should have_last_search_with(
      :"hl.snippets" => 3
    )
  end

  it 'should set max snippets specific to highlight fields' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :max_snippets => 3
      end
    end
    connection.should have_last_search_with(
      :"hl.fl"       => %w(title_text),
      :"f.title_text.hl.snippets" => 3
    )
  end

  it 'should set the maximum size' do
    session.search(Post) do
      keywords 'text' do
        highlight :fragment_size => 200
      end
    end
    connection.should have_last_search_with(
      :"hl.fragsize" => 200
    )
  end

  it 'should set the maximum size for specific fields' do
    session.search(Post) do
      keywords 'text' do
        highlight :title, :fragment_size => 200
      end
    end
    connection.should have_last_search_with(
      :"f.title_text.hl.fragsize" => 200
    )
  end

  it 'enables merging of contiguous fragments' do
    session.search(Post) do
      keywords 'test' do
        highlight :merge_contiguous_fragments => true
      end
    end
    connection.should have_last_search_with(
      :"hl.mergeContiguous" => 'true'
    )
  end

  it 'enables merging of contiguous fragments for specific fields' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :merge_contiguous_fragments => true
      end
    end
    connection.should have_last_search_with(
      :"f.title_text.hl.mergeContiguous" => 'true'
    )
  end

  it 'enables use of phrase highlighter' do
    session.search(Post) do
      keywords 'test' do
        highlight :phrase_highlighter => true
      end
    end
    connection.should have_last_search_with(
      :"hl.usePhraseHighlighter" => 'true'
    )
  end

  it 'enables use of phrase highlighter for specific fields' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :phrase_highlighter => true
      end
    end
    connection.should have_last_search_with(
      :"f.title_text.hl.usePhraseHighlighter" => 'true'
    )
  end

  it 'requires field match if requested' do
    session.search(Post) do
      keywords 'test' do
        highlight :phrase_highlighter => true, :require_field_match => true
      end
    end
    connection.should have_last_search_with(
      :"hl.requireFieldMatch" => 'true'
    )
  end

  it 'requires field match for specified field if requested' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :phrase_highlighter => true, :require_field_match => true
      end
    end
    connection.should have_last_search_with(
      :"f.title_text.hl.requireFieldMatch" => 'true'
    )
  end

  it 'sets field specific params for different fields if different params given' do
    session.search(Post) do
      keywords 'test' do
        highlight :title, :max_snippets => 2
        highlight :body, :max_snippets => 1
      end
    end
    connection.should have_last_search_with(
      :"hl.fl" => %w(title_text body_texts),
      :"f.title_text.hl.snippets" => 2,
      :"f.body_texts.hl.snippets" => 1
    )
  end
end