load_balanced_tire 0.1

data/test/integration/script_fields_test.rb

@@ -0,0 +1,38 @@
+require 'test_helper'
+
+module Tire
+
+  class ScriptFieldsIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "ScriptFields" do
+
+      should "add multiple fields to the results" do
+        # 1.json > title: "One", words: 125
+
+        s = Tire.search('articles-test') do
+          query { string "One" }
+          script_field :double_words, :script => "doc['words'].value * 2"
+          script_field :triple_words, :script => "doc['words'].value * 3"
+        end
+
+        assert_equal 250, s.results.first.double_words
+        assert_equal 375, s.results.first.triple_words
+      end
+
+      should "allow passing parameters to the script" do
+        # 1.json > title: "One", words: 125
+
+        s = Tire.search('articles-test') do
+          query { string "One" }
+          script_field :double_words, :script => "doc['words'].value * factor", :params => { :factor => 2 }
+        end
+
+        assert_equal 250, s.results.first.double_words
+      end
+
+    end
+
+  end
+
+end

data/test/integration/sort_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+
+module Tire
+
+  class SortIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Sort" do
+
+      should "sort by title" do
+        q = '*'
+        s = Tire.search('articles-test') do
+          query { string q }
+          sort  { by :title }
+        end
+
+        assert_equal 5, s.results.count
+        assert_equal 'Five', s.results.first[:title]
+      end
+
+      should "sort by title, descending" do
+        q = '*'
+        s = Tire.search('articles-test') do
+          query { string q }
+          sort  { by :title, :desc }
+        end
+
+        assert_equal 5, s.results.count
+        assert_equal 'Two', s.results.first[:title]
+      end
+
+    end
+
+  end
+
+end

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,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