gojee-sunspot 2.0.2

data/lib/sunspot/version.rb

@@ -0,0 +1,3 @@
+module Sunspot
+  VERSION = '2.0.2'
+end

data/script/console

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+require 'irb'
+require 'rubygems'
+require 'bundler'
+Bundler.setup(:default, :test)
+require 'sunspot'
+def test_env!
+  require File.join(File.dirname(__FILE__), '..', 'spec', 'integration', 'spec_helper')
+end
+IRB.start

data/spec/api/adapters_spec.rb

@@ -0,0 +1,33 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe Sunspot::Adapters::InstanceAdapter do
+  it "finds adapter by superclass" do
+    Sunspot::Adapters::InstanceAdapter::for(Model).should be(AbstractModelInstanceAdapter)
+  end
+
+  it "finds adapter by mixin" do
+    Sunspot::Adapters::InstanceAdapter::for(MixModel).should be(MixInModelInstanceAdapter)
+  end
+
+  it 'throws NoAdapterError if anonymous module passed in' do
+    lambda do
+      Sunspot::Adapters::InstanceAdapter::for(Module.new)
+    end.should raise_error(Sunspot::NoAdapterError)
+  end
+end
+
+describe Sunspot::Adapters::DataAccessor do
+  it "finds adapter by superclass" do
+    Sunspot::Adapters::DataAccessor::for(Model).should be(AbstractModelDataAccessor)
+  end
+
+  it "finds adapter by mixin" do
+    Sunspot::Adapters::DataAccessor::for(MixModel).should be(MixInModelDataAccessor)
+  end
+
+  it 'throws NoAdapterError if anonymous module passed in' do
+    lambda do
+      Sunspot::Adapters::DataAccessor::for(Module.new)
+    end.should raise_error(Sunspot::NoAdapterError)
+  end
+end

data/spec/api/batcher_spec.rb

@@ -0,0 +1,112 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe Sunspot::Batcher do
+  it "includes Enumerable" do
+    described_class.should include Enumerable
+  end
+
+  describe "#each" do
+    let(:current) { [:foo, :bar] }
+    before { subject.stub(:current).and_return current }
+
+    it "iterates over current" do
+      yielded_values = []
+
+      subject.each do |value|
+        yielded_values << value
+      end
+
+      yielded_values.should eq current
+    end
+  end
+
+  describe "adding to current batch" do
+    it "#push pushes to current" do
+      subject.push :foo
+      subject.current.should include :foo
+    end
+
+    it "#<< pushes to current" do
+      subject.push :foo
+      subject.current.should include :foo
+    end
+
+    it "#concat concatinates on current batch" do
+      subject << :foo
+      subject.concat [:bar, :mix]
+      should include :foo, :bar, :mix
+    end
+  end
+
+
+  describe "#current" do
+    context "no current" do
+      it "starts a new" do
+        expect { subject.current }.to change(subject, :depth).by 1
+      end
+
+      it "is empty by default" do
+        subject.current.should be_empty
+      end
+    end
+
+    context "with a current" do
+      before { subject.start_new }
+
+      it "does not start a new" do
+        expect { subject.current }.to_not change(subject, :depth)
+      end
+
+      it "returns the same as last time" do
+        subject.current.should eq subject.current
+      end
+    end
+  end
+
+  describe "#start_new" do
+    it "creates a new batches" do
+      expect { 2.times { subject.start_new } }.to change(subject, :depth).by 2
+    end
+
+    it "changes current" do
+      subject << :foo
+      subject.start_new
+      should_not include :foo
+    end
+  end
+
+  describe "#end_current" do
+    context "no current batch" do
+      it "fails" do
+        expect { subject.end_current }.to raise_error Sunspot::Batcher::NoCurrentBatchError
+      end
+    end
+
+    context "with current batch" do
+      before { subject.start_new }
+
+      it "changes current" do
+        subject << :foo
+        subject.end_current
+        should_not include :foo
+      end
+
+      it "returns current" do
+        subject << :foo
+        subject.end_current.should include :foo
+      end
+    end
+  end
+
+  describe "#batching?" do
+    it "is false when depth is 0" do
+      subject.should_receive(:depth).and_return 0
+      should_not be_batching
+    end
+
+    it "is true when depth is more than 0" do
+      subject.should_receive(:depth).and_return 1
+      should be_batching
+    end
+  end
+end

data/spec/api/binding_spec.rb

