gojee-sunspot 2.0.3 → 2.0.4

data/spec/api/session_proxy/class_sharding_session_proxy_spec.rb

@@ -0,0 +1,85 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+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, :optimize].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.expand_path('spec_helper', File.dirname(__FILE__))
+
+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.expand_path('spec_helper', File.dirname(__FILE__))
+
+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.expand_path('spec_helper', File.dirname(__FILE__))
+
+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, :optimize].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/silent_fail_session_proxy_spec.rb

@@ -0,0 +1,24 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe Sunspot::SessionProxy::ShardingSessionProxy do
+  
+  FakeException = Class.new(StandardError)
+  SUPPORTED_METHODS = Sunspot::SessionProxy::SilentFailSessionProxy::SUPPORTED_METHODS
+
+  before do
+    @search_session = mock(Sunspot::Session.new)
+    @proxy = Sunspot::SessionProxy::SilentFailSessionProxy.new(@search_session)
+  end
+  
+  it "should call rescued_exception when an exception is caught" do
+    SUPPORTED_METHODS.each do |method|
+      e = FakeException.new(method)
+      @search_session.stub!(method).and_raise(e)
+      @proxy.should_receive(:rescued_exception).with(method, e)
+      @proxy.send(method)
+    end
+  end
+  
+  it_should_behave_like 'session proxy'
+  
+end

data/spec/api/session_proxy/spec_helper.rb

@@ -0,0 +1,9 @@
+require File.expand_path('spec_helper', File.join(File.dirname(__FILE__), '..'))
+
+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

data/spec/api/session_proxy/thread_local_session_proxy_spec.rb

@@ -0,0 +1,39 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+require 'weakref'
+
+describe Sunspot::SessionProxy::ThreadLocalSessionProxy do
+  before :each do
+    @config = Sunspot::Configuration.build
+    @proxy = Sunspot::SessionProxy::ThreadLocalSessionProxy.new(@config)
+  end
+
+  it 'should have the same session for the same thread' do
+    @proxy.session.should eql(@proxy.session)
+  end
+
+  it 'should not have the same session for different threads' do
+    session1 = @proxy.session
+    session2 = nil
+    Thread.new do
+      session2 = @proxy.session
+    end.join
+    session1.should_not eql(session2)
+  end
+
+  it 'should not have the same session for the same thread in different proxy instances' do
+    proxy2 = Sunspot::SessionProxy::ThreadLocalSessionProxy.new(@config)
+    @proxy.session.should_not eql(proxy2.session)
+  end
+
+  (Sunspot::Session.public_instance_methods(false) - ['config', :config]).each do |method|
+    it "should delegate #{method.inspect} to its session" do
+      args = Array.new(Sunspot::Session.instance_method(method).arity.abs) do
+        stub('arg')
+      end
+      @proxy.session.should_receive(method).with(*args)
+      @proxy.send(method, *args)
+    end
+  end
+
+  it_should_behave_like 'session proxy'
+end

data/spec/api/session_spec.rb

