searchkick-sinneduy 0.9.0

data/test/suggest_test.rb

@@ -0,0 +1,82 @@
+require_relative "test_helper"
+
+class TestSuggest < Minitest::Test
+
+  def test_basic
+    store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
+    assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [:name]
+  end
+
+  def test_perfect
+    store_names ["Tiger Shark", "Great White Shark"]
+    assert_suggest "Tiger Shark", nil, fields: [:name] # no correction
+  end
+
+  def test_phrase
+    store_names ["Big Tiger Shark", "Tiger Sharp Teeth", "Tiger Sharp Mind"]
+    assert_suggest "How to catch a big tiger shar", "how to catch a big tiger shark", fields: [:name]
+  end
+
+  def test_without_option
+    store_names ["hi"] # needed to prevent ElasticsearchException - seed 668
+    assert_raises(RuntimeError) { Product.search("hi").suggestions }
+  end
+
+  def test_multiple_fields
+    store [
+      {name: "Shark", color: "Sharp"},
+      {name: "Shark", color: "Sharp"}
+    ]
+    assert_suggest_all "shar", ["shark", "sharp"]
+  end
+
+  def test_multiple_fields_highest_score_first
+    store [
+      {name: "Tiger Shark", color: "Sharp"}
+    ]
+    assert_suggest "tiger shar", "tiger shark"
+  end
+
+  def test_multiple_fields_same_value
+    store [
+      {name: "Shark", color: "Shark"}
+    ]
+    assert_suggest_all "shar", ["shark"]
+  end
+
+  def test_fields_option
+    store [
+      {name: "Shark", color: "Sharp"}
+    ]
+    assert_suggest_all "shar", ["shark"], fields: [:name]
+  end
+
+  def test_fields_option_multiple
+    store [
+      {name: "Shark"}
+    ]
+    assert_suggest "shar", "shark", fields: [:name, :unknown]
+  end
+
+  def test_fields_partial_match
+    store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
+    assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [{name: :word_start}]
+  end
+
+  def test_fields_partial_match_boost
+    store_names ["Great White Shark", "Hammerhead Shark", "Tiger Shark"]
+    assert_suggest "How Big is a Tigre Shar", "how big is a tiger shark", fields: [{"name^2" => :word_start}]
+  end
+
+  protected
+
+  def assert_suggest(term, expected, options = {})
+    assert_equal expected, Product.search(term, options.merge(suggest: true)).suggestions.first
+  end
+
+  # any order
+  def assert_suggest_all(term, expected, options = {})
+    assert_equal expected.sort, Product.search(term, options.merge(suggest: true)).suggestions.sort
+  end
+
+end

data/test/synonyms_test.rb

