ssickles-tire 0.4.2

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/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
+    { :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,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

@@ -0,0 +1,88 @@
+require 'rubygems'
+require 'bundler/setup'
+
+require 'pathname'
+require 'test/unit'
+
+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={})
+    Tire::HTTP::Response.new(body, code, headers)
+  end
+
+  def fixtures_path
+    Pathname( File.expand_path( 'fixtures', File.dirname(__FILE__) ) )
+  end
+
+  def fixture_file(path)
+    File.read File.expand_path( path, fixtures_path )
+  end
+
+end
+
+module Test::Integration
+  URL = "http://localhost:9200"
+
+  def setup
+    begin; Object.send(:remove_const, :Rails); rescue; end
+
+    begin
+      ::RestClient.get URL
+    rescue Errno::ECONNREFUSED
+      abort "\n\n#{'-'*87}\n[ABORTED] You have to run ElasticSearch on #{URL} for integration tests\n#{'-'*87}\n\n"
+    end
+
+    ::RestClient.delete "#{URL}/articles-test"     rescue nil
+    ::RestClient.post   "#{URL}/articles-test", ''
+    fixtures_path.join('articles').entries.each do |f|
+      filename = f.to_s
+      next if filename =~ /^\./
+      ::RestClient.put "#{URL}/articles-test/article/#{File.basename(filename, '.*')}",
+                       fixtures_path.join('articles').join(f).read
+    end
+    ::RestClient.post "#{URL}/articles-test/_refresh", ''
+
+    Dir[File.dirname(__FILE__) + '/models/**/*.rb'].each { |m| load m }
+  end
+
+  def teardown
+    %w[
+      articles-test
+      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
+end