sunspot 0.9.7

data/spec/integration/spec_helper.rb

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

data/spec/integration/stored_fields_spec.rb

@@ -0,0 +1,10 @@
+describe 'stored fields' do
+  before :all do
+    Sunspot.remove_all
+    Sunspot.index!(Post.new(:title => 'A Title'))
+  end
+
+  it 'should return stored fields' do
+    Sunspot.search(Post).hits.first.stored(:title).should == 'A Title'
+  end
+end

data/spec/integration/test_pagination.rb

@@ -0,0 +1,32 @@
+require File.join(File.dirname(__FILE__), 'spec_helper')
+
+describe 'pagination' do
+  before :all do
+    Sunspot.remove_all
+    @posts = (0..19).map do |i|
+      Post.new(:blog_id => i)
+    end
+    Sunspot.index(*@posts)
+  end
+
+  it 'should return all by default' do
+    results = Sunspot.search(Post) { order_by :blog_id }.results
+    results.should == @posts
+  end
+
+  it 'should return first page of 10' do
+    results = Sunspot.search(Post) do
+      order_by :blog_id
+      paginate :page => 1, :per_page => 10
+    end.results
+    results.should == @posts[0,10]
+  end
+
+  it 'should return second page of 10' do
+    results = Sunspot.search(Post) do
+      order_by :blog_id
+      paginate :page => 2, :per_page => 10
+    end.results
+    results.should == @posts[10,10]
+  end
+end

data/spec/mocks/adapters.rb

@@ -0,0 +1,32 @@
+class AbstractModel
+end
+
+class Model < AbstractModel
+end
+
+class AbstractModelInstanceAdapter < Sunspot::Adapters::InstanceAdapter
+end
+
+class AbstractModelDataAccessor < Sunspot::Adapters::DataAccessor
+end
+
+Sunspot::Adapters::InstanceAdapter.register(AbstractModelInstanceAdapter, AbstractModel)
+Sunspot::Adapters::DataAccessor.register(AbstractModelDataAccessor, AbstractModel)
+
+
+module MixInModel
+end
+
+class MixModel
+  include MixInModel
+end
+
+class MixInModelInstanceAdapter < Sunspot::Adapters::InstanceAdapter
+end
+
+class MixInModelDataAccessor < Sunspot::Adapters::DataAccessor
+end
+
+Sunspot::Adapters::InstanceAdapter.register(MixInModelInstanceAdapter, MixInModel)
+Sunspot::Adapters::DataAccessor.register(MixInModelDataAccessor, MixInModel)
+

data/spec/mocks/blog.rb

@@ -0,0 +1,3 @@
+class Blog < MockRecord
+  attr_accessor :name
+end

data/spec/mocks/comment.rb

@@ -0,0 +1,19 @@
+module Namespaced
+  class Comment < MockRecord
+    attr_reader :id
+    attr_accessor :author_name, :published_at, :body, :average_rating, :boost
+
+    def custom_string
+      @custom_string ||= {}
+    end
+  end
+end
+
+Sunspot.setup(Namespaced::Comment) do
+  text :body, :author_name
+  string :author_name
+  time :published_at
+  integer :average_rating
+  dynamic_string :custom_string
+  boost :boost
+end

data/spec/mocks/connection.rb

@@ -0,0 +1,84 @@
+module Mock
+  class ConnectionFactory
+    def new(adapter = nil, opts = nil)
+      if @instance
+        raise('Factory can only create an instance once!')
+      else
+        @instance = Connection.new(adapter, opts)
+      end
+    end
+
+    def instance
+      @instance ||= Connection.new
+    end
+  end
+
+  class Connection
+    attr_reader :adds, :commits, :searches
+    attr_accessor :adapter, :opts
+
+    def initialize(adapter = nil, opts = nil)
+      @adapter, @opts = adapter, opts
+      @adds, @deletes, @deletes_by_query, @commits, @searches = Array.new(5) { [] }
+    end
+
+    def add(documents)
+      @adds << Array(documents)
+    end
+
+    def delete_by_id(*ids)
+      @deletes << ids
+    end
+
+    def delete_by_query(query)
+      @deletes_by_query << query
+    end
+
+    def commit
+      @commits << Time.now
+    end
+
+    def select(params)
+      @searches << @last_search = params
+    end
+
+    def has_add_with?(*documents)
+      @adds.any? do |add|
+        documents.all? do |document|
+          add.any? do |added|
+            if document.is_a?(Hash)
+              document.all? do |field, value|
+                added.fields_by_name(field).map do |field|
+                  field.value
+                end == Array(value)
+              end
+            else
+              !added.fields_by_name(document).empty?
+            end
+          end
+        end
+      end
+    end
+
+    def has_delete?(*ids)
+      @deletes.any? do |delete|
+        delete & ids == ids
+      end
+    end
+
+    def has_delete_by_query?(query)
+      @deletes_by_query.include?(query)
+    end
+
+    def has_last_search_with?(params)
+      return unless @last_search
+      if params.respond_to?(:all?)
+        params.all? do |key, value|
+          @last_search.has_key?(key) && @last_search[key] == value
+        end
+      else
+        @last_search.has_key?(params)
+      end
+    end
+  end
+end