@@ -0,0 +1,50 @@
+require_relative "test_helper"
+
+class TestSynonyms < Minitest::Test
+
+  def test_bleach
+    store_names ["Clorox Bleach", "Kroger Bleach"]
+    assert_search "clorox", ["Clorox Bleach", "Kroger Bleach"]
+  end
+
+  def test_saran_wrap
+    store_names ["Saran Wrap", "Kroger Plastic Wrap"]
+    assert_search "saran wrap", ["Saran Wrap", "Kroger Plastic Wrap"]
+  end
+
+  def test_burger_buns
+    store_names ["Hamburger Buns"]
+    assert_search "burger buns", ["Hamburger Buns"]
+  end
+
+  def test_bandaids
+    store_names ["Band-Aid", "Kroger 12-Pack Bandages"]
+    assert_search "bandaids", ["Band-Aid", "Kroger 12-Pack Bandages"]
+  end
+
+  def test_qtips
+    store_names ["Q Tips", "Kroger Cotton Swabs"]
+    assert_search "q tips", ["Q Tips", "Kroger Cotton Swabs"]
+  end
+
+  def test_reverse
+    store_names ["Scallions"]
+    assert_search "green onions", ["Scallions"]
+  end
+
+  def test_exact
+    store_names ["Green Onions", "Yellow Onions"]
+    assert_search "scallion", ["Green Onions"]
+  end
+
+  def test_stemmed
+    store_names ["Green Onions", "Yellow Onions"]
+    assert_search "scallions", ["Green Onions"]
+  end
+
+  # def test_wordnet
+  #   store_names ["Creature", "Beast", "Dragon"], Animal
+  #   assert_search "animal", ["Creature", "Beast"], {}, Animal
+  # end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,276 @@
+require "bundler/setup"
+Bundler.require(:default)
+require "minitest/autorun"
+require "minitest/pride"
+require "logger"
+
+ENV["RACK_ENV"] = "test"
+
+Minitest::Test = Minitest::Unit::TestCase unless defined?(Minitest::Test)
+
+File.delete("elasticsearch.log") if File.exist?("elasticsearch.log")
+Searchkick.client.transport.logger = Logger.new("elasticsearch.log")
+
+puts "Running against Elasticsearch #{Searchkick.server_version}"
+
+I18n.config.enforce_available_locales = true
+
+ActiveJob::Base.logger = nil if defined?(ActiveJob)
+
+if defined?(Mongoid)
+
+  def mongoid2?
+    Mongoid::VERSION.starts_with?("2.")
+  end
+
+  if mongoid2?
+    # enable comparison of BSON::ObjectIds
+    module BSON
+      class ObjectId
+        def <=>(other)
+          data <=> other.data
+        end
+      end
+    end
+  end
+
+  Mongoid.configure do |config|
+    if mongoid2?
+      config.master = Mongo::Connection.new.db("searchkick_test")
+    else
+      config.connect_to "searchkick_test"
+    end
+  end
+
+  class Product
+    include Mongoid::Document
+    include Mongoid::Timestamps
+
+    field :name
+    field :store_id, type: Integer
+    field :in_stock, type: Boolean
+    field :backordered, type: Boolean
+    field :orders_count, type: Integer
+    field :price, type: Integer
+    field :color
+    field :latitude, type: BigDecimal
+    field :longitude, type: BigDecimal
+    field :description
+  end
+
+  class Store
+    include Mongoid::Document
+
+    field :name
+  end
+
+  class Animal
+    include Mongoid::Document
+
+    field :name
+  end
+
+  class Dog < Animal
+  end
+
+  class Cat < Animal
+  end
+elsif defined?(NoBrainer)
+  NoBrainer.configure do |config|
+    config.app_name = :searchkick
+    config.environment = :test
+  end
+
+  class Product
+    include NoBrainer::Document
+    include NoBrainer::Document::Timestamps
+
+    field :name,         type: String
+    field :store_id,     type: Integer
+    field :in_stock,     type: Boolean
+    field :backordered,  type: Boolean
+    field :orders_count, type: Integer
+    field :price,        type: Integer
+    field :color,        type: String
+    field :latitude
+    field :longitude
+    field :description,  type: String
+  end
+
+  class Store
+    include NoBrainer::Document
+
+    field :name, type: String
+  end
+
+  class Animal
+    include NoBrainer::Document
+
+    field :name, type: String
+  end
+
+  class Dog < Animal
+  end
+
+  class Cat < Animal
+  end
+else
+  require "active_record"
+
+  # for debugging
+  # ActiveRecord::Base.logger = Logger.new(STDOUT)
+
+  # rails does this in activerecord/lib/active_record/railtie.rb
+  ActiveRecord::Base.default_timezone = :utc
+  ActiveRecord::Base.time_zone_aware_attributes = true
+
+  # migrations
+  ActiveRecord::Base.establish_connection adapter: "sqlite3", database: ":memory:"
+
+  ActiveRecord::Base.raise_in_transactional_callbacks = true if ActiveRecord::Base.respond_to?(:raise_in_transactional_callbacks=)
+
+  ActiveRecord::Migration.create_table :products do |t|
+    t.string :name
+    t.integer :store_id
+    t.boolean :in_stock
+    t.boolean :backordered
+    t.integer :orders_count
+    t.integer :price
+    t.string :color
+    t.decimal :latitude, precision: 10, scale: 7
+    t.decimal :longitude, precision: 10, scale: 7
+    t.text :description
+    t.timestamps null: true
+  end
+
+  ActiveRecord::Migration.create_table :stores do |t|
+    t.string :name
+  end
+
+  ActiveRecord::Migration.create_table :animals do |t|
+    t.string :name
+    t.string :type
+  end
+
+  class Product < ActiveRecord::Base
+  end
+
+  class Store < ActiveRecord::Base
+  end
+
+  class Animal < ActiveRecord::Base
+  end
+
+  class Dog < Animal
+  end
+
+  class Cat < Animal
+  end
+end
+
+class Product
+  belongs_to :store
+
+  searchkick \
+    synonyms: [
+      ["clorox", "bleach"],
+      ["scallion", "greenonion"],
+      ["saranwrap", "plasticwrap"],
+      ["qtip", "cottonswab"],
+      ["burger", "hamburger"],
+      ["bandaid", "bandag"]
+    ],
+    autocomplete: [:name],
+    suggest: [:name, :color],
+    conversions: "conversions",
+    personalize: "user_ids",
+    locations: ["location", "multiple_locations"],
+    text_start: [:name],
+    text_middle: [:name],
+    text_end: [:name],
+    word_start: [:name],
+    word_middle: [:name],
+    word_end: [:name],
+    highlight: [:name],
+    unsearchable: [:description]
+
+  attr_accessor :conversions, :user_ids, :aisle
+
+  def search_data
+    serializable_hash.except("id").merge(
+      conversions: conversions,
+      user_ids: user_ids,
+      location: [latitude, longitude],
+      multiple_locations: [[latitude, longitude], [0, 0]],
+      aisle: aisle
+    )
+  end
+
+  def should_index?
+    name != "DO NOT INDEX"
+  end
+end
+
+class Store
+  searchkick \
+    routing: :name,
+    merge_mappings: true,
+    mappings: {
+      store: {
+        properties: {
+          name: {type: "string", analyzer: "keyword"},
+        }
+      }
+  }
+end
+
+class Animal
+  searchkick \
+    autocomplete: [:name],
+    suggest: [:name],
+    index_name: -> { "#{name.tableize}-#{Date.today.year}" }
+    # wordnet: true
+end
+
+Product.searchkick_index.delete if Product.searchkick_index.exists?
+Product.reindex
+Product.reindex # run twice for both index paths
+
+Store.reindex
+Animal.reindex
+
+class Minitest::Test
+
+  def setup
+    Product.destroy_all
+    Store.destroy_all
+    Animal.destroy_all
+  end
+
+  protected
+
+  def store(documents, klass = Product)
+    documents.shuffle.each do |document|
+      klass.create!(document)
+    end
+    klass.searchkick_index.refresh
+  end
+
+  def store_names(names, klass = Product)
+    store names.map { |name| {name: name} }, klass
+  end
+
+  # no order
+  def assert_search(term, expected, options = {}, klass = Product)
+    assert_equal expected.sort, klass.search(term, options).map(&:name).sort
+  end
+
+  def assert_order(term, expected, options = {}, klass = Product)
+    assert_equal expected, klass.search(term, options).map(&:name)
+  end
+
+  def assert_first(term, expected, options = {}, klass = Product)
+    assert_equal expected, klass.search(term, options).map(&:name).first
+  end
+
+end

