sunspot_rails 2.0.0.pre.120720 → 2.0.0.pre.120924

data/sunspot/spec/api/query/fulltext_examples.rb

@@ -0,0 +1,313 @@
+shared_examples_for 'fulltext query' do
+  it 'searches by keywords' do
+    search do
+      keywords 'keyword search'
+    end
+    connection.should have_last_search_with(:q => 'keyword search')
+  end
+
+  it 'ignores keywords if empty' do
+    search do
+      keywords ''
+    end
+    connection.should_not have_last_search_with(:defType => 'dismax')
+  end
+
+  it 'ignores keywords if nil' do
+    search do
+      keywords nil
+    end
+    connection.should_not have_last_search_with(:defType => 'dismax')
+  end
+
+  it 'ignores keywords with only whitespace' do
+    search do
+      keywords "  \t"
+    end
+    connection.should_not have_last_search_with(:defType => 'dismax')
+  end
+
+  it 'gracefully ignores keywords block if keywords ignored' do
+    search do
+      keywords(nil) { fields(:title) }
+    end
+  end
+
+  it 'sets default query parser to dismax when keywords used' do
+    search do
+      keywords 'keyword search'
+    end
+    connection.should have_last_search_with(:defType => 'dismax')
+  end
+
+  it 'searches types in filter query if keywords used' do
+    search do
+      keywords 'keyword search'
+    end
+    connection.should have_last_search_with(:fq => ['type:Post'])
+  end
+
+  describe 'with multiple keyword components' do
+    before :each do
+      session.search Post do
+        keywords 'first search', :fields => :title
+        keywords 'second search'
+      end
+    end
+
+    it 'puts specified keywords in subquery' do
+      subqueries(:q).map { |subquery| subquery[:v] }.should ==
+        ['first search', 'second search']
+    end
+
+    it 'puts specified dismax parameters in subquery' do
+      subqueries(:q).first[:qf].should == 'title_text'
+    end
+
+    it 'puts default dismax parameters in subquery' do
+      subqueries(:q).last[:qf].split(' ').sort.should == %w(backwards_title_text body_textsv tags_textv title_text)
+    end
+
+    it 'puts field list in main query' do
+      connection.should have_last_search_with(:fl => '* score')
+    end
+  end
+
+  it 'searches all text fields for searched class' do
+    search = search do
+      keywords 'keyword search'
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(backwards_title_text body_textsv tags_textv title_text)
+  end
+
+  it 'searches both stored and unstored text fields' do
+    search Post, Namespaced::Comment do
+      keywords 'keyword search'
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(author_name_text backwards_title_text body_text body_textsv tags_textv title_text)
+  end
+
+  it 'searches only specified text fields when specified' do
+    search do
+      keywords 'keyword search', :fields => [:title, :body]
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(body_textsv title_text)
+  end
+
+  it 'excludes text fields when instructed' do
+    search do
+      keywords 'keyword search' do
+        exclude_fields :backwards_title, :body_mlt
+      end
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(body_textsv tags_textv title_text)
+  end
+
+  it 'assigns boost to fields when specified' do
+    search do
+      keywords 'keyword search' do
+        fields :title => 2.0, :body => 0.75
+      end
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(body_textsv^0.75 title_text^2.0)
+  end
+
+  it 'allows assignment of boosted and unboosted fields' do
+    search do
+      keywords 'keyword search' do
+        fields :body, :title => 2.0
+      end
+    end
+  end
+
+  it 'searches both unstored and stored text field with same name when specified' do
+    search Post, Namespaced::Comment do
+      keywords 'keyword search', :fields => [:body]
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(body_text body_textsv)
+  end
+
+  it 'requests score when keywords used' do
+    search do
+      keywords 'keyword search'
+    end
+    connection.should have_last_search_with(:fl => '* score')
+  end
+
+  it 'does not request score when keywords not used' do
+    search Post
+    connection.should_not have_last_search_with(:fl)
+  end
+
+  it 'sets phrase fields' do
+    search do
+      keywords 'great pizza' do
+        phrase_fields :title => 2.0
+      end
+    end
+    connection.should have_last_search_with(:pf => 'title_text^2.0')
+  end
+
+  it 'sets phrase fields with boost' do
+    search do
+      keywords 'great pizza' do
+        phrase_fields :title => 1.5
+      end
+    end
+    connection.should have_last_search_with(:pf => 'title_text^1.5')
+  end
+
+  it 'sets phrase slop from DSL' do
+    search do
+      keywords 'great pizza' do
+        phrase_slop 2
+      end
+    end
+    connection.should have_last_search_with(:ps => 2)
+  end
+
+  it 'sets boost for certain fields without restricting fields' do
+    search do
+      keywords 'great pizza' do
+        boost_fields :title => 1.5
+      end
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(backwards_title_text body_textsv tags_textv title_text^1.5)
+  end
+
+  it 'ignores boost fields that do not apply' do
+    search do
+      keywords 'great pizza' do
+        boost_fields :bogus => 1.2, :title => 1.5
+      end
+    end
+    connection.searches.last[:qf].split(' ').sort.should == %w(backwards_title_text body_textsv tags_textv title_text^1.5)
+  end
+
+  it 'sets default boost with default fields' do
+    search Photo do
+      keywords 'great pizza'
+    end
+    connection.should have_last_search_with(:qf => 'caption_text^1.5')
+  end
+
+  it 'sets default boost with fields specified in options' do
+    search Photo do
+      keywords 'great pizza', :fields => [:caption]
+    end
+    connection.should have_last_search_with(:qf => 'caption_text^1.5')
+  end
+
+  it 'sets default boost with fields specified in DSL' do
+    search Photo do
+      keywords 'great pizza' do
+        fields :caption
+      end
+    end
+    connection.should have_last_search_with(:qf => 'caption_text^1.5')
+  end
+
+  it 'overrides default boost when specified in DSL' do
+    search Photo do
+      keywords 'great pizza' do
+        fields :caption => 2.0
+      end
+    end
+    connection.should have_last_search_with(:qf => 'caption_text^2.0')
+  end
+
+  it 'creates boost query' do
+    search do
+      keywords 'great pizza' do
+        boost 2.0 do
+          with(:average_rating).greater_than(2.0)
+        end
+      end
+    end
+    connection.should have_last_search_with(:bq => ['average_rating_ft:{2\.0 TO *}^2.0'])
+  end
+
+  it 'creates multiple boost queries' do
+    search do
+      keywords 'great pizza' do
+        boost(2.0) do
+          with(:average_rating).greater_than(2.0)
+        end
+        boost(1.5) do
+          with(:featured, true)
+        end
+      end
+    end
+    connection.should have_last_search_with(
+      :bq => [
+        'average_rating_ft:{2\.0 TO *}^2.0',
+        'featured_bs:true^1.5'
+      ]
+    )
+  end
+
+  it 'sends minimum match parameter from options' do
+    search do
+      keywords 'great pizza', :minimum_match => 2
+    end
+    connection.should have_last_search_with(:mm => 2)
+  end
+
+  it 'sends minimum match parameter from DSL' do
+    search do
+      keywords('great pizza') { minimum_match(2) }
+    end
+    connection.should have_last_search_with(:mm => 2)
+  end
+
+  it 'sends tiebreaker parameter from options' do
+    search do
+      keywords 'great pizza', :tie => 0.1
+    end
+    connection.should have_last_search_with(:tie => 0.1)
+  end
+
+  it 'sends tiebreaker parameter from DSL' do
+    search do
+      keywords('great pizza') { tie(0.1) }
+    end
+    connection.should have_last_search_with(:tie => 0.1)
+  end
+
+  it 'sends query phrase slop from options' do
+    search do
+      keywords 'great pizza', :query_phrase_slop => 2
+    end
+    connection.should have_last_search_with(:qs => 2)
+  end
+
+  it 'sends query phrase slop from DSL' do
+    search do
+      keywords('great pizza') { query_phrase_slop(2) }
+    end
+    connection.should have_last_search_with(:qs => 2)
+  end
+
+  it 'allows specification of a text field that only exists in one type' do
+    search Post, Namespaced::Comment do
+      keywords 'keywords', :fields => :author_name
+    end
+    connection.searches.last[:qf].should == 'author_name_text'
+  end
+
+  it 'raises Sunspot::UnrecognizedFieldError for nonexistant fields in keywords' do
+    lambda do
+      search do
+        keywords :text, :fields => :bogus
+      end
+    end.should raise_error(Sunspot::UnrecognizedFieldError)
+  end
+
+  it 'raises Sunspot::UnrecognizedFieldError if a text field that does not exist for any type is specified' do
+    lambda do
+      search Post, Namespaced::Comment do
+        keywords 'fulltext', :fields => :bogus
+      end
+    end.should raise_error(Sunspot::UnrecognizedFieldError)
+  end
+end

