ruben-sunspot 1.1.0

data/spec/api/search/results_spec.rb

@@ -0,0 +1,79 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe 'search results', :type => :search do
+  it 'loads single result' do
+    post = Post.new
+    stub_results(post)
+    session.search(Post).results.should == [post]
+  end
+
+  it 'loads multiple results in order' do
+    post_1, post_2 = Post.new, Post.new
+    stub_results(post_1, post_2)
+    session.search(Post).results.should == [post_1, post_2]
+    stub_results(post_2, post_1)
+    session.search(Post).results.should == [post_2, post_1]
+  end
+
+  # This is a reduction of a crazy bug I found in production where some hits
+  # were inexplicably not being populated.
+  it 'properly loads results of multiple classes that have the same primary key' do
+    Post.reset!
+    Namespaced::Comment.reset!
+    results = [Post.new, Namespaced::Comment.new]
+    stub_results(*results)
+    session.search(Post, Namespaced::Comment).results.should == results
+  end
+
+  it 'gracefully returns empty results when response is nil' do
+    stub_nil_results
+    session.search(Post).results.should == []
+  end
+
+  if ENV['USE_WILL_PAGINATE']
+
+    it 'returns search total as attribute of results' do
+      stub_results(Post.new, 4)
+      session.search(Post) do
+        paginate(:page => 1)
+      end.results.total_entries.should == 4
+    end
+
+  else
+
+    it 'returns vanilla array if WillPaginate is not available' do
+      stub_results(Post.new)
+      session.search(Post) do
+        paginate(:page => 1)
+      end.results.should_not respond_to(:total_entries)
+    end
+
+  end
+
+  it 'returns total' do
+    stub_results(Post.new, Post.new, 4)
+    session.search(Post) { paginate(:page => 1) }.total.should == 4
+  end
+
+  it 'returns total for nil search' do
+    stub_nil_results
+    session.search(Post).total.should == 0
+  end
+
+  it 'returns available results if some results are not available from data store' do
+    posts = [Post.new, Post.new]
+    posts.last.destroy
+    stub_results(*posts)
+    session.search(Post).results.should == posts[0..0]
+  end
+
+  it 'does not attempt to query the data store more than once when results are unavailable' do
+    posts = [Post.new, Post.new]
+    posts.each { |post| post.destroy }
+    stub_results(*posts)
+    search = session.search(Post) do
+      data_accessor_for(Post).should_receive(:load_all).once.and_return([])
+    end
+    search.results.should == []
+  end
+end

data/spec/api/search/search_spec.rb

@@ -0,0 +1,23 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot::Search do
+  it 'should allow access to the data accessor' do
+    stub_results(posts = Post.new)
+    search = session.search Post do
+      data_accessor_for(Post).custom_title = 'custom title'
+    end
+    search.results.first.title.should == 'custom title'
+  end
+  
+  it 'should re-execute search' do
+    post_1, post_2 = Post.new, Post.new
+    
+    stub_results(post_1)
+    search = session.search Post
+    search.results.should == [post_1]
+    
+    stub_results(post_2)
+    search.execute!
+    search.results.should == [post_2]
+  end
+end

data/spec/api/search/spec_helper.rb

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

data/spec/api/server_spec.rb

