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

data/sunspot/spec/api/indexer/batch_spec.rb

@@ -0,0 +1,72 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'batch indexing', :type => :indexer do
+  let(:posts) { Array.new(2) { |index| Post.new :title => "Post number #{index}!" } }
+
+  it 'should send all batched adds in a single request' do
+    session.batch do
+      for post in posts
+        session.index(post)
+      end
+    end
+    connection.adds.length.should == 1
+  end
+
+  it 'should add all batched adds' do
+    session.batch do
+      for post in posts
+        session.index(post)
+      end
+    end
+    add = connection.adds.last
+    connection.adds.first.map { |add| add.field_by_name(:id).value }.should ==
+      posts.map { |post| "Post #{post.id}" }
+  end
+
+  it 'should not index changes to models that happen after index call' do
+    post = Post.new
+    session.batch do
+      session.index(post)
+      post.title = 'Title'
+    end
+    connection.adds.first.first.field_by_name(:title_ss).should be_nil
+  end
+
+  it 'should batch an add and a delete' do
+    pending 'batching all operations'
+    connection.should_not_receive(:add)
+    connection.should_not_receive(:remove)
+    session.batch do
+      session.index(posts[0])
+      session.remove(posts[1])
+    end
+    connection.adds
+  end
+
+  describe "nesting of batches" do
+    let(:a_nested_batch) do
+      session.batch do
+        session.index posts[0]
+
+        session.batch do
+          session.index posts[1]
+        end
+      end
+    end
+
+    it "behaves like two sets of batches, does the inner first, then outer" do
+      session.batch { session.index posts[1] }
+      session.batch { session.index posts[0] }
+
+      two_sets_of_batches_adds = connection.adds.dup
+      connection.adds.clear
+
+      a_nested_batch
+      nested_batches_adds = connection.adds
+
+      nested_batches_adds.first.first.field_by_name(:title_ss).value.should eq(
+        two_sets_of_batches_adds.first.first.field_by_name(:title_ss).value
+      )
+    end
+  end
+end

data/sunspot/spec/api/indexer/dynamic_fields_spec.rb

@@ -0,0 +1,42 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'indexing dynamic fields' do
+  it 'indexes string data' do
+    session.index(post(:custom_string => { :test => 'string' }))
+    connection.should have_add_with(:"custom_string:test_ss" => 'string')
+  end
+
+  it 'indexes integer data with virtual accessor' do
+    session.index(post(:category_ids => [1, 2]))
+    connection.should have_add_with(:"custom_integer:1_i" => '1', :"custom_integer:2_i" => '1')
+  end
+
+  it 'indexes float data' do
+    session.index(post(:custom_fl => { :test => 1.5 }))
+    connection.should have_add_with(:"custom_float:test_fm" => '1.5')
+  end
+
+  it 'indexes time data' do
+    session.index(post(:custom_time => { :test => Time.parse('2009-05-18 18:05:00 -0400') }))
+    connection.should have_add_with(:"custom_time:test_d" => '2009-05-18T22:05:00Z')
+  end
+
+  it 'indexes boolean data' do
+    session.index(post(:custom_boolean => { :test => false }))
+    connection.should have_add_with(:"custom_boolean:test_b" => 'false')
+  end
+
+  it 'indexes multiple values for a field' do
+    session.index(post(:custom_fl => { :test => [1.0, 2.1, 3.2] }))
+    connection.should have_add_with(:"custom_float:test_fm" => %w(1.0 2.1 3.2))
+  end
+
+  it 'should throw a NoMethodError if dynamic text field defined' do
+    lambda do
+      Sunspot.setup(Post) do
+        dynamic_text :custom_text
+      end
+    end.should raise_error(NoMethodError)
+  end
+end
+

data/sunspot/spec/api/indexer/fixed_fields_spec.rb