data/sunspot/spec/api/query/function_spec.rb

@@ -0,0 +1,79 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'function query' do
+  it "should send query to solr with boost function" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { :average_rating })
+      end
+    end
+    connection.should have_last_search_including(:bf, 'average_rating_ft')
+  end
+
+  it "should handle boost function with constant float" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { 10.5 })
+      end
+    end
+    connection.should have_last_search_including(:bf, '10.5')
+  end
+
+  it "should handle boost function with time literal" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { Time.parse('2010-03-25 14:13:00 EDT') })
+      end
+    end
+    connection.should have_last_search_including(:bf, '2010-03-25T18:13:00Z')
+  end
+ 
+  it "should handle arbitrary functions in a function query block" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { product(:average_rating, 10) })
+      end
+    end
+    connection.should have_last_search_including(:bf, 'product(average_rating_ft,10)')
+  end
+
+  it "should handle the sub function in a function query block" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { sub(:average_rating, 10) })
+      end
+    end
+    connection.should have_last_search_including(:bf, 'sub(average_rating_ft,10)')
+  end
+ 
+  it "should handle nested functions in a function query block" do
+    session.search Post do
+      keywords('pizza') do
+        boost(function { product(:average_rating, sum(:average_rating, 20)) })
+      end
+    end
+    connection.should have_last_search_including(:bf, 'product(average_rating_ft,sum(average_rating_ft,20))')
+  end
+
+  # TODO SOLR 1.5
+  it "should raise ArgumentError if string literal passed" do
+    lambda do
+      session.search Post do
+        keywords('pizza') do
+          boost(function { "hello world" })
+        end
+      end
+    end.should raise_error(ArgumentError)
+  end
+
+  it "should raise UnrecognizedFieldError if bogus field name passed" do
+    lambda do
+      session.search Post do
+        keywords('pizza') do
+          boost(function { :bogus })
+        end
+      end
+    end.should raise_error(Sunspot::UnrecognizedFieldError)
+  end
+end
+