metadata

@@ -0,0 +1,194 @@
+--- !ruby/object:Gem::Specification
+name: searchkick-sinneduy
+version: !ruby/object:Gem::Version
+  version: 0.9.0
+platform: ruby
+authors:
+- Andrew Kane
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-06-12 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: activemodel
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: elasticsearch-sinneduy
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.13
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 1.0.13
+- !ruby/object:Gem::Dependency
+  name: hashie
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.6'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Intelligent search made easy
+email:
+- andrew@chartkick.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .travis.yml
+- CHANGELOG.md
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- ci/before_install.sh
+- gemfiles/activerecord31.gemfile
+- gemfiles/activerecord32.gemfile
+- gemfiles/activerecord40.gemfile
+- gemfiles/activerecord41.gemfile
+- gemfiles/mongoid2.gemfile
+- gemfiles/mongoid3.gemfile
+- gemfiles/mongoid4.gemfile
+- gemfiles/nobrainer.gemfile
+- lib/searchkick.rb
+- lib/searchkick/index.rb
+- lib/searchkick/logging.rb
+- lib/searchkick/model.rb
+- lib/searchkick/query.rb
+- lib/searchkick/reindex_job.rb
+- lib/searchkick/reindex_v2_job.rb
+- lib/searchkick/results.rb
+- lib/searchkick/tasks.rb
+- lib/searchkick/version.rb
+- searchkick.gemspec
+- test/autocomplete_test.rb
+- test/boost_test.rb
+- test/facets_test.rb
+- test/highlight_test.rb
+- test/index_test.rb
+- test/inheritance_test.rb
+- test/match_test.rb
+- test/model_test.rb
+- test/query_test.rb
+- test/reindex_job_test.rb
+- test/reindex_v2_job_test.rb
+- test/routing_test.rb
+- test/should_index_test.rb
+- test/similar_test.rb
+- test/sql_test.rb
+- test/suggest_test.rb
+- test/synonyms_test.rb
+- test/test_helper.rb
+homepage: https://github.com/ankane/searchkick
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.0.14
+signing_key: 
+specification_version: 4
+summary: Searchkick learns what your users are looking for. As more people search,
+  it gets smarter and the results get better. It’s friendly for developers - and magical
+  for your users.
+test_files:
+- test/autocomplete_test.rb
+- test/boost_test.rb
+- test/facets_test.rb
+- test/highlight_test.rb
+- test/index_test.rb
+- test/inheritance_test.rb
+- test/match_test.rb
+- test/model_test.rb
+- test/query_test.rb
+- test/reindex_job_test.rb
+- test/reindex_v2_job_test.rb
+- test/routing_test.rb
+- test/should_index_test.rb
+- test/similar_test.rb
+- test/sql_test.rb
+- test/suggest_test.rb
+- test/synonyms_test.rb
+- test/test_helper.rb
+has_rdoc: