tire-erez 0.5.4

data/test/integration/match_query_test.rb

@@ -0,0 +1,79 @@
+require 'test_helper'
+
+module Tire
+
+  class MatchQueryIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Match query" do
+      setup do
+        Tire.index 'match-query-test' do
+          delete
+          create settings: { index: { number_of_shards: 1, number_of_replicas: 0 } },
+                 mappings: {
+                  document: { properties: {
+                    last_name: { type: 'string', analyzer: 'english' },
+                    age:       { type: 'integer' }
+                  } }
+                 }
+
+          store first_name: 'John', last_name: 'Smith',    age: 30, gender: 'male'
+          store first_name: 'John', last_name: 'Smithson', age: 25, gender: 'male'
+          store first_name: 'Adam', last_name: 'Smith',    age: 75, gender: 'male'
+          store first_name: 'Mary', last_name: 'John',     age: 30, gender: 'female'
+          refresh
+        end
+      end
+
+      teardown do
+        Tire.index('match-query-test').delete
+      end
+
+      should "find documents by single field" do
+        s = Tire.search 'match-query-test' do
+          query do
+            match :last_name, 'Smith'
+          end
+        end
+
+        assert_equal 2, s.results.count
+      end
+
+      should "find document by multiple fields with multi_match" do
+        s = Tire.search 'match-query-test' do
+          query do
+            match [:first_name, :last_name], 'John'
+          end
+        end
+
+        assert_equal 3, s.results.count
+      end
+
+      should "find documents by prefix" do
+        s = Tire.search 'match-query-test' do
+          query do
+            match :last_name, 'Smi', type: 'phrase_prefix'
+          end
+        end
+
+        assert_equal 3, s.results.count
+      end
+
+      should "automatically create a boolean query when called repeatedly" do
+        s = Tire.search 'match-query-test' do
+          query do
+            match [:first_name, :last_name], 'John'
+            match :age, 30
+            match :gender, 'male'
+          end
+          # puts to_curl
+        end
+
+        assert_equal 1, s.results.count
+      end
+
+    end
+
+  end
+
+end

data/test/integration/mongoid_searchable_test.rb