@@ -0,0 +1,50 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe "DSL bindings" do
+  it 'should give access to calling context\'s methods in search DSL' do
+    value = nil
+    session.search(Post) do
+      value = test_method
+    end
+    value.should == 'value'
+  end
+
+  it 'should give access to calling context\'s id method in search DSL' do
+    value = nil
+    session.search(Post) do
+      value = id
+    end
+    value.should == 16
+  end
+
+  it 'should give access to calling context\'s methods in nested DSL block' do
+    value = nil
+    session.search(Post) do
+      any_of do
+        value = test_method
+      end
+    end
+    value.should == 'value'
+  end
+
+  it 'should give access to calling context\'s methods in double-nested DSL block' do
+    value = nil
+    session.search(Post) do
+      any_of do
+        all_of do
+          value = test_method
+        end
+      end
+    end
+  end
+
+  private
+
+  def test_method
+    'value'
+  end
+
+  def id
+    16
+  end
+end

data/spec/api/class_set_spec.rb

@@ -0,0 +1,24 @@
+require "spec_helper"
+
+describe Sunspot::ClassSet do
+  it "is enumerable" do
+    class1, class2 = stub(:name => "Class1"), stub(:name => "Class2")
+
+    set = described_class.new
+    set << class1 << class2
+
+    set.to_a.should =~ [class1, class2]
+  end
+
+  it "replaces classes with the same name" do
+    set = described_class.new
+
+    class1 = stub(:name => "Class1")
+    set << class1
+    set.to_a.should == [class1]
+
+    class1_dup = stub(:name => "Class1")
+    set << class1_dup
+    set.to_a.should == [class1_dup]
+  end
+end

data/spec/api/hit_enumerable_spec.rb

@@ -0,0 +1,47 @@
+require File.join(File.dirname(__FILE__), "spec_helper")
+
+describe Sunspot::Search::HitEnumerable do
+  subject do
+    Class.new do
+      include Sunspot::Search::HitEnumerable
+    end.new
+  end
+
+  describe "#hits" do
+    before do
+      subject.stub(:solr_docs).and_return([{"id" => "Post 1", "score" => 3.14}])
+      subject.stub(:highlights_for)
+    end
+
+    it "retrieves the raw Solr response from #solr_docs and constructs Hit objects" do
+      Sunspot::Search::Hit.should_receive(:new).
+                           with({"id" => "Post 1", "score" => 3.14}, anything, anything)
+
+      subject.hits
+    end
+
+    it "constructs Hit objects with highlights" do
+      subject.should_receive(:highlights_for).with({"id" => "Post 1", "score" => 3.14})
+
+      subject.hits
+    end
+
+    it "returns only verified hits if :verify => true is passed" do
+      Sunspot::Search::Hit.any_instance.stub(:result).and_return(nil)
+
+      subject.hits(:verify => true).should be_empty
+    end
+
+    it "returns an empty array if no results are available from Solr" do
+      subject.stub(:solr_docs).and_return(nil)
+
+      subject.hits.should == []
+    end
+
+    it "provides #populate_hits so that querying for one hit result will eager load the rest" do
+      Sunspot::Search::Hit.any_instance.should_receive(:result=)
+
+      subject.populate_hits
+    end
+  end
+end

data/spec/api/indexer/attributes_spec.rb

