ssickles-tire 0.4.2.7 → 0.4.3

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,122 @@
+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

data/test/models/mongoid_models.rb

@@ -0,0 +1,97 @@
+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"
+
+  # def index
+  #   "KEEP OFF MY INDEX!!!"
+  # end
+  #
+  # def self.settings
+  #   "KEEP OFF MY SETTINGS!!!"
+  # end
+
+  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
+
+  # tire.mapping do
+  #   indexes :title,      :type => 'string', :boost => 10, :analyzer => 'snowball'
+  # 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_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

data/test/models/persistent_article_with_defaults.rb

@@ -0,0 +1,11 @@
+class PersistentArticleWithDefaults
+
+  include Tire::Model::Persistence
+
+  property :title
+  property :published_on
+  property :tags,   :default => []
+  property :hidden, :default => false
+  property :options,  :default => {:switches => []}
+
+end

data/test/models/persistent_articles_with_custom_index_name.rb

@@ -0,0 +1,10 @@
+# Example class with ElasticSearch persistence and custom index name
+
+class PersistentArticleWithCustomIndexName
+
+  include Tire::Model::Persistence
+
+  property :title
+
+  index_name 'custom-index-name'
+end

data/test/models/supermodel_article.rb

@@ -0,0 +1,17 @@
+# Example ActiveModel class for testing :searchable mode
+
+require 'rubygems'
+require 'redis/persistence'
+
+class SupermodelArticle
+  include Redis::Persistence
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  property :title
+
+  mapping do
+    indexes :title,      :type => 'string', :boost => 15, :analyzer => 'czech'
+  end
+end

data/test/models/validated_model.rb

@@ -0,0 +1,11 @@
+# Example ActiveModel with validations
+
+class ValidatedModel
+
+  include Tire::Model::Persistence
+
+  property :name
+
+  validates_presence_of :name
+
+end

data/test/test_helper.rb

@@ -4,15 +4,32 @@ require 'bundler/setup'
 require 'pathname'
 require 'test/unit'
 
-require 'yajl'
-#require 'yajl/json_gem'
+require 'yajl/json_gem'
+require 'sqlite3'
 
 require 'shoulda'
 require 'turn/autorun' unless ENV["TM_FILEPATH"] || ENV["CI"] || defined?(RUBY_VERSION) && RUBY_VERSION < '1.9'
 require 'mocha'
 
+require 'active_support/core_ext/hash/indifferent_access'
+
 require 'tire'
 
+# Require basic model files
+#
+require File.dirname(__FILE__) + '/models/active_model_article'
+require File.dirname(__FILE__) + '/models/active_model_article_with_callbacks'
+require File.dirname(__FILE__) + '/models/active_model_article_with_custom_document_type'
+require File.dirname(__FILE__) + '/models/active_model_article_with_custom_index_name'
+require File.dirname(__FILE__) + '/models/active_record_models'
+require File.dirname(__FILE__) + '/models/article'
+require File.dirname(__FILE__) + '/models/persistent_article'
+require File.dirname(__FILE__) + '/models/persistent_article_in_namespace'
+require File.dirname(__FILE__) + '/models/persistent_article_with_casting'
+require File.dirname(__FILE__) + '/models/persistent_article_with_defaults'
+require File.dirname(__FILE__) + '/models/persistent_articles_with_custom_index_name'
+require File.dirname(__FILE__) + '/models/validated_model'
+
 class Test::Unit::TestCase
 
   def mock_response(body, code=200, headers={})
@@ -57,7 +74,14 @@ module Test::Integration
   def teardown
     %w[
       articles-test
-      dynamic_index ].each do |index|
+      active_record_articles
+      active_model_article_with_custom_as_serializations
+      active_record_class_with_tire_methods
+      mongoid_articles
+      mongoid_class_with_tire_methods
+      supermodel_articles
+      dynamic_index
+      model_with_nested_documents ].each do |index|
         ::RestClient.delete "#{URL}/#{index}" rescue nil
     end
   end