@@ -0,0 +1,309 @@
+require 'test_helper'
+require File.expand_path('../../models/mongoid_models', __FILE__)
+
+begin
+  require "mongo"
+  Mongo::Connection.new("localhost", 27017)
+
+  ENV["MONGODB_IS_AVAILABLE"] = 'true'
+rescue Mongo::ConnectionFailure => e
+  ENV["MONGODB_IS_AVAILABLE"] = nil
+end
+
+if ENV["MONGODB_IS_AVAILABLE"]
+  module Tire
+
+    class MongoidSearchableIntegrationTest < Test::Unit::TestCase
+      include Test::Integration
+
+      def setup
+        super
+        Mongoid.configure do |config|
+          config.master = Mongo::Connection.new.db("tire_mongoid_integration_test")
+        end
+      end
+
+      context "Mongoid integration" do
+
+        setup    do
+          MongoidArticle.destroy_all
+          Tire.index('mongoid_articles').delete
+
+          load File.expand_path('../../models/mongoid_models.rb', __FILE__)
+        end
+        teardown do
+          MongoidArticle.destroy_all
+          Tire.index('mongoid_articles').delete
+        end
+
+        should "configure mapping" do
+          assert_equal 'snowball', MongoidArticle.mapping[:title][:analyzer]
+          assert_equal 10, MongoidArticle.mapping[:title][:boost]
+
+          assert_equal 'snowball', MongoidArticle.tire.index.mapping['mongoid_article']['properties']['title']['analyzer']
+        end
+
+        should "save document into index on save and find it" do
+          a = MongoidArticle.new :title => 'Test'
+          a.save!
+          id = a.id
+
+          a.index.refresh
+
+          results = MongoidArticle.tire.search 'test'
+
+          assert       results.any?
+          assert_equal 1, results.count
+
+          assert_instance_of Results::Item, results.first
+          assert_not_nil results.first.id
+          assert_equal   id.to_s, results.first.id.to_s
+          assert         results.first.persisted?, "Record should be persisted"
+          assert_not_nil results.first._score
+          assert_equal   'Test', results.first.title
+        end
+
+        context "with eager loading" do
+          setup do
+            MongoidArticle.destroy_all
+
+            @first_article  = MongoidArticle.create! :title => "Test 1"
+            @second_article = MongoidArticle.create! :title => "Test 2"
+            @third_article  = MongoidArticle.create! :title => "Test 3"
+            @fourth_article = MongoidArticle.create! :title => "Test 4"
+            @fifth_article  = MongoidArticle.create! :title => "Test 5"
+
+            MongoidArticle.tire.index.refresh
+          end
+
+          should "load records on query search" do
+            results = MongoidArticle.tire.search '"Test 1"', :load => true
+
+            assert       results.any?
+            assert_equal MongoidArticle.all.first, results.first
+          end
+
+          should "load records on block search" do
+            results = MongoidArticle.tire.search :load => true do
+              query { string '"Test 1"' }
+            end
+
+            assert_equal MongoidArticle.all.first, results.first
+          end
+
+          should "load records with options on query search" do
+            assert_equal MongoidArticle.find([@first_article[:_id]], :include => 'comments').first,
+            MongoidArticle.tire.search('"Test 1"',
+                                  :load => { :include => 'comments' }).results.first
+          end
+
+          should "return empty collection for nonmatching query" do
+            assert_nothing_raised do
+              results = MongoidArticle.tire.search :load => true do
+                query { string '"Hic Sunt Leones"' }
+              end
+              assert_equal 0, results.size
+              assert !results.any?
+            end
+          end
+        end
+
+        should "remove document from index on destroy" do
+          a = MongoidArticle.new :title => 'Test remove...'
+          a.save!
+          assert_equal 1, MongoidArticle.count
+
+          a.destroy
+          assert_equal 0, MongoidArticle.all.size
+
+          a.index.refresh
+          results = MongoidArticle.tire.search 'test'
+          assert_equal 0, results.count
+        end
+
+        should "return documents with scores" do
+          MongoidArticle.create! :title => 'foo'
+          MongoidArticle.create! :title => 'bar'
+
+          MongoidArticle.tire.index.refresh
+          results = MongoidArticle.tire.search 'foo OR bar^100'
+          assert_equal 2, results.count
+
+          assert_equal 'bar', results.first.title
+        end
+
+        context "with pagination" do
+          setup do
+            1.upto(9) { |number| MongoidArticle.create :title => "Test#{number}" }
+            MongoidArticle.tire.index.refresh
+          end
+
+          context "and parameter searches" do
+
+            should "find first page with five results" do
+              results = MongoidArticle.tire.search 'test*', :sort => 'title', :per_page => 5, :page => 1
+              assert_equal 5, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 1, results.current_page
+              assert_equal nil, results.previous_page
+              assert_equal 2, results.next_page
+
+              assert_equal 'Test1', results.first.title
+            end
+
+            should "find next page with five results" do
+              results = MongoidArticle.tire.search 'test*', :sort => 'title', :per_page => 5, :page => 2
+              assert_equal 4, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 2, results.current_page
+              assert_equal 1, results.previous_page
+              assert_equal nil, results.next_page
+
+              assert_equal 'Test6', results.first.title
+            end
+
+            should "find not find missing page" do
+              results = MongoidArticle.tire.search 'test*', :sort => 'title', :per_page => 5, :page => 3
+              assert_equal 0, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 3, results.current_page
+              assert_equal 2, results.previous_page
+              assert_equal nil, results.next_page
+
+              assert_nil results.first
+            end
+
+          end
+
+          context "and block searches" do
+            setup { @q = 'test*' }
+
+            should "find first page with five results" do
+              results = MongoidArticle.tire.search do |search|
+                search.query { |query| query.string @q }
+                search.sort  { by :title }
+                search.from 0
+                search.size 5
+              end
+              assert_equal 5, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 1, results.current_page
+              assert_equal nil, results.previous_page
+              assert_equal 2, results.next_page
+
+              assert_equal 'Test1', results.first.title
+            end
+
+            should "find next page with five results" do
+              results = MongoidArticle.tire.search do |search|
+                search.query { |query| query.string @q }
+                search.sort  { by :title }
+                search.from 5
+                search.size 5
+              end
+              assert_equal 4, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 2, results.current_page
+              assert_equal 1, results.previous_page
+              assert_equal nil, results.next_page
+
+              assert_equal 'Test6', results.first.title
+            end
+
+            should "not find a missing page" do
+              results = MongoidArticle.tire.search do |search|
+                search.query { |query| query.string @q }
+                search.sort  { by :title }
+                search.from 10
+                search.size 5
+              end
+              assert_equal 0, results.size
+
+              assert_equal 2, results.total_pages
+              assert_equal 3, results.current_page
+              assert_equal 2, results.previous_page
+              assert_equal nil, results.next_page
+
+              assert_nil results.first
+            end
+
+          end
+
+        end
+
+        context "with proxy" do
+
+          should "allow access to Tire instance methods" do
+            a = MongoidClassWithTireMethods.create :title => 'One'
+            assert_equal "THIS IS MY INDEX!", a.index
+            assert_instance_of Tire::Index, a.tire.index
+            assert a.tire.index.exists?, "Index should exist"
+          end
+
+          should "allow access to Tire class methods" do
+            class ::MongoidClassWithTireMethods
+              include Mongoid::Document
+              def self.search(*)
+                "THIS IS MY SEARCH!"
+              end
+            end
+
+            MongoidClassWithTireMethods.create :title => 'One'
+            MongoidClassWithTireMethods.tire.index.refresh
+
+            assert_equal "THIS IS MY SEARCH!", MongoidClassWithTireMethods.search
+
+            results = MongoidClassWithTireMethods.tire.search 'one'
+
+            assert_equal 'One', results.first.title
+          end
+
+        end
+
+        context "within Rails" do
+
+          setup do
+            module ::Rails; end
+
+            a = MongoidArticle.new :title => 'Test'
+            c = a.comments.build :author => 'fool', :body => 'Works!'
+            s = a.stats.build    :pageviews  => 12, :period => '2011-08'
+            a.save!
+            c.save!
+            s.save!
+            @id = a.id.to_s
+
+            a.index.refresh
+            @item = MongoidArticle.tire.search('test').first
+          end
+
+          should "have access to indexed properties" do
+            assert_equal 'Test', @item.title
+            assert_equal 'fool', @item.comments.first.author
+            assert_equal 12,     @item.stats.first.pageviews
+          end
+
+          should "load the underlying models" do
+            assert_instance_of Results::Item, @item
+            assert_instance_of MongoidArticle, @item.load
+            assert_equal      'Test', @item.load.title
+
+            assert_instance_of Results::Item, @item.comments.first
+            assert_instance_of MongoidComment, @item.comments.first.load
+            assert_equal      'fool', @item.comments.first.load.author
+          end
+
+          should "load the underlying model with options" do
+            assert_equal MongoidArticle.find(@id), @item.load(:include => 'comments')
+          end
+
+        end
+      end
+    end
+  end
+end