@@ -0,0 +1,232 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+shared_examples_for 'all sessions' do
+  context '#index()' do
+    before :each do
+      @session.index(Post.new)
+    end
+
+    it 'should add document to connection' do
+      connection.should have(1).adds
+    end
+  end
+
+  context '#index!()' do
+    before :each do
+      @session.index!(Post.new)
+    end
+
+    it 'should add document to connection' do
+      connection.should have(1).adds
+    end
+
+    it 'should commit' do
+      connection.should have(1).commits
+    end
+  end
+
+  context '#commit()' do
+    before :each do
+      @session.commit
+    end
+
+    it 'should commit' do
+      connection.should have(1).commits
+    end
+  end
+
+  context '#optimize()' do
+    before :each do
+      @session.optimize
+    end
+
+    it 'should optimize' do
+      connection.should have(1).optims
+    end
+  end
+
+  context '#search()' do
+    before :each do
+      @session.search(Post)
+    end
+
+    it 'should search' do
+      connection.should have(1).searches
+    end
+  end
+end
+
+describe 'Session' do
+  before :each do
+    @connection_factory = Mock::ConnectionFactory.new
+    Sunspot::Session.connection_class = @connection_factory
+  end
+
+  after :each do
+    Sunspot::Session.connection_class = nil
+    Sunspot.reset!
+  end
+
+  context 'singleton session' do
+    before :each do
+      Sunspot.reset!
+      @session = Sunspot
+    end
+
+    it_should_behave_like 'all sessions'
+
+    it 'should open connection with defaults if nothing specified' do
+      Sunspot.commit
+      connection.opts[:url].should == 'http://127.0.0.1:8983/solr'
+    end
+
+    it 'should open a connection with custom host' do
+      Sunspot.config.solr.url = 'http://127.0.0.1:8981/solr'
+      Sunspot.commit
+      connection.opts[:url].should == 'http://127.0.0.1:8981/solr'
+    end
+
+    it 'should open a connection with custom read timeout' do
+      Sunspot.config.solr.read_timeout = 0.5
+      Sunspot.commit
+      connection.opts[:read_timeout].should == 0.5
+    end
+
+    it 'should open a connection with custom open timeout' do
+      Sunspot.config.solr.open_timeout = 0.5
+      Sunspot.commit
+      connection.opts[:open_timeout].should == 0.5
+    end
+  end
+
+  context 'custom session' do
+    before :each do
+      @session = Sunspot::Session.new
+    end
+
+    it_should_behave_like 'all sessions'
+
+    it 'should open a connection with custom host' do
+      session = Sunspot::Session.new do |config|
+        config.solr.url = 'http://127.0.0.1:8982/solr'
+      end
+      session.commit
+      connection.opts[:url].should == 'http://127.0.0.1:8982/solr'
+    end
+  end
+
+  context 'dirty sessions' do
+    before :each do
+      @session = Sunspot::Session.new
+    end
+
+    it 'should start out not dirty' do
+      @session.dirty?.should be_false
+    end
+    
+    it 'should start out not delete_dirty' do
+      @session.delete_dirty?.should be_false
+    end
+
+    it 'should be dirty after adding an item' do
+      @session.index(Post.new)
+      @session.dirty?.should be_true
+    end
+    
+    it 'should be not be delete_dirty after adding an item' do
+      @session.index(Post.new)
+      @session.delete_dirty?.should be_false
+    end
+
+    it 'should be dirty after deleting an item' do
+      @session.remove(Post.new)
+      @session.dirty?.should be_true
+    end
+
+    it 'should be delete_dirty after deleting an item' do
+      @session.remove(Post.new)
+      @session.delete_dirty?.should be_true
+    end
+
+    it 'should be dirty after a remove_all for a class' do
+      @session.remove_all(Post)
+      @session.dirty?.should be_true
+    end
+
+    it 'should be delete_dirty after a remove_all for a class' do
+      @session.remove_all(Post)
+      @session.delete_dirty?.should be_true
+    end
+
+    it 'should be dirty after a global remove_all' do
+      @session.remove_all
+      @session.dirty?.should be_true
+    end
+    
+    it 'should be delete_dirty after a global remove_all' do
+      @session.remove_all
+      @session.delete_dirty?.should be_true
+    end
+    
+    it 'should not be dirty after a commit' do
+      @session.index(Post.new)
+      @session.commit
+      @session.dirty?.should be_false
+    end
+
+    it 'should not be dirty after an optimize' do
+      @session.index(Post.new)
+      @session.optimize
+      @session.dirty?.should be_false
+    end
+
+    it 'should not be delete_dirty after a commit' do
+      @session.remove(Post.new)
+      @session.commit
+      @session.delete_dirty?.should be_false
+    end
+
+    it 'should not be delete_dirty after an optimize' do
+      @session.remove(Post.new)
+      @session.optimize
+      @session.delete_dirty?.should be_false
+    end
+
+    it 'should not commit when commit_if_dirty called on clean session' do
+      @session.commit_if_dirty
+      connection.should have(0).commits
+    end
+
+    it 'should not commit when commit_if_delete_dirty called on clean session' do
+      @session.commit_if_delete_dirty
+      connection.should have(0).commits
+    end
+
+    it 'should commit when commit_if_dirty called on dirty session' do
+      @session.index(Post.new)
+      @session.commit_if_dirty
+      connection.should have(1).commits
+    end
+    
+    it 'should commit when commit_if_delete_dirty called on delete_dirty session' do
+      @session.remove(Post.new)
+      @session.commit_if_delete_dirty
+      connection.should have(1).commits
+    end
+  end
+
+  context 'session proxy' do
+    it 'should send messages to manually assigned session proxy' do
+      stub_session = stub!('session')
+      Sunspot.session = stub_session
+      post = Post.new
+      stub_session.should_receive(:index).with(post)
+      Sunspot.index(post)
+      Sunspot.reset!
+    end
+  end
+
+  def connection
+    @connection_factory.instance
+  end
+end