@@ -0,0 +1,91 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+require 'tempfile'
+
+describe Sunspot::Server do
+  SUNSPOT_START_JAR = File.expand_path(
+    File.join(File.dirname(__FILE__), '..', '..', 'solr', 'start.jar')
+  )
+
+  before :each do
+    @server = Sunspot::Server.new
+  end
+
+  it 'runs server in current process' do
+    @server.should_not_receive(:fork)
+    @server.should_receive(:exec).with(/java .*-jar start.jar/)
+    @server.run
+  end
+
+  it 'runs Java with min memory' do
+    @server.min_memory = 1024
+    @server.should_receive(:exec).with(/-Xms1024/)
+    @server.run
+  end
+
+  it 'runs Java with max memory' do
+    @server.max_memory = 2048
+    @server.should_receive(:exec).with(/-Xmx2048/)
+    @server.run
+  end
+
+  it 'runs Jetty with specified port' do
+    @server.port = 8981
+    @server.should_receive(:exec).with(/-Djetty\.port=8981/)
+    @server.run
+  end
+
+  it 'runs Solr with specified data dir' do
+    @server.solr_data_dir = '/var/solr/data'
+    @server.should_receive(:exec).with(%r(-Dsolr\.data\.dir=/var/solr/data))
+    @server.run
+  end
+
+  it 'runs Solr with specified Solr home' do
+    @server.solr_home = '/var/solr'
+    @server.should_receive(:exec).with(%r(-Dsolr\.solr\.home=/var/solr))
+    @server.run
+  end
+
+  it 'runs Solr with specified Solr jar' do
+    @server.solr_jar = SUNSPOT_START_JAR
+    FileUtils.should_receive(:cd).with(File.dirname(SUNSPOT_START_JAR))
+    @server.run
+  end
+
+  describe 'with logging' do
+    before :each do
+      @server.log_level = 'info'
+      @server.log_file = 'log/sunspot-development.log'
+      Tempfile.should_receive(:new).with('logging.properties').and_return(@tempfile = StringIO.new)
+      @tempfile.should_receive(:flush)
+      @tempfile.should_receive(:close)
+      @tempfile.stub!(:path).and_return('/tmp/logging.properties.12345')
+      @server.stub!(:exec)
+    end
+
+    it 'runs Solr with logging properties file' do
+      @server.should_receive(:exec).with(%r(-Djava\.util\.logging\.config\.file=/tmp/logging\.properties\.12345))
+      @server.run
+    end
+
+    it 'sets logging level' do
+      @server.run
+      @tempfile.string.should =~ /^java\.util\.logging\.FileHandler\.level *= *INFO$/
+    end
+
+    it 'sets handler' do
+      @server.run
+      @tempfile.string.should =~ /^handlers *= *java.util.logging.FileHandler$/
+    end
+
+    it 'sets formatter' do
+      @server.run
+      @tempfile.string.should =~ /^java\.util\.logging\.FileHandler\.formatter *= *java\.util\.logging\.SimpleFormatter$/
+    end
+
+    it 'sets log file' do
+      @server.run
+      @tempfile.string.should =~ /^java\.util\.logging\.FileHandler\.pattern *= *log\/sunspot-development\.log$/
+    end
+  end
+end

data/spec/api/session_proxy/class_sharding_session_proxy_spec.rb

