sunspot_suggest 0.0.1

data/spec/mocks/adapters.rb

@@ -0,0 +1,36 @@
+class AbstractModel
+end
+
+class Model < AbstractModel
+end
+
+class UnseenModel < AbstractModel
+end
+
+class AbstractModelInstanceAdapter < Sunspot::Adapters::InstanceAdapter
+end
+
+class AbstractModelDataAccessor < Sunspot::Adapters::DataAccessor
+  attr_accessor :to_be_injected
+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,21 @@
+module Namespaced
+  class Comment < MockRecord
+    attr_reader :id
+    attr_accessor :author_name, :published_at, :body, :average_rating, :boost,
+                  :hash
+
+    def custom_float
+      @custom_float ||= {}
+    end
+  end
+end
+
+Sunspot.setup(Namespaced::Comment) do
+  text :body, :author_name
+  string :author_name
+  time :published_at, :trie => true
+  long :hash
+  double :average_rating
+  dynamic_float :custom_float, :multiple => true
+  boost :boost
+end

data/spec/mocks/connection.rb

@@ -0,0 +1,128 @@
+module Mock
+  class ConnectionFactory
+    def connect(opts)
+      if @instance
+        raise('Factory can only create an instance once!')
+      else
+        @instance = Connection.new(opts)
+      end
+    end
+
+    def new(url = nil)
+      if @instance
+        raise('Factory can only create an instance once!')
+      else
+        @instance = Connection.new(url)
+      end
+    end
+
+    def instance
+      @instance ||= Connection.new
+    end
+  end
+
+  class Connection
+    attr_reader :adds, :commits, :optims, :searches, :message, :opts, :deletes_by_query
+    attr_accessor :response
+    attr_writer :expected_handler
+    attr_reader :last_search
+
+    undef_method :select # annoyingly defined on Object
+
+    def initialize(opts = {})
+      @opts = opts
+      @message = OpenStruct.new
+      @adds, @deletes, @deletes_by_query, @commits, @optims, @searches = Array.new(6) { [] }
+      @expected_handler = :select
+    end
+
+    def add(documents)
+      @adds << Array(documents)
+    end
+
+    def delete_by_id(ids)
+      @deletes << Array(ids)
+    end
+
+    def delete_by_query(query)
+      @deletes_by_query << query
+    end
+
+    def commit
+      @commits << Time.now
+    end
+
+    def optimize
+      @optims << Time.now
+    end
+
+    def post(path, params)
+      unless path == "#{@expected_handler}"
+        raise ArgumentError, "Expected request to #{@expected_handler} request handler"
+      end
+      @searches << @last_search = params[:data]
+      @response || {}
+    end
+
+    def method_missing(method, *args, &block)
+      get("#{method}", *args)
+    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.to_s
+                end == Array(value).map { |v| v.to_s }
+              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)
+      with?(@last_search, params) if @last_search
+    end
+
+    def has_last_search_including?(key, *values)
+      return unless @last_search
+      if @last_search.has_key?(key)
+        if @last_search[key].is_a?(Array)
+          (@last_search[key] & values).length == values.length
+        elsif values.length == 1
+          @last_search[key] == values.first
+        end
+      end
+    end
+
+    private
+
+    def with?(request, params)
+      if params.respond_to?(:all?)
+        params.all? do |key, value|
+          if request.has_key?(key)
+            request[key] == value
+          end
+        end
+      else
+        request.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_class_sharding_session_proxy.rb

@@ -0,0 +1,24 @@
+class MockClassShardingSessionProxy < Sunspot::SessionProxy::ClassShardingSessionProxy
+  attr_reader :post_session, :photo_session
+
+  def initialize(search_session)
+    super
+    @post_session, @photo_session = Sunspot::Session.new, Sunspot::Session.new
+    @post_session.config.solr.url = 'http://posts.solr.local/solr'
+    @photo_session.config.solr.url = 'http://photos.solr.local/solr'
+    @sessions = {
+      Post => @post_session,
+      Photo => @photo_session
+    }
+  end
+
+  def session_for_class(clazz)
+    @sessions[clazz]
+  end
+
+  def all_sessions
+    @sessions.values.sort do |lsession, rsession|
+      lsession.config.solr.url <=> rsession.config.solr.url
+    end
+  end
+end