data/test/integration/multi_search_test.rb

@@ -0,0 +1,114 @@
+require 'test_helper'
+
+module Tire
+
+  class MultiSearchIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Multi search" do
+      # Tire.configure { logger STDERR }
+      setup do
+        Tire.index 'multi-search-test-1' do
+          delete
+          create
+          store first_name: 'John', last_name: 'Smith',    age: 30, gender: 'male'
+          store first_name: 'John', last_name: 'Smithson', age: 25, gender: 'male'
+          store first_name: 'Mary', last_name: 'Smith',    age: 20, gender: 'female'
+          refresh
+        end
+        Tire.index 'multi-search-test-2' do
+          delete
+          create
+          store first_name: 'John', last_name: 'Milton', age: 35, gender: 'male'
+          store first_name: 'Mary', last_name: 'Milson', age: 44, gender: 'female'
+          store first_name: 'Mary', last_name: 'Reilly', age: 55, gender: 'female'
+          refresh
+        end
+      end
+
+      teardown do
+        Tire.index('multi-search-test-1').delete
+        Tire.index('multi-search-test-2').delete
+      end
+
+      should "return multiple results" do
+        s = Tire.multi_search 'multi-search-test-1' do
+          search :johns do
+            query { match :_all, 'john' }
+          end
+          search :males do
+            query { match :gender, 'male' }
+          end
+          search :facets, search_type: 'count' do
+            facet('age') { statistical :age }
+          end
+        end
+
+        assert_equal 3,    s.results.size
+
+        assert_equal 2,    s.results[:johns].size
+        assert_equal 2,    s.results[:males].size
+
+        assert s.results[:facets].results.empty?, "Results not empty? #{s.results[:facets].results}"
+        assert_equal 75.0, s.results[:facets].facets['age']['total']
+      end
+
+      should "mix named and numbered searches" do
+        s = Tire.multi_search 'multi-search-test-1' do
+          search(:johns) { query { match :_all, 'john' }  }
+          search         { query { match :_all, 'mary' }  }
+        end
+
+        assert_equal 2, s.results.size
+
+        assert_equal 2, s.results[:johns].size
+        assert_equal 1, s.results[1].size
+      end
+
+      should "iterate over mixed searches" do
+        s = Tire.multi_search 'multi-search-test-1' do
+          search(:johns) { query { match :_all, 'john' }  }
+          search         { query { match :_all, 'mary' }  }
+        end
+
+        assert_equal [:johns, 1], s.searches.names
+        assert_equal [:johns, 1], s.results.to_hash.keys
+
+        s.results.each_with_index do |results, i|
+          assert_equal 2, results.size if i == 0
+          assert_equal 1, results.size if i == 1
+        end
+
+        s.results.each_pair do |name, results|
+          assert_equal 2, results.size if name == :johns
+          assert_equal 1, results.size if name == 1
+        end
+      end
+
+      should "return results from different indices" do
+        s = Tire.multi_search do
+          search( index: 'multi-search-test-1' ) { query { match :_all, 'john' }  }
+          search( index: 'multi-search-test-2' ) { query { match :_all, 'john' }  }
+        end
+
+        assert_equal 2, s.results[0].size
+        assert_equal 1, s.results[1].size
+      end
+
+      should "return error for failed searches" do
+        s = Tire.multi_search 'multi-search-test-1' do
+          search() { query { match :_all, 'john' }  }
+          search() { query { string '[x' }  }
+        end
+
+        assert_equal 2, s.results[0].size
+        assert          s.results[0].success?
+
+        assert_equal 0, s.results[1].size
+        assert          s.results[1].failure?
+      end
+    end
+
+  end
+
+end