data/sunspot/spec/api/query/geo_examples.rb

@@ -0,0 +1,68 @@
+require 'bigdecimal'
+
+shared_examples_for 'geohash query' do
+  it 'searches for nearby points with defaults' do
+    search do
+      with(:coordinates).near(40.7, -73.5)
+    end
+    connection.should have_last_search_including(:q, build_geo_query)
+  end
+
+  it 'searches for nearby points with non-Float arguments' do
+    search do
+      with(:coordinates).near(BigDecimal.new('40.7'), BigDecimal.new('-73.5'))
+    end
+    connection.should have_last_search_including(:q, build_geo_query)
+  end
+
+  it 'searches for nearby points with given precision' do
+    search do
+      with(:coordinates).near(40.7, -73.5, :precision => 10)
+    end
+    connection.should have_last_search_including(:q, build_geo_query(:precision => 10))
+  end
+
+  it 'searches for nearby points with given precision factor' do
+    search do
+      with(:coordinates).near(40.7, -73.5, :precision_factor => 1.5)
+    end
+    connection.should have_last_search_including(:q, build_geo_query(:precision_factor => 1.5))
+  end
+
+  it 'searches for nearby points with given boost' do
+    search do
+      with(:coordinates).near(40.7, -73.5, :boost => 2.0)
+    end
+    connection.should have_last_search_including(:q, build_geo_query(:boost => 2.0))
+  end
+
+  it 'performs both dismax search and location search' do
+    search do
+      fulltext 'pizza', :fields => :title
+      with(:coordinates).near(40.7, -73.5)
+    end
+    expected = 
+      "{!dismax fl='* score' qf='title_text'}pizza (#{build_geo_query})"
+    connection.should have_last_search_including(
+      :q,
+      %Q(_query_:"{!dismax qf='title_text'}pizza" (#{build_geo_query}))
+    )
+  end
+
+  private
+
+  def build_geo_query(options = {})
+    precision = options[:precision] || 7
+    precision_factor = options[:precision_factor] || 16.0
+    boost = options[:boost] || 1.0
+    hash = 'dr5xx3nytvgs'
+    (precision..12).map do |i|
+      phrase = 
+        if i == 12 then hash
+        else "#{hash[0, i]}*"
+        end
+      precision_boost = Sunspot::Util.format_float(boost*precision_factor**(i-12.0), 3)
+      "coordinates_s:#{phrase}^#{precision_boost}"
+    end.reverse.join(' OR ')
+  end
+end