data/spec/mocks/mock_record.rb

@@ -0,0 +1,52 @@
+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 = {})
+    @id = attrs.delete(:id) || IDS[self.class.name.to_sym] += 1
+    INSTANCES[self.class.name.to_sym][@id] = self
+    attrs.each_pair do |name, value|
+      send(:"#{name}=", value)
+    end
+  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) }.compact.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
+
+  def destroy
+    INSTANCES[self.class.name.to_sym].delete(@id)
+  end
+end

data/spec/mocks/mock_sharding_session_proxy.rb

@@ -0,0 +1,15 @@
+class MockShardingSessionProxy < Sunspot::SessionProxy::ShardingSessionProxy
+  attr_reader :sessions
+  alias_method :all_sessions, :sessions
+
+  def initialize(search_session)
+    super
+    @sessions = Array.new(2) { Sunspot::Session.new }.each_with_index do |session, i|
+      session.config.solr.url = "http://localhost:898#{i}/solr"
+    end
+  end
+
+  def session_for(object)
+    @sessions[object.blog_id.to_i % 2]
+  end
+end

data/spec/mocks/photo.rb

@@ -0,0 +1,11 @@
+class Photo < MockRecord
+  attr_accessor :caption, :lat, :lng, :size, :average_rating, :created_at
+end
+
+Sunspot.setup(Photo) do
+  text :caption, :default_boost => 1.5
+  boost 0.75
+  integer :size, :trie => true
+  float :average_rating, :trie => true
+  time :created_at, :trie => true
+end

data/spec/mocks/post.rb

@@ -0,0 +1,86 @@
+require File.join(File.dirname(__FILE__), 'blog')
+require File.join(File.dirname(__FILE__), 'super_class')
+
+class Post < SuperClass
+  attr_accessor :title, :body, :blog_id, :published_at, :ratings_average,
+                :author_name, :featured, :expire_date, :coordinates, :tags
+  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, :stored => true, :more_like_this => true
+  text :backwards_title do
+    title.reverse if title
+  end
+  text :tags, :more_like_this => true
+  string :title, :stored => true
+  integer :blog_id, :references => Blog
+  integer :category_ids, :multiple => true
+  float :average_rating, :using => :ratings_average, :trie => true
+  time :published_at, :trie => true
+  date :expire_date
+  boolean :featured, :using => :featured?, :stored => true
+  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
+  location :coordinates
+  latlon(:coordinates_new) { coordinates }
+
+  dynamic_string :custom_string, :stored => true
+  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
+
+  string :legacy, :as => :legacy_field_s do
+    "legacy #{title}"
+  end
+
+  string :legacy_array, :as => :legacy_array_field_sm, :multiple => true do
+    ['first string', 'second string']
+ 	end
+end
+
+class PhotoPost < Post
+end
+

data/spec/mocks/super_class.rb

@@ -0,0 +1,2 @@
+class SuperClass < MockRecord
+end

data/spec/mocks/user.rb

@@ -0,0 +1,13 @@
+class User
+  attr_accessor :name, :roles
+
+  def id
+    1
+  end
+end
+
+Sunspot.setup(User) do
+  text :name
+  string :name
+  integer :role_ids, :multiple => true, :stored => true
+end

data/spec/spec_helper.rb