@@ -0,0 +1,57 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'indexing fixed fields', :type => :indexer do
+  it 'should index id' do
+    session.index post
+    connection.should have_add_with(:id => "Post #{post.id}")
+  end
+
+  it 'should index type' do
+    session.index post
+    connection.should have_add_with(:type => ['Post', 'SuperClass', 'MockRecord'])
+  end
+
+  it 'should index class name' do
+    session.index post
+    connection.should have_add_with(:class_name => 'Post')
+  end
+
+  it 'should index the array of objects supplied' do
+    posts = Array.new(2) { Post.new }
+    session.index posts
+    connection.should have_add_with(
+      { :id => "Post #{posts.first.id}" },
+      { :id => "Post #{posts.last.id}" }
+    )
+  end
+
+  it 'should index an array containing more than one type of object' do
+    post1, comment, post2 = objects = [Post.new, Namespaced::Comment.new, Post.new]
+    session.index objects
+    connection.should have_add_with(
+      { :id => "Post #{post1.id}", :type => ['Post', 'SuperClass', 'MockRecord'] },
+      { :id => "Namespaced::Comment #{comment.id}", :type => ['Namespaced::Comment', 'MockRecord'] },
+      { :id => "Post #{post2.id}", :type => ['Post', 'SuperClass', 'MockRecord'] }
+    )
+  end
+
+  it 'commits immediately after index! called' do
+    connection.should_receive(:add).ordered
+    connection.should_receive(:commit).ordered
+    session.index!(post)
+  end
+
+  it 'raises an ArgumentError if an attempt is made to index an object that has no configuration' do
+    lambda { session.index(Blog.new) }.should raise_error(Sunspot::NoSetupError)
+  end
+
+  it 'raises a NoAdapterError if class without adapter is indexed' do
+    lambda { session.index(User.new) }.should raise_error(Sunspot::NoAdapterError)
+  end
+
+  it 'raises an ArgumentError if a non-word character is included in the field name' do
+    lambda do
+      Sunspot.setup(Post) { string :"bad name" }
+    end.should raise_error(ArgumentError)
+  end
+end

data/sunspot/spec/api/indexer/fulltext_spec.rb

@@ -0,0 +1,43 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'indexing fulltext fields' do
+  it 'indexes text field' do
+    session.index(post(:title => 'A Title'))
+    connection.should have_add_with(:title_text => 'A Title')
+  end
+
+  it 'indexes stored text field' do
+    session.index(post(:body => 'Test body'))
+    connection.should have_add_with(:body_textsv => 'Test body')
+  end
+
+  it 'indexes text field with boost' do
+    session.index(post(:title => 'A Title'))
+    connection.adds.last.first.field_by_name(:title_text).attrs[:boost].should == 2
+  end
+
+  it 'indexes multiple values for a text field' do
+    session.index(post(:body => %w(some title)))
+    connection.should have_add_with(:body_textsv => %w(some title))
+  end
+
+  it 'indexes text via a block accessor' do
+    session.index(post(:title => 'backwards'))
+    connection.should have_add_with(:backwards_title_text => 'sdrawkcab')
+  end
+
+  it 'indexes document level boost using block' do
+    session.index(post(:ratings_average => 4.0))
+    connection.adds.last.first.attrs[:boost].should == 1.25
+  end
+
+  it 'indexes document level boost using attribute' do
+    session.index(Namespaced::Comment.new(:boost => 1.5))
+    connection.adds.last.first.attrs[:boost].should == 1.5
+  end
+
+  it 'indexes document level boost defined statically' do
+    session.index(Photo.new)
+    connection.adds.last.first.attrs[:boost].should == 0.75
+  end
+end

data/sunspot/spec/api/indexer/removal_spec.rb

