tire-erez 0.5.4

data/test/integration/text_query_test.rb

@@ -0,0 +1,39 @@
+require 'test_helper'
+
+module Tire
+
+  class QueryStringIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Text query" do
+      setup do
+        Tire.index('articles-test') do
+          store :type => 'article', :title => '+1 !!!'
+          store :type => 'article', :title => 'Furry Kitten'
+          refresh
+        end
+      end
+
+      should "find article by title" do
+        results = Tire.search('articles-test') do
+          query { text :title, '+1' }
+        end.results
+
+        assert_equal 1,        results.count
+        assert_equal "+1 !!!", results.first[:title]
+      end
+
+      should "allow to pass options (fuzziness)" do
+        results = Tire.search('articles-test') do
+          query { text :title, 'fuzzy mitten', :fuzziness => 0.5, :operator => 'and' }
+        end.results
+
+        assert_equal 1,        results.count
+        assert_equal "Furry Kitten", results.first[:title]
+      end
+
+    end
+
+  end
+
+end

data/test/models/active_model_article.rb

@@ -0,0 +1,31 @@
+# Example ActiveModel class
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticle
+
+  extend  ActiveModel::Naming
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+
+  include Tire::Model::Search
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def id;         attributes['id'] || attributes['_id']; end
+  def id=(value); attributes['id'] = value;              end
+
+  def method_missing(name, *args, &block)
+    attributes[name.to_sym] || attributes[name.to_s] || super
+  end
+
+  def persisted?; true; end
+  def save;       true; end
+
+end

data/test/models/active_model_article_with_callbacks.rb

@@ -0,0 +1,49 @@
+# Example ActiveModel class with callbacks
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticleWithCallbacks
+
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Validations
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+  include ActiveModel::Naming
+
+  extend  ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def method_missing(id, *args, &block)
+    attributes[id.to_sym] || attributes[id.to_s] || super
+  end
+
+  def persisted?
+    true
+  end
+
+  def save
+    _run_save_callbacks do
+      STDERR.puts "[Saving ...]"
+    end
+  end
+
+  def destroy
+    _run_destroy_callbacks do
+      STDERR.puts "[Destroying ...]"
+      @destroyed = true
+    end
+  end
+
+  def destroyed?; !!@destroyed; end
+
+end

data/test/models/active_model_article_with_custom_document_type.rb

@@ -0,0 +1,7 @@
+# Example ActiveModel class with custom document type
+
+require File.expand_path('../active_model_article', __FILE__)
+
+class ActiveModelArticleWithCustomDocumentType < ActiveModelArticle
+  document_type 'my_custom_type'
+end

data/test/models/active_model_article_with_custom_index_name.rb

@@ -0,0 +1,7 @@
+# Example ActiveModel class with custom index name
+
+require File.expand_path('../active_model_article', __FILE__)
+
+class ActiveModelArticleWithCustomIndexName < ActiveModelArticle
+  index_name 'custom-index-name'
+end

data/test/models/active_record_models.rb

@@ -0,0 +1,131 @@
+require 'active_record'
+
+class ActiveRecordArticle < ActiveRecord::Base
+  has_many :comments, :class_name => "ActiveRecordComment", :foreign_key => "article_id"
+  has_many :stats,    :class_name => "ActiveRecordStat",    :foreign_key => "article_id"
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  tire do
+    mapping do
+      indexes :title,      :type => 'string', :boost => 10, :analyzer => 'snowball'
+      indexes :created_at, :type => 'date'
+
+      indexes :comments do
+        indexes :author
+        indexes :body
+      end
+    end
+  end
+
+  def to_indexed_json
+    {
+      :title        => title,
+      :length       => length,
+
+      :comments     => comments.map { |c| { :_type  => 'active_record_comment',
+                                            :_id    => c.id,
+                                            :author => c.author,
+                                            :body   => c.body  } },
+      :stats        => stats.map    { |s| { :pageviews  => s.pageviews } }
+    }.to_json
+  end
+
+  def length
+    title.length
+  end
+
+  def comment_authors
+    comments.map(&:author).to_sentence
+  end
+end
+
+class ActiveRecordComment < ActiveRecord::Base
+  belongs_to :article, :class_name => "ActiveRecordArticle", :foreign_key => "article_id"
+end
+
+class ActiveRecordStat < ActiveRecord::Base
+  belongs_to :article, :class_name => "ActiveRecordArticle", :foreign_key => "article_id"
+end
+
+class ActiveRecordClassWithTireMethods < ActiveRecord::Base
+
+  def self.mapping
+    "THIS IS MY MAPPING!"
+  end
+
+  def index
+    "THIS IS MY INDEX!"
+  end
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  tire do
+    mapping do
+      indexes :title, :type => 'string', :analyzer => 'snowball'
+    end
+  end
+end
+
+class ActiveRecordClassWithDynamicIndexName < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  index_name do
+    "dynamic" + '_' + "index"
+  end
+end
+
+# Used in test for multiple class instances in one index,
+# and single table inheritance (STI) support.
+
+class ActiveRecordModelOne < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+  self.table_name = 'active_record_model_one'
+  index_name 'active_record_model_one'
+end
+
+class ActiveRecordModelTwo < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+  self.table_name = 'active_record_model_two'
+  index_name 'active_record_model_two'
+end
+
+class ActiveRecordAsset < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+end
+
+class ActiveRecordVideo < ActiveRecordAsset
+  index_name 'active_record_assets'
+end
+
+class ActiveRecordPhoto < ActiveRecordAsset
+  index_name 'active_record_assets'
+end
+
+# Namespaced ActiveRecord models
+
+module ActiveRecordNamespace
+  def self.table_name_prefix
+    'active_record_namespace_'
+  end
+end
+
+class ActiveRecordNamespace::MyModel < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+end
+
+# Model with percolation
+
+class ActiveRecordModelWithPercolation < ActiveRecord::Base
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  percolate!
+end