data/spec/api/spec_helper.rb

@@ -0,0 +1,3 @@
+require File.expand_path('spec_helper', File.join(File.dirname(__FILE__), '..'))
+
+Dir.glob(File.join(File.dirname(__FILE__), '**', '*_examples.rb')).each { |shared| require(shared) }

data/spec/api/sunspot_spec.rb

@@ -0,0 +1,29 @@
+require File.expand_path('spec_helper', File.dirname(__FILE__))
+
+describe Sunspot do
+
+  describe "setup" do
+    it "should register the class in Sunspot.searchable" do
+      Sunspot.setup(Blog) do
+        text :name
+      end
+      Sunspot.searchable.should_not be_empty
+      Sunspot.searchable.should include(Blog)
+    end
+  end
+
+  describe "reset!" do
+    it "should reset current session" do
+      old_session = Sunspot.send(:session)
+      Sunspot.reset!(true)
+      Sunspot.send(:session).should_not == old_session
+    end
+
+    it "should keep keep configuration if specified" do
+      Sunspot.config.solr.url = "http://localhost:9999/path/solr"
+      config_before_reset = Sunspot.config
+      Sunspot.reset!(true)
+      Sunspot.config.should == config_before_reset
+    end
+  end
+end

data/spec/ext.rb

@@ -0,0 +1,11 @@
+module Solr
+  class Document
+    def field_by_name(field_name)
+      @fields.find { |field| field.name.to_s == field_name.to_s }
+    end
+
+    def fields_by_name(field_name)
+      @fields.select { |field| field.name.to_s == field_name.to_s }
+    end
+  end
+end

data/spec/helpers/indexer_helper.rb

@@ -0,0 +1,17 @@
+module IndexerHelper
+  def post(attrs = {})
+    @post ||= Post.new(attrs)
+  end
+
+  def last_add
+    @connection.adds.last
+  end
+
+  def value_in_last_document_for(field_name)
+    @connection.adds.last.last.field_by_name(field_name).value
+  end
+
+  def values_in_last_document_for(field_name)
+    @connection.adds.last.last.fields_by_name(field_name).map { |field| field.value }
+  end
+end

data/spec/helpers/integration_helper.rb

@@ -0,0 +1,8 @@
+module IntegrationHelper
+  def self.included(base)
+    base.before(:all) do
+      Sunspot.config.solr.url = ENV['SOLR_URL'] || 'http://localhost:8983/solr'
+      Sunspot.reset!(true)
+    end
+  end
+end

data/spec/helpers/mock_session_helper.rb

@@ -0,0 +1,13 @@
+module MockSessionHelper
+  def config
+    @config ||= Sunspot::Configuration.build
+  end
+
+  def connection
+    @connection ||= Mock::Connection.new
+  end
+
+  def session
+    @session ||= Sunspot::Session.new(config, connection)
+  end
+end

data/spec/helpers/query_helper.rb

@@ -0,0 +1,26 @@
+module QueryHelper
+  def get_filter_tag(boolean_query)
+    connection.searches.last[:fq].each do |fq|
+      if match = fq.match(/^\{!tag=(.+)\}#{Regexp.escape(boolean_query)}$/)
+        return match[1]
+      end
+    end
+    nil
+  end
+
+  def subqueries(param)
+    q = connection.searches.last[:q]
+    subqueries = []
+    subqueries = q.scan(%r(_query_:"\{!dismax (.*?)\}(.*?)"))
+    subqueries.map do |subquery|
+      params = {}
+      subquery[0].scan(%r((\S+?)='(.+?)')) do |key, value|
+        params[key.to_sym] = value
+      end
+      unless subquery[1].empty?
+        params[:v] = subquery[1]
+      end
+      params
+    end
+  end
+end