data/spec/mocks/mock_adapter.rb

@@ -0,0 +1,30 @@
+require File.join(File.dirname(__FILE__), 'post')
+
+module MockAdapter
+  class InstanceAdapter < Sunspot::Adapters::InstanceAdapter
+    def id
+      @instance.id
+    end
+  end
+
+  class DataAccessor < Sunspot::Adapters::DataAccessor
+    def load(id)
+      @clazz.get(id.to_i)
+    end
+
+    def load_all(ids)
+      all = @clazz.get_all(ids.map { |id| id.to_i })
+      if @custom_title
+        all.each { |item| item.title = @custom_title }
+      end
+      all
+    end
+
+    def custom_title=(custom_title)
+      @custom_title = custom_title
+    end
+  end
+end
+
+Sunspot::Adapters::DataAccessor.register(MockAdapter::DataAccessor, MockRecord)
+Sunspot::Adapters::InstanceAdapter.register(MockAdapter::InstanceAdapter, MockRecord)

data/spec/mocks/mock_record.rb

@@ -0,0 +1,48 @@
+class MockRecord
+  IDS = Hash.new { |h, k| h[k] = 0 }
+  QUERY_COUNTS = Hash.new { |h, k| h[k] = 0 }
+  INSTANCES = Hash.new { |h, k| h[k] = {} }
+
+  attr_reader :id
+
+  class <<self
+    def reset!
+      IDS[name.to_sym] = 0
+      INSTANCES[name.to_sym] = {}
+    end
+  end
+
+  def initialize(attrs = {})
+    attrs.each_pair do |name, value|
+      send(:"#{name}=", value)
+    end
+    @id = IDS[self.class.name.to_sym] += 1
+    INSTANCES[self.class.name.to_sym][@id] = self
+  end
+
+  def self.inherited(base)
+    base.extend(ClassMethods)
+  end
+
+  module ClassMethods
+    def get(id)
+      QUERY_COUNTS[self.name.to_sym] += 1
+      get_instance(id)
+    end
+
+    def get_all(ids)
+      QUERY_COUNTS[self.name.to_sym] += 1
+      ids.map { |id| get_instance(id) }.sort_by { |instance| instance.id }
+    end
+
+    def query_count
+      QUERY_COUNTS[self.name.to_sym]
+    end
+
+    private
+
+    def get_instance(id)
+      INSTANCES[self.name.to_sym][id]
+    end
+  end
+end

data/spec/mocks/photo.rb

@@ -0,0 +1,8 @@
+class Photo < MockRecord
+  attr_accessor :caption
+end
+
+Sunspot.setup(Photo) do
+  text :caption
+  boost 0.75
+end

data/spec/mocks/post.rb

@@ -0,0 +1,73 @@
+require File.join(File.dirname(__FILE__), 'blog')
+
+class Post < MockRecord
+  attr_accessor :title, :body, :blog_id, :published_at, :ratings_average,
+                :author_name, :featured, :expire_date
+  alias_method :featured?, :featured
+
+  def category_ids
+    @category_ids ||= []
+  end
+
+  def custom_string
+    @custom_string ||= {}
+  end
+
+  def custom_fl
+    @custom_fl ||= {}
+  end
+
+  def custom_time
+    @custom_time ||= {}
+  end
+
+  def custom_boolean
+    @custom_boolean ||= {}
+  end
+
+  private
+  attr_writer :category_ids, :custom_string, :custom_fl, :custom_time, :custom_boolean
+end
+
+Sunspot.setup(Post) do
+  text :title, :boost => 2
+  text :body
+  text :backwards_title do
+    title.reverse if title
+  end
+  string :title, :stored => true
+  integer :blog_id, :references => Blog
+  integer :category_ids, :multiple => true
+  float :average_rating, :using => :ratings_average
+  time :published_at
+  date :expire_date
+  boolean :featured, :using => :featured?
+  string :sort_title do
+    title.downcase.sub(/^(a|an|the)\W+/, '') if title
+  end
+  integer :primary_category_id do |post|
+    post.category_ids.first
+  end
+  time :last_indexed_at, :stored => true do
+    Time.now
+  end
+
+  dynamic_string :custom_string
+  dynamic_float :custom_float, :multiple => true, :using => :custom_fl
+  dynamic_integer :custom_integer do
+    category_ids.inject({}) do |hash, category_id|
+      hash.merge(category_id => 1)
+    end
+  end
+  dynamic_time :custom_time
+  dynamic_boolean :custom_boolean
+
+  boost do
+    if ratings_average
+      1 + (ratings_average - 3.0) / 4.0
+    end
+  end
+end
+
+class PhotoPost < Post
+end