data/test/models/article.rb

@@ -0,0 +1,15 @@
+# Example non-ActiveModel custom wrapper for result item
+
+class Article
+  attr_reader :id, :title, :body
+
+  def initialize(attributes={})
+    attributes.each { |k,v| instance_variable_set(:"@#{k}", v) }
+  end
+
+  def to_json(options={})
+    { :id => @id, :title => @title, :body => @body }.to_json
+  end
+
+  alias :to_indexed_json :to_json
+end

data/test/models/mongoid_models.rb

@@ -0,0 +1,85 @@
+require 'mongoid'
+
+class MongoidArticle
+
+  include Mongoid::Document
+
+
+  has_many :comments, :class_name => "MongoidComment", :foreign_key => "article_id"
+  has_many :stats,    :class_name => "MongoidStat",    :foreign_key => "article_id"
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  tire do
+    mapping do
+      indexes :title,      :type => 'string', :boost => 10, :analyzer => 'snowball'
+      indexes :created_at, :type => 'date'
+
+      indexes :comments do
+        indexes :author
+        indexes :body
+      end
+    end
+  end
+
+  def to_indexed_json
+    {
+      :title        => title,
+      :length       => length,
+
+      :comments     => comments.map { |c| { :_type  => 'mongoid_comment',
+                                            :_id    => c.id,
+                                            :author => c.author,
+                                            :body   => c.body  } },
+      :stats        => stats.map    { |s| { :pageviews  => s.pageviews } }
+    }.to_json
+  end
+
+  def length
+    title.length
+  end
+
+  def comment_authors
+    comments.map(&:author).to_sentence
+  end
+end
+
+class MongoidComment
+
+  include Mongoid::Document
+
+
+  belongs_to :article, :class_name => "MongoidArticle", :foreign_key => "article_id"
+end
+
+class MongoidStat
+
+  include Mongoid::Document
+
+
+  belongs_to :article, :class_name => "MongoidArticle", :foreign_key => "article_id"
+end
+
+class MongoidClassWithTireMethods
+
+  include Mongoid::Document
+
+
+  def self.mapping
+    "THIS IS MY MAPPING!"
+  end
+
+  def index
+    "THIS IS MY INDEX!"
+  end
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  tire do
+    mapping do
+      indexes :title, :type => 'string', :analyzer => 'snowball'
+    end
+  end
+end

data/test/models/persistent_article.rb

@@ -0,0 +1,11 @@
+# Example class with Elasticsearch persistence
+
+class PersistentArticle
+
+  include Tire::Model::Persistence
+
+  property :title
+  property :published_on
+  property :tags
+
+end

data/test/models/persistent_article_in_index.rb

@@ -0,0 +1,16 @@
+# Example class with Elasticsearch persistence in index `persistent_articles`
+#
+# The `index` is `persistent_articles`
+#
+
+class PersistentArticleInIndex
+
+  include Tire::Model::Persistence
+
+  property :title
+  property :published_on
+  property :tags
+
+  index_name "persistent_articles"
+
+end

data/test/models/persistent_article_in_namespace.rb

@@ -0,0 +1,12 @@
+# Example namespaced class with Elasticsearch persistence
+#
+# The `document_type` is `my_namespace/persistent_article_in_namespace`
+#
+
+module MyNamespace
+  class PersistentArticleInNamespace
+    include Tire::Model::Persistence
+
+    property :title
+  end
+end

data/test/models/persistent_article_with_casting.rb

@@ -0,0 +1,28 @@
+class Author
+  attr_accessor :first_name, :last_name
+  def initialize(attributes)
+    @first_name = HashWithIndifferentAccess.new(attributes)[:first_name]
+    @last_name  = HashWithIndifferentAccess.new(attributes)[:last_name]
+  end
+end
+
+class Comment
+  def initialize(params);                      @attributes = HashWithIndifferentAccess.new(params);  end
+  def method_missing(method_name, *arguments); @attributes[method_name];                             end
+  def as_json(*);                              @attributes;                                          end
+end
+
+class PersistentArticleWithCastedItem
+  include Tire::Model::Persistence
+
+  property :title
+  property :author, :class => Author
+  property :stats
+end
+
+class PersistentArticleWithCastedCollection
+  include Tire::Model::Persistence
+
+  property :title
+  property :comments, :class => [Comment]
+end