@@ -0,0 +1,149 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+require 'bigdecimal'
+
+describe 'indexing attribute fields', :type => :indexer do
+  it 'should correctly index a stored string attribute field' do
+    session.index(post(:title => 'A Title'))
+    connection.should have_add_with(:title_ss => 'A Title')
+  end
+
+  it 'should correctly index an integer attribute field' do
+    session.index(post(:blog_id => 4))
+    connection.should have_add_with(:blog_id_i => '4')
+  end
+
+  it 'should correctly index a long attribute field' do
+    session.index(Namespaced::Comment.new(:hash => 2**30))
+    connection.should have_add_with(:hash_l => '1073741824')
+  end
+
+  it 'should correctly index a float attribute field' do
+    session.index(post(:ratings_average => 2.23))
+    connection.should have_add_with(:average_rating_ft => '2.23')
+  end
+
+  it 'should correctly index a double attribute field' do
+    session.index(Namespaced::Comment.new(:average_rating => 2.23))
+    connection.should have_add_with(:average_rating_e => '2.23')
+  end
+
+  it 'should correctly index a trie integer attribute field' do
+    session.index(Photo.new(:size => 104856))
+    connection.should have_add_with(:size_it => '104856')
+  end
+
+  it 'should correctly index a trie float attribute field' do
+    session.index(Photo.new(:average_rating => 2.23))
+    connection.should have_add_with(:average_rating_ft => '2.23')
+  end
+
+  it 'should correctly index a trie time attribute field' do
+    session.index(Photo.new(:created_at => Time.parse('2009-12-16 15:00:00 -0400')))
+    connection.should have_add_with(:created_at_dt => '2009-12-16T19:00:00Z')
+  end
+
+  it 'should allow indexing by a multiple-value field' do
+    session.index(post(:category_ids => [3, 14]))
+    connection.should have_add_with(:category_ids_im => ['3', '14'])
+  end
+
+  it 'should not index a single-value field with newlines as multiple' do
+    session.index(post(:title => "Multi\nLine"))
+    connection.adds.last.first.field_by_name(:title_ss).value.should == "Multi\nLine"
+  end
+
+  it 'should correctly index a time field' do
+    session.index(
+      post(:published_at => Time.parse('1983-07-08 05:00:00 -0400'))
+    )
+    connection.should have_add_with(:published_at_dt => '1983-07-08T09:00:00Z')
+  end
+
+  it 'should correctly index a time field that\'s after 32-bit Y2K' do
+    session.index(
+      post(:published_at => DateTime.parse('2050-07-08 05:00:00 -0400'))
+    )
+    connection.should have_add_with(:published_at_dt => '2050-07-08T09:00:00Z')
+  end
+
+  it 'should correctly index a date field' do
+    session.index(post(:expire_date => Date.new(2009, 07, 13)))
+    connection.should have_add_with(:expire_date_d => '2009-07-13T00:00:00Z')
+  end
+
+  it 'should correctly index a boolean field' do
+    session.index(post(:featured => true))
+    connection.should have_add_with(:featured_bs => 'true')
+  end
+
+  it 'should correctly index a false boolean field' do
+    session.index(post(:featured => false))
+    connection.should have_add_with(:featured_bs => 'false')
+  end
+
+  it 'should not index a nil boolean field' do
+    session.index(post)
+    connection.should_not have_add_with(:featured_bs)
+  end
+
+  it 'should index latitude and longitude as a pair' do
+    session.index(post(:coordinates => Sunspot::Util::Coordinates.new(40.7, -73.5)))
+    connection.should have_add_with(:coordinates_s => 'dr5xx3nytvgs')
+  end
+
+  it 'should index latitude and longitude passed as non-Floats' do
+    coordinates = Sunspot::Util::Coordinates.new(
+      BigDecimal.new('40.7'), BigDecimal.new('-73.5'))
+    session.index(post(:coordinates => coordinates))
+    connection.should have_add_with(:coordinates_s => 'dr5xx3nytvgs')
+  end
+
+  it 'should correctly index an attribute field with block access' do
+    session.index(post(:title => 'The Blog Post'))
+    connection.should have_add_with(:sort_title_s => 'blog post')
+  end
+
+  it 'should correctly index an attribute field with instance-external block access' do
+    session.index(post(:category_ids => [1, 2, 3]))
+    connection.should have_add_with(:primary_category_id_i => '1')
+  end
+
+  it 'should correctly index a field that is defined on a superclass' do
+    Sunspot.setup(SuperClass) { string :author_name }
+    session.index(post(:author_name => 'Mat Brown'))
+    connection.should have_add_with(:author_name_s => 'Mat Brown')
+  end
+
+  it 'should throw a NoMethodError only if a nonexistent type is defined' do
+    lambda { Sunspot.setup(Post) { string :author_name }}.should_not raise_error
+    lambda { Sunspot.setup(Post) { bogus :journey }}.should raise_error(NoMethodError)
+  end
+
+  it 'should throw a NoMethodError if a nonexistent field argument is passed' do
+    lambda { Sunspot.setup(Post) { string :author_name, :bogus => :argument }}.should raise_error(ArgumentError)
+  end
+
+  it 'should throw an ArgumentError if single-value field tries to index multiple values' do
+    lambda do
+      Sunspot.setup(Post) { string :author_name }
+      session.index(post(:author_name => ['Mat Brown', 'Matthew Brown']))
+    end.should raise_error(ArgumentError)
+  end
+
+  it 'should throw an ArgumentError if specifying more_like_this on type that does not support it' do
+    lambda do
+      Sunspot.setup(Post) { integer :popularity, :more_like_this => true }
+    end.should raise_error(ArgumentError)
+  end
+
+  it 'should use a specified field name when the :as option is set' do
+    session.index(post(:title => 'A Title'))
+    connection.should have_add_with(:legacy_field_s => 'legacy A Title')
+  end
+
+  it 'should use a specified field name when the :as option is set for array values' do
+    session.index(post(:title => 'Another Title'))
+    connection.should have_add_with(:legacy_array_field_sm => ['first string', 'second string'])
+ 	end
+end
+

data/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