ruben-sunspot 1.1.0

data/spec/api/session_proxy/thread_local_session_proxy_spec.rb

@@ -0,0 +1,50 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+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
+
+  it 'should garbage collect session instance when proxy dereferenced' do
+    ref = WeakRef.new(@proxy.session)
+    @proxy = nil
+    GC.start
+    # need to do this a second time since the reference to the session is
+    # destroyed in the finalizer during the first GC run, and thus isn't picked
+    # up by that run.
+    GC.start 
+    lambda { ref.inspect }.should raise_error(WeakRef::RefError)
+  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,198 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+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 '#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
+  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 delete_dirty after a commit' do
+      @session.remove(Post.new)
+      @session.commit
+      @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(File.join(File.dirname(__FILE__), '..', 'spec_helper'))
+
+Dir.glob(File.join(File.dirname(__FILE__), '**', '*_examples.rb')).each { |shared| require(shared) }

data/spec/api/sunspot_spec.rb

@@ -0,0 +1,18 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe Sunspot do
+  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,29 @@
+module IndexerHelper
+  def config
+    Sunspot::Configuration.build
+  end
+
+  def connection
+    @connection ||= Mock::Connection.new
+  end
+
+  def session
+    @session ||= Sunspot::Session.new(config, connection)
+  end
+
+  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/query_helper.rb

@@ -0,0 +1,38 @@
+module QueryHelper
+  def config
+    @config ||= Sunspot::Configuration.build
+  end
+
+  def connection
+    @connection ||= Mock::Connection.new
+  end
+
+  def session
+    @session ||= Sunspot::Session.new(config, connection)
+  end
+
+  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

data/spec/helpers/search_helper.rb

@@ -0,0 +1,80 @@
+module SearchHelper
+  def stub_nil_results
+    connection.response = { 'response' => nil }
+  end
+
+  def stub_full_results(*results)
+    count =
+      if results.last.is_a?(Integer) then results.pop
+      else results.length
+      end
+    docs = results.map do |result|
+      instance = result.delete('instance')
+      result.merge('id' => "#{instance.class.name} #{instance.id}")
+    end
+    response = {
+      'response' => {
+        'docs' => docs,
+        'numFound' => count
+      }
+    }
+    connection.response = response
+    response
+  end
+
+  def stub_results(*results)
+    stub_full_results(
+      *results.map do |result|
+        if result.is_a?(Integer)
+          result
+        else
+          { 'instance' => result }
+        end
+      end
+    )
+  end
+
+  def stub_facet(name, values)
+    connection.response = {
+      'facet_counts' => {
+        'facet_fields' => {
+          name.to_s => values.to_a.sort_by { |value, count| -count }.flatten
+        }
+      }
+    }
+  end
+
+  def stub_date_facet(name, gap, values)
+    connection.response = {
+      'facet_counts' => {
+        'facet_dates' => {
+          name.to_s => { 'gap' => "+#{gap}SECONDS" }.merge(values)
+        }
+      }
+    }
+  end
+
+  def stub_query_facet(values)
+    connection.response = { 'facet_counts' => { 'facet_queries' => values } }
+  end
+
+  def facet_values(result, field_name)
+    result.facet(field_name).rows.map { |row| row.value }
+  end
+
+  def facet_counts(result, field_name)
+    result.facet(field_name).rows.map { |row| row.count }
+  end
+
+  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/integration/dynamic_fields_spec.rb

@@ -0,0 +1,55 @@
+describe 'dynamic fields' do
+  before :each do
+    Sunspot.remove_all
+    @posts = Post.new(:custom_string => { :cuisine => 'Pizza' }),
+             Post.new(:custom_string => { :cuisine => 'Greek' }),
+             Post.new(:custom_string => { :cuisine => 'Greek' })
+    Sunspot.index!(@posts)
+  end
+
+  it 'should search for dynamic string field' do
+    Sunspot.search(Post) do
+      dynamic(:custom_string) do
+        with(:cuisine, 'Pizza')
+      end
+    end.results.should == [@posts.first]
+  end
+
+  describe 'faceting' do
+    before :each do
+      @search = Sunspot.search(Post) do
+        dynamic :custom_string do
+          facet :cuisine
+        end
+      end
+    end
+
+    it 'should return value "value" with count 2' do
+      row = @search.dynamic_facet(:custom_string, :cuisine).rows[0]
+      row.value.should == 'Greek'
+      row.count.should == 2
+    end
+
+    it 'should return value "other" with count 1' do
+      row = @search.dynamic_facet(:custom_string, :cuisine).rows[1]
+      row.value.should == 'Pizza'
+      row.count.should == 1
+    end
+  end
+
+  it 'should order by dynamic string field ascending' do
+    Sunspot.search(Post) do
+      dynamic :custom_string do
+        order_by :cuisine, :asc
+      end
+    end.results.last.should == @posts.first
+  end
+
+  it 'should order by dynamic string field descending' do
+    Sunspot.search(Post) do
+      dynamic :custom_string do
+        order_by :cuisine, :desc
+      end
+    end.results.first.should == @posts.first
+  end
+end