@@ -0,0 +1,53 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'document removal', :type => :indexer do
+  it 'removes an object from the index' do
+    session.remove(post)
+    connection.should have_delete("Post #{post.id}")
+  end
+
+  it 'removes an object by type and id' do
+    session.remove_by_id(Post, 1)
+    connection.should have_delete('Post 1')
+  end
+
+  it 'removes an object by type and id and immediately commits' do
+    connection.should_receive(:delete_by_id).with(['Post 1']).ordered
+    connection.should_receive(:commit).ordered
+    session.remove_by_id!(Post, 1)
+  end
+
+  it 'removes an object from the index and immediately commits' do
+    connection.should_receive(:delete_by_id).ordered
+    connection.should_receive(:commit).ordered
+    session.remove!(post)
+  end
+
+  it 'removes everything from the index' do
+    session.remove_all
+    connection.should have_delete_by_query("*:*")
+  end
+
+  it 'removes everything from the index and immediately commits' do
+    connection.should_receive(:delete_by_query).ordered
+    connection.should_receive(:commit).ordered
+    session.remove_all!
+  end
+
+  it 'removes everything of a given class from the index' do
+    session.remove_all(Post)
+    connection.should have_delete_by_query("type:Post")
+  end
+
+  it 'correctly escapes namespaced classes when removing everything from the index' do
+    connection.should_receive(:delete_by_query).with('type:Namespaced\:\:Comment')
+    session.remove_all(Namespaced::Comment)
+  end
+
+  it 'should remove by query' do
+    session.remove(Post) do
+      with(:title, 'monkeys')
+    end
+    connection.should have_delete_by_query("(type:Post AND title_ss:monkeys)")
+  end
+end

data/sunspot/spec/api/indexer/spec_helper.rb

@@ -0,0 +1 @@
+require File.expand_path('spec_helper', File.join(File.dirname(__FILE__), '..'))

data/sunspot/spec/api/indexer_spec.rb

@@ -0,0 +1,14 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe 'indexer', :type => :indexer do
+  it 'should completely wipe setup if class redefined (reloaded)' do
+    Object::ReloadableClass = Class.new(MockRecord)
+    Sunspot.setup(ReloadableClass) { string(:title) }
+    Object.class_eval { remove_const(:ReloadableClass) }
+    Object::ReloadableClass = Class.new(MockRecord)
+    Sunspot.setup(ReloadableClass) {}
+    lambda do
+      Sunspot.search(ReloadableClass) { with(:title, 'title') }
+    end.should raise_error(Sunspot::UnrecognizedFieldError)
+  end
+end

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

@@ -0,0 +1,35 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+shared_examples_for "query with advanced manipulation" do
+  describe 'adjust_solr_params' do
+    before :each do
+      search do
+        adjust_solr_params do |params|
+          params[:rows] = 40
+          params[:qt] = 'complicated'
+        end
+      end
+    end
+
+    it "modifies existing param" do
+      connection.should have_last_search_with(:rows  => 40)
+    end
+
+    it "adds new param" do
+      connection.should have_last_search_with(:qt => 'complicated')
+    end
+  end
+
+  describe 'request_handler' do
+    before :each do
+      connection.expected_handler = :myRequestHandler
+      search do
+        request_handler :myRequestHandler
+      end
+    end
+
+    it 'should use specified request handler' do
+      connection.should have_last_search_with({})
+    end
+  end
+end

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