@@ -0,0 +1,85 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot::SessionProxy::ClassShardingSessionProxy do
+  before do
+    @proxy = MockClassShardingSessionProxy.new(session)
+  end
+
+  [:index, :index!, :remove, :remove!].each do |method|
+    it "should delegate #{method} to appropriate shard" do
+      post = Post.new
+      photo = Photo.new
+      @proxy.post_session.should_receive(method).with([post])
+      @proxy.photo_session.should_receive(method).with([photo])
+      @proxy.send(method, post)
+      @proxy.send(method, photo)
+    end
+  end
+
+  [:remove_by_id, :remove_by_id!].each do |method|
+    it "should delegate #{method} to appropriate shard" do
+      @proxy.post_session.should_receive(method).with(Post, 1)
+      @proxy.photo_session.should_receive(method).with(Photo, 1)
+      @proxy.send(method, Post, 1)
+      @proxy.send(method, Photo, 1)
+    end
+  end
+
+  [:remove_all, :remove_all!].each do |method|
+    it "should delegate #{method} with argument to appropriate shard" do
+      @proxy.post_session.should_receive(method).with(Post)
+      @proxy.photo_session.should_receive(method).with(Photo)
+      @proxy.send(method, Post)
+      @proxy.send(method, Photo)
+    end
+
+    it "should delegate #{method} without argument to all shards" do
+      @proxy.post_session.should_receive(method)
+      @proxy.photo_session.should_receive(method)
+      @proxy.send(method)
+    end
+  end
+
+  [:commit, :commit_if_dirty, :commit_if_delete_dirty].each do |method|
+    it "should delegate #{method} to all sessions" do
+      [@proxy.post_session, @proxy.photo_session].each do |session|
+        session.should_receive(method)
+      end
+      @proxy.send(method)
+    end
+  end
+
+  it "should not support the :batch method" do
+    lambda { @proxy.batch }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+  end
+
+  it "should delegate new_search to search session, adding in shards parameter" do
+    search = @proxy.new_search(Post)
+    search.query[:shards].should ==
+      'http://photos.solr.local/solr,http://posts.solr.local/solr'
+  end
+
+  it "should delegate search to search session, adding in shards parameter" do
+    @proxy.search(Post)
+    connection.should have_last_search_with(
+      :shards => 'http://photos.solr.local/solr,http://posts.solr.local/solr'
+    )
+  end
+
+  [:dirty, :delete_dirty].each do |method|
+    it "should be dirty if any of the sessions are dirty" do
+      @proxy.post_session.stub!(:"#{method}?").and_return(true)
+      @proxy.should send("be_#{method}")
+    end
+
+    it "should not be dirty if none of the sessions are dirty" do
+      @proxy.should_not send("be_#{method}")
+    end
+  end
+
+  it "should raise a NotSupportedError when :config is called" do
+    lambda { @proxy.config }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+  end
+
+  it_should_behave_like 'session proxy'
+end

data/spec/api/session_proxy/id_sharding_session_proxy_spec.rb

@@ -0,0 +1,30 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot::SessionProxy::ShardingSessionProxy do
+  before do
+    search_session = Sunspot::Session.new
+    @sessions = Array.new(2) { Sunspot::Session.new }
+    @proxy = Sunspot::SessionProxy::IdShardingSessionProxy.new(search_session, @sessions)
+  end
+
+  [:index, :index!, :remove, :remove!].each do |method|
+    it "should delegate #{method} to appropriate shard" do
+      posts = [Post.new(:id => 2), Post.new(:id => 1)]
+      @proxy.sessions[0].should_receive(method).with([posts[0]])
+      @proxy.sessions[1].should_receive(method).with([posts[1]])
+      @proxy.send(method, posts[0])
+      @proxy.send(method, posts[1])
+    end
+  end
+
+  [:remove_by_id, :remove_by_id!].each do |method|
+    it "should delegate #{method} to appropriate session" do
+      @proxy.sessions[0].should_receive(method).with(Post, 2)
+      @proxy.sessions[1].should_receive(method).with(Post, 1)
+      @proxy.send(method, Post, 1)
+      @proxy.send(method, Post, 2)
+    end
+  end
+
+  it_should_behave_like 'session proxy'
+end

data/spec/api/session_proxy/master_slave_session_proxy_spec.rb

@@ -0,0 +1,41 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot::SessionProxy::MasterSlaveSessionProxy do
+  before :each do
+    @master_session, @slave_session = Sunspot::Session.new, Sunspot::Session.new
+    @proxy = Sunspot::SessionProxy::MasterSlaveSessionProxy.new(@master_session, @slave_session)
+  end
+
+  {
+    :master_session => Sunspot::Session.public_instance_methods(false) - [:search, 'search', :new_search, 'new_search', :more_like_this, 'more_like_this', :new_more_like_this, 'new_more_like_this', :config, 'config'],
+    :slave_session => [:search, :new_search, :more_like_this, :new_more_like_this]
+  }.each_pair do |delegate, methods|
+    methods.each do |method|
+      it "should delegate #{method} to #{delegate}" do
+        args = Array.new(Sunspot::Session.instance_method(method).arity.abs) do
+          stub('arg')
+        end
+        instance_variable_get(:"@#{delegate}").should_receive(method).with(*args)
+        @proxy.send(method, *args)
+      end
+    end
+  end
+
+  it 'should return master session config by default' do
+    @proxy.config.should eql(@master_session.config)
+  end
+
+  it 'should return master session config when specified' do
+    @proxy.config(:master).should eql(@master_session.config)
+  end
+
+  it 'should return slave session config when specified' do
+    @proxy.config(:slave).should eql(@slave_session.config)
+  end
+
+  it 'should raise ArgumentError when bogus config specified' do
+    lambda { @proxy.config(:bogus) }.should raise_error
+  end
+
+  it_should_behave_like 'session proxy'
+end