@@ -0,0 +1,50 @@
+# encoding: utf-8
+
+require 'pry'
+require 'ostruct'
+require 'forwardable'
+require 'sunspot'
+require 'sunspot_suggest'
+
+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
+Dir.glob(File.join(File.dirname(__FILE__), "helpers", "*.rb")).each do |file|
+  require file
+end
+
+require_relative 'article'
+
+RSpec.configure do |config|
+  config.include SpellcheckHelper
+
+
+  # config.filter_run :focus => true
+
+  # Mock session available to all spec/api tests
+  config.include MockSessionHelper,
+                 :type => :api,
+                 :example_group => {:file_path => /spec[\\\/]api/}
+
+  # Real Solr instance is available to integration tests
+  config.include IntegrationHelper,
+                 :type => :integration,
+                 :example_group => {:file_path => /spec[\\\/]integration/}
+
+  # Nested under spec/api
+  [:indexer, :query, :search].each do |spec_type|
+    helper_name = "#{spec_type}_helper"
+
+    config.include Sunspot::Util.full_const_get(Sunspot::Util.camel_case(helper_name)),
+                   :type          => spec_type,
+                   :example_group => {:file_path => /spec[\\\/]api[\\\/]#{spec_type}/}
+  end
+
+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/sunspot_suggest.gemspec

@@ -0,0 +1,26 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'sunspot_suggest/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "sunspot_suggest"
+  spec.version       = SunspotSuggest::VERSION
+  spec.authors       = ["rainkinz"]
+  spec.email         = ["brendan.grainger@gmail.com"]
+  spec.description   = %q{Adds suggest and spellcheck DSL methods}
+  spec.summary       = %q{Adds suggest and spellcheck DSL methods}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency "sunspot", "~> 2.0"
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency 'rspec', '~>2.6.0'
+  spec.add_development_dependency 'pry'
+end

metadata

@@ -0,0 +1,184 @@
+--- !ruby/object:Gem::Specification
+name: sunspot_suggest
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+  prerelease: 
+platform: ruby
+authors:
+- rainkinz
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-07-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: sunspot
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.6.0
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Adds suggest and spellcheck DSL methods
+email:
+- brendan.grainger@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/sunspot_suggest.rb
+- lib/sunspot_suggest/plugin.rb
+- lib/sunspot_suggest/sunspot/dsl/standard_query.rb
+- lib/sunspot_suggest/sunspot/query/common_query.rb
+- lib/sunspot_suggest/sunspot/query/spellcheck.rb
+- lib/sunspot_suggest/sunspot/query/suggest.rb
+- lib/sunspot_suggest/sunspot/search/abstract_search.rb
+- lib/sunspot_suggest/version.rb
+- spec/api/query/spellcheck_spec.rb
+- spec/api/search/spellcheck_search_spec.rb
+- spec/article.rb
+- spec/helpers/indexer_helper.rb
+- spec/helpers/integration_helper.rb
+- spec/helpers/mock_session_helper.rb
+- spec/helpers/query_helper.rb
+- spec/helpers/search_helper.rb
+- spec/helpers/spellcheck_helper.rb
+- spec/mocks/adapters.rb
+- spec/mocks/blog.rb
+- spec/mocks/comment.rb
+- spec/mocks/connection.rb
+- spec/mocks/mock_adapter.rb
+- spec/mocks/mock_class_sharding_session_proxy.rb
+- spec/mocks/mock_record.rb
+- spec/mocks/mock_sharding_session_proxy.rb
+- spec/mocks/photo.rb
+- spec/mocks/post.rb
+- spec/mocks/super_class.rb
+- spec/mocks/user.rb
+- spec/spec_helper.rb
+- sunspot_suggest.gemspec
+homepage: ''
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: Adds suggest and spellcheck DSL methods
+test_files:
+- spec/api/query/spellcheck_spec.rb
+- spec/api/search/spellcheck_search_spec.rb
+- spec/article.rb
+- spec/helpers/indexer_helper.rb
+- spec/helpers/integration_helper.rb
+- spec/helpers/mock_session_helper.rb
+- spec/helpers/query_helper.rb
+- spec/helpers/search_helper.rb
+- spec/helpers/spellcheck_helper.rb
+- spec/mocks/adapters.rb
+- spec/mocks/blog.rb
+- spec/mocks/comment.rb
+- spec/mocks/connection.rb
+- spec/mocks/mock_adapter.rb
+- spec/mocks/mock_class_sharding_session_proxy.rb
+- spec/mocks/mock_record.rb
+- spec/mocks/mock_sharding_session_proxy.rb
+- spec/mocks/photo.rb
+- spec/mocks/post.rb
+- spec/mocks/super_class.rb
+- spec/mocks/user.rb
+- spec/spec_helper.rb