data/spec/mocks/user.rb

@@ -0,0 +1,8 @@
+class User
+  attr_accessor :name
+end
+
+Sunspot.setup(User) do
+  text :name
+  string :name
+end

data/spec/spec_helper.rb

@@ -0,0 +1,47 @@
+using_gems = false
+begin
+  require 'spec'
+  begin
+    require 'ruby-debug'
+  rescue LoadError => e
+    if using_gems
+      module Kernel
+        def debugger
+          STDERR.puts('Debugger is not available')
+        end
+      end
+    else
+      raise(e)
+    end
+  end
+  if ENV['USE_WILL_PAGINATE']
+    require 'will_paginate'
+    require 'will_paginate/collection'
+  end
+rescue LoadError => e
+  require 'rubygems'
+  if using_gems
+    raise(e)
+  else
+    using_gems = true
+    retry
+  end
+end
+
+unless gem_name = ENV['SUNSPOT_TEST_GEM']
+  $:.unshift(File.dirname(__FILE__) + '/../lib')
+else
+  gem gem_name
+end
+require 'sunspot'
+
+require File.join(File.dirname(__FILE__), 'mocks', 'mock_record.rb')
+Dir.glob(File.join(File.dirname(__FILE__), 'mocks', '**', '*.rb')).each do |file|
+  require file unless File.basename(file) == 'mock_record.rb'
+end
+
+def without_class(clazz)
+  Object.class_eval { remove_const(clazz.name.to_sym) }
+  yield
+  Object.class_eval { const_set(clazz.name.to_sym, clazz) }
+end

data/tasks/gemspec.rake

@@ -0,0 +1,25 @@
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |s|
+    s.name = 'sunspot'
+    s.executables = ['sunspot-solr', 'sunspot-configure-solr']
+    s.summary = 'Library for expressive, powerful interaction with the Solr search engine'
+    s.email = 'mat@patch.com'
+    s.homepage = 'http://github.com/outoftime/sunspot'
+    s.description = 'Library for expressive, powerful interaction with the Solr search engine'
+    s.authors = ['Mat Brown', 'Peer Allan', 'Dmitriy Dzema', 'Benjamin Krause']
+    s.rubyforge_project = 'sunspot'
+    s.files = FileList['[A-Z]*', '{bin,lib,spec,tasks,templates}/**/*', 'solr/{etc,lib,webapps}/**/*', 'solr/solr/conf/*', 'solr/start.jar']
+    s.add_dependency 'mwmitchell-rsolr', '>= 0.8.9'
+    s.add_dependency 'daemons', '~> 1.0'
+    s.add_dependency 'optiflag', '~> 0.6.5'
+    s.add_dependency 'haml', '~> 2.2'
+    s.add_development_dependency 'rspec', '~> 1.1'
+    s.add_development_dependency 'ruby-debug', '~> 0.10'
+    s.extra_rdoc_files = ['README.rdoc']
+    s.rdoc_options << '--webcvs=http://github.com/outoftime/sunspot/tree/master/%s' <<
+                      '--title' << 'Sunspot - Solr-powered search for Ruby objects - API Documentation' <<
+                      '--main' << 'README.rdoc'
+
+  end
+end

data/tasks/rcov.rake

@@ -0,0 +1,28 @@
+require 'rake'
+require 'spec/rake/spectask'
+
+desc 'run specs with rcov'
+Spec::Rake::SpecTask.new('rcov') do |t|
+  t.spec_files = FileList['spec/**/*_spec.rb']
+  t.rcov = true
+  t.rcov_dir = File.join('coverage', 'all')
+  t.rcov_opts.concat(['--exclude', 'spec', '--sort', 'coverage', '--only-uncovered'])
+end
+
+namespace :rcov do
+  desc 'run api specs with rcov'
+  Spec::Rake::SpecTask.new('api') do |t|
+    t.spec_files = FileList['spec/api/*_spec.rb']
+    t.rcov = true
+    t.rcov_dir = File.join('coverage', 'api')
+    t.rcov_opts.concat(['--exclude', 'spec', '--sort', 'coverage', '--only-uncovered'])
+  end
+
+  desc 'run integration specs with rcov'
+  Spec::Rake::SpecTask.new('integration') do |t|
+    t.spec_files = FileList['spec/integration/*_spec.rb']
+    t.rcov = true
+    t.rcov_dir = File.join('coverage', 'integration')
+    t.rcov_opts.concat(['--exclude', 'spec', '--sort', 'coverage', '--only-uncovered'])
+  end
+end