@@ -0,0 +1,201 @@
+shared_examples_for "query with connective scope" do
+  it 'creates a disjunction between two restrictions' do
+    search do
+      any_of do
+        with :category_ids, 1
+        with :blog_id, 2
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(category_ids_im:1 OR blog_id_i:2)'
+    )
+  end
+
+  it 'creates a conjunction inside of a disjunction' do
+    search do
+      any_of do
+        with :blog_id, 2
+        all_of do
+          with :category_ids, 1
+          with(:average_rating).greater_than(3.0)
+        end
+      end
+    end
+    connection.should have_last_search_including(
+      :fq,
+      '(blog_id_i:2 OR (category_ids_im:1 AND average_rating_ft:{3\.0 TO *}))'
+    )
+  end
+
+  it 'creates a disjunction with nested conjunction with negated restrictions' do
+    search do
+      any_of do
+        with :category_ids, 1
+        all_of do
+          without(:average_rating).greater_than(3.0)
+          with(:blog_id, 1)
+        end
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(category_ids_im:1 OR (-average_rating_ft:{3\.0 TO *} AND blog_id_i:1))'
+    )
+  end
+
+  it 'does nothing special if #all_of called from the top level' do
+    search do
+      all_of do
+        with :blog_id, 2
+        with :category_ids, 1
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, 'blog_id_i:2', 'category_ids_im:1'
+    )
+  end
+
+  it 'creates a disjunction with negated restrictions' do
+    search do
+      any_of do
+        with :category_ids, 1
+        without(:average_rating).greater_than(3.0)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '-(-category_ids_im:1 AND average_rating_ft:{3\.0 TO *})'
+    )
+  end
+
+  it 'creates a disjunction with a negated restriction and a nested disjunction in a conjunction with a negated restriction' do
+    search do
+      any_of do
+        without(:title, 'Yes')
+        all_of do
+          with(:blog_id, 1)
+          any_of do
+            with(:category_ids, 4)
+            without(:average_rating, 2.0)
+          end
+        end
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '-(title_ss:Yes AND -(blog_id_i:1 AND -(-category_ids_im:4 AND average_rating_ft:2\.0)))'
+    )
+  end
+  it 'creates a disjunction with nested conjunction with nested disjunction with negated restriction' do
+    search do
+      any_of do
+        with(:title, 'Yes')
+        all_of do
+          with(:blog_id, 1)
+          any_of do
+            with(:category_ids, 4)
+            without(:average_rating, 2.0)
+          end
+        end
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(title_ss:Yes OR (blog_id_i:1 AND -(-category_ids_im:4 AND average_rating_ft:2\.0)))'
+    )
+  end
+
+  #
+  # This is important because if a disjunction could be nested in another
+  # disjunction, then the inner disjunction could denormalize (and thus
+  # become negated) after the outer disjunction denormalized (checking to
+  # see if the inner one is negated). Since conjunctions never need to
+  # denormalize, if a disjunction can only contain conjunctions or restrictions,
+  # we can guarantee that the negation state of a disjunction's components will
+  # not change when #to_params is called on them.
+  #
+  # Since disjunction is associative, this behavior has no effect on the actual
+  # logical semantics of the disjunction.
+  #
+  it 'creates a single disjunction when disjunctions nested' do
+    search do
+      any_of do
+        with(:title, 'Yes')
+        any_of do
+          with(:blog_id, 1)
+          with(:category_ids, 4)
+        end
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(title_ss:Yes OR blog_id_i:1 OR category_ids_im:4)'
+    )
+  end
+
+  it 'creates a disjunction with instance exclusion' do
+    post = Post.new
+    search do
+      any_of do
+        without(post)
+        with(:category_ids, 1)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, "-(id:(Post\\ #{post.id}) AND -category_ids_im:1)"
+    )
+  end
+
+  it 'creates a disjunction with empty restriction' do
+    search do
+      any_of do
+        with(:average_rating, nil)
+        with(:average_rating).greater_than(3.0)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '-(average_rating_ft:[* TO *] AND -average_rating_ft:{3\.0 TO *})'
+    )
+  end
+
+  it 'creates a disjunction with instance inclusion' do
+    post = Post.new
+    search do
+      any_of do
+        with(post)
+        with(:average_rating).greater_than(3.0)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, "(id:(Post\\ #{post.id}) OR average_rating_ft:{3\\.0 TO *})"
+    )
+  end
+
+  it 'creates a disjunction with some text field components' do
+    search do
+      any_of do
+        text_fields do
+          with(:title).starting_with('test')
+        end
+        with(:blog_id, 1)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(title_text:test* OR blog_id_i:1)'
+    )
+  end
+
+  it 'should ignore empty connectives' do
+    search do
+      any_of {}
+    end
+    connection.should_not have_last_search_including(:fq, '')
+  end
+
+  it 'creates a conjunction of in_radius queries' do
+    search do
+      any_of do
+        with(:coordinates_new).in_radius(23, -46, 100)
+        with(:coordinates_new).in_radius(42, 56, 50)
+      end
+    end
+    connection.should have_last_search_including(
+      :fq, '(_query_:"{!geofilt sfield=coordinates_new_ll pt=23,-46 d=100}" OR _query_:"{!geofilt sfield=coordinates_new_ll pt=42,56 d=50}")'
+    )
+  end
+end