data/spec/api/session_proxy/sharding_session_proxy_spec.rb

@@ -0,0 +1,77 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot::SessionProxy::ShardingSessionProxy do
+  before do
+    @proxy = MockShardingSessionProxy.new(session)
+  end
+
+  [:index, :index!, :remove, :remove!].each do |method|
+    it "should delegate #{method} to appropriate shard" do
+      posts = [Post.new(:blog_id => 2), Post.new(:blog_id => 3)]
+      @proxy.sessions[0].should_receive(method).with([posts[0]])
+      @proxy.sessions[1].should_receive(method).with([posts[1]])
+      @proxy.send(method, posts[0])
+      @proxy.send(method, posts[1])
+    end
+  end
+
+  [:remove_by_id, :remove_by_id!].each do |method|
+    it "should raise NotSupportedError when #{method} called" do
+      lambda { @proxy.send(method, Post, 1) }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+    end
+  end
+
+  [:remove_all, :remove_all!].each do |method|
+    it "should raise NotSupportedError when #{method} called with argument" do
+      lambda { @proxy.send(method, Post) }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+    end
+
+    it "should delegate #{method} without argument to all shards" do
+      @proxy.sessions.each { |session| session.should_receive(method) }
+      @proxy.send(method)
+    end
+  end
+
+  [:commit, :commit_if_dirty, :commit_if_delete_dirty].each do |method|
+    it "should delegate #{method} to all sessions" do
+      @proxy.sessions.each do |session|
+        session.should_receive(method)
+      end
+      @proxy.send(method)
+    end
+  end
+
+  it "should not support the :batch method" do
+    lambda { @proxy.batch }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+  end
+
+  it "should delegate new_search to search session, adding in shards parameter" do
+    search = @proxy.new_search(Post)
+    search.query[:shards].should ==
+      'http://localhost:8980/solr,http://localhost:8981/solr'
+  end
+
+  it "should delegate search to search session, adding in shards parameter" do
+    @proxy.search(Post)
+    connection.should have_last_search_with(
+      :shards => 'http://localhost:8980/solr,http://localhost:8981/solr'
+    )
+  end
+
+  [:dirty, :delete_dirty].each do |method|
+    it "should be dirty if any of the sessions are dirty" do
+      @proxy.sessions[0].stub!(:"#{method}?").and_return(true)
+      @proxy.should send("be_#{method}")
+    end
+
+    it "should not be dirty if none of the sessions are dirty" do
+      @proxy.should_not send("be_#{method}")
+    end
+  end
+
+  it "should raise a NotSupportedError when :config is called" do
+    lambda { @proxy.config }.should raise_error(Sunspot::SessionProxy::NotSupportedError)
+  end
+
+  it_should_behave_like 'session proxy'
+end

data/spec/api/session_proxy/spec_helper.rb

@@ -0,0 +1,9 @@
+require File.join(File.dirname(__FILE__), '..', 'spec_helper')
+
+shared_examples_for 'session proxy' do
+  Sunspot::Session.public_instance_methods(false).each do |method|
+    it "should respond to #{method.inspect}" do
+      @proxy.should respond_to(method)
+    end
+  end
+end