tire-erez 0.5.4

data/test/integration/fuzzy_queries_test.rb

@@ -0,0 +1,20 @@
+require 'test_helper'
+
+module Tire
+
+  class FuzzyQueryIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Fuzzy query" do
+      should "fuzzily find article by tag" do
+        results = Tire.search('articles-test') { query { fuzzy :tags, 'irlang' } }.results
+
+        assert_equal 1,        results.count
+        assert_equal ["erlang"], results.first[:tags]
+      end
+
+    end
+
+  end
+
+end

data/test/integration/highlight_test.rb

@@ -0,0 +1,64 @@
+require 'test_helper'
+
+module Tire
+
+  class HighlightIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Highlight" do
+      teardown { Tire.index('highlight-test').delete }
+
+      should "add 'highlight' field to the result item" do
+        # Tire::Configuration.logger STDERR, :level => 'debug'
+        s = Tire.search('articles-test') do
+          query { string 'Two' }
+          highlight :title
+        end
+
+        doc = s.results.first
+
+        assert_equal 1, doc.highlight.title.size
+        assert doc.highlight.title.to_s.include?('<em>'), "Highlight does not include default highlight tag"
+      end
+
+      should "highlight multiple fields with custom highlight tag" do
+        s = Tire.search('articles-test') do
+          query { string 'Two OR ruby' }
+          highlight :tags, :title, :options => { :tag => '<strong>' }
+        end
+
+        doc = s.results.first
+
+        assert_equal 1, doc.highlight.title.size
+        assert_equal "<strong>Two</strong>", doc.highlight.title.first, "Highlight does not include highlight tag"
+        assert_equal "<strong>ruby</strong>", doc.highlight.tags.first, "Highlight does not include highlight tag"
+      end
+
+      should "return entire content with highlighted fragments" do
+        # Tire::Configuration.logger STDERR, :level => 'debug'
+
+        content = "A Fox one day fell into a deep well and could find no means of escape. A Goat, overcome with thirst, came to the same well, and seeing the Fox, inquired if the water was good. Concealing his sad plight under a merry guise, the Fox indulged in a lavish praise of the water, saying it was excellent beyond measure, and encouraging him to descend. The Goat, mindful only of his thirst, thoughtlessly jumped down, but just as he drank, the Fox informed him of the difficulty they were both in and suggested a scheme for their common escape. \"If,\" said he, \"you will place your forefeet upon the wall and bend your head, I will run up your back and escape, and will help you out afterwards.\" The Goat readily assented and the Fox leaped upon his back. Steadying himself with the Goat horns, he safely reached the mouth of the well and made off as fast as he could. When the Goat upbraided him for breaking his promise, he turned around and cried out, \"You foolish old fellow! If you had as many brains in your head as you have hairs in your beard, you would never have gone down before you had inspected the way up, nor have exposed yourself to dangers from which you had no means of escape.\" Look before you leap."
+
+        Tire.index 'highlight-test' do
+          delete
+          create
+          store   :id => 1, :content => content
+          refresh
+        end
+
+        s = Tire.search('highlight-test') do
+          query { string 'fox' }
+          highlight :content => { :number_of_fragments => 0 }
+        end
+
+        doc = s.results.first
+        assert_not_nil doc.highlight.content
+
+        highlight = doc.highlight.content
+        assert highlight.to_s.include?('<em>'), "Highlight does not include default highlight tag"
+      end
+
+    end
+
+  end
+end

data/test/integration/index_aliases_test.rb

@@ -0,0 +1,122 @@
+require 'test_helper'
+
+require 'active_support/core_ext/numeric'
+require 'active_support/core_ext/date/calculations'
+
+module Tire
+
+  class IndexAliasesIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "With a filtered alias" do
+      setup do
+
+        @index = Tire.index 'index-original' do
+          delete
+          create
+        end
+        
+      end
+
+      teardown { Tire.index('index-original').delete }
+      
+       should "create the alias" do
+         @index.add_alias 'index-aliased'
+         assert_equal 1, @index.aliases.size
+       end
+      
+       should "find only portion of documents in the filtered alias" do
+         Tire.index 'index-original' do
+           add_alias 'index-aliased', :filter => { :term => { :user => 'anne' } }
+           store :title => 'Document 1', :user => 'anne'
+           store :title => 'Document 2', :user => 'mary'
+      
+           refresh
+         end
+      
+         assert_equal 2, Tire.search('index-original') { query { all } }.results.size
+         assert_equal 1, Tire.search('index-aliased')  { query { all } }.results.size
+       end
+      
+       should "remove the alias" do
+         @index.add_alias 'index-aliased'
+         assert_equal 1, @index.aliases.size
+      
+         @index.remove_alias 'index-aliased'
+         assert_equal 0, @index.aliases.size
+      
+         assert_raise Tire::Search::SearchRequestFailed do
+           Tire.search('index-aliased')  { query { all } }.results
+         end
+       end
+      
+       should "retrieve a list of aliases for an index" do
+         @index.add_alias 'index-aliased'
+      
+         assert_equal ['index-aliased'], @index.aliases.map(&:name)
+       end
+ 
+      should "retrieve the properties of an alias" do
+        @index.add_alias 'index-aliased', :routing => '1'
+
+        assert_equal '1', @index.aliases('index-aliased').search_routing
+      end
+    end
+
+    context "In the 'sliding window' scenario" do
+
+      setup do
+        WINDOW_SIZE_IN_WEEKS = 4
+
+        @indices = WINDOW_SIZE_IN_WEEKS.times.map { |number| "articles_#{number.weeks.ago.strftime('%Y-%m-%d')}" }
+
+        @indices.each_with_index do |name,i|
+          Tire.index(name) do
+            delete
+            create
+            store   :title => "Document #{i}"
+            refresh
+          end
+          Alias.new(:name => "articles_current") { |a| a.indices(name) and a.save }
+        end
+      end
+
+      teardown do
+        @indices.each { |index| Tire.index(index).delete }
+      end
+
+      should "add a new index to alias" do
+        @indices << "articles_#{(WINDOW_SIZE_IN_WEEKS+1).weeks.ago.strftime('%Y-%m-%d')}"
+        Tire.index(@indices.last).create
+        Alias.new(:name => "articles_current") { |a| a.index @indices.last and a.save }
+
+        a = Alias.find("articles_current")
+        assert_equal 5, a.indices.size
+      end
+
+      should "remove the stale index from the alias" do
+        Alias.find("articles_current") do |a|
+          # Remove all indices older then 2 weeks from the alias
+          a.indices.delete_if do |i|
+            Time.parse( i.gsub(/articles_/, '') ) < 2.weeks.ago rescue false
+          end
+          a.save
+        end
+
+        assert_equal 2, Alias.find("articles_current").indices.size
+      end
+
+      should "search within the alias" do
+        Alias.find("articles_current") do |a|
+          a.indices.clear and a.indices @indices[0..1] and a.save
+        end
+
+        assert_equal 4, Tire.search(@indices) { query {all} }.results.size
+        assert_equal 2, Tire.search("articles_current") { query {all} }.results.size
+      end
+
+    end
+
+  end
+
+end

data/test/integration/index_mapping_test.rb

@@ -0,0 +1,43 @@
+require 'test_helper'
+
+module Tire
+
+  class IndexMappingIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Default mapping" do
+      teardown { Tire.index('mapped-index').delete; sleep 0.1 }
+
+      should "create and return the default mapping" do
+
+        index = Tire.index 'mapped-index' do
+          create
+          store :type => :article, :title => 'One'
+          refresh
+          sleep 1
+        end
+
+        assert_equal 'string', index.mapping['article']['properties']['title']['type'],  index.mapping.inspect
+        assert_nil             index.mapping['article']['properties']['title']['boost'], index.mapping.inspect
+      end
+    end
+
+    context "Creating index with mapping" do
+      teardown { Tire.index('mapped-index').delete; sleep 0.1 }
+    
+      should "create the specified mapping" do
+    
+        index = Tire.index 'mapped-index' do
+          create :mappings => { :article => { :properties => { :title => { :type => 'string', :boost => 2.0, :store => 'yes' } } } }
+        end
+
+        # p index.mapping    
+        assert_equal 2.0,   index.mapping['article']['properties']['title']['boost'],  index.mapping.inspect
+        assert_equal 'yes', index.mapping['article']['properties']['title']['store'],  index.mapping.inspect
+    
+      end
+    end
+
+  end
+
+end

data/test/integration/index_store_test.rb

@@ -0,0 +1,112 @@
+require 'test_helper'
+
+module Tire
+
+  class IndexStoreIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Storing the documents in index" do
+
+      setup do
+        Tire.index 'articles-test-ids' do
+          delete
+          create
+
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+          store :id => 3, :title => 'Three'
+          store :id => 4, :title => 'Four'
+          store :id => 4, :title => 'Four'
+
+          refresh
+        end
+      end
+
+      teardown do
+        Tire.index('articles-test-ids').delete
+        Tire.index('articles-test-types').delete
+      end
+
+      should "happen in existing index" do
+        assert   Tire.index("articles-test-ids").exists?
+        assert ! Tire.index("four-oh-four-index").exists?
+      end
+
+      should "store hashes under their IDs" do
+        s = Tire.search('articles-test-ids') { query { string '*' } }
+
+        assert_equal 4, s.results.count
+
+        document = Tire.index('articles-test-ids').retrieve :document, 4
+        assert_equal 'Four', document.title
+        assert_equal 2,      document._version.to_i
+
+      end
+
+      should "store documents as proper types" do
+        Tire.index 'articles-test-types' do
+          delete
+          create
+          store :type => 'my_type', :title => 'One'
+          refresh
+        end
+
+        s = Tire.search('articles-test-types/my_type') { query { all } }
+        assert_equal 1, s.results.count
+        assert_equal 'my_type', s.results.first.type
+      end
+
+    end
+
+    context "Removing documents from the index" do
+
+      teardown { Tire.index('articles-test-remove').delete }
+
+      setup do
+        Tire.index 'articles-test-remove' do
+          delete
+          create
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+          refresh
+        end
+      end
+
+      should "remove document from the index" do
+        assert_equal 2, Tire.search('articles-test-remove') { query { string '*' } }.results.count
+
+        assert_nothing_raised do
+          assert Tire.index('articles-test-remove').remove 1
+          assert ! Tire.index('articles-test-remove').remove(1)
+        end
+      end
+
+    end
+
+    context "Retrieving documents from the index" do
+
+      teardown { Tire.index('articles-test-retrieve').delete }
+
+      setup do
+        Tire.index 'articles-test-retrieve' do
+          delete
+          create
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+          refresh
+        end
+      end
+
+      should "retrieve document from the index" do
+        assert_instance_of Tire::Results::Item, Tire.index('articles-test-retrieve').retrieve(:document, 1)
+      end
+
+      should "return nil when retrieving missing document" do
+        assert_nil Tire.index('articles-test-retrieve').retrieve :document, 4
+      end
+
+    end
+
+  end
+
+end

data/test/integration/index_update_document_test.rb

@@ -0,0 +1,121 @@
+require 'test_helper'
+
+module Tire
+
+  class IndexUpdateDocumentIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Updating a document" do
+
+      setup do
+        Tire.index 'articles-with-tags' do
+          delete
+          create
+
+          store :type => 'article', :id => 1, :title => 'One',   :tags => ['foo'],        :views => 0
+          store :type => 'article', :id => 2, :title => 'Two',   :tags => ['foo', 'bar'], :views => 10
+          store :type => 'article', :id => 3, :title => 'Three', :tags => ['foobar']
+
+          refresh
+        end
+      end
+
+      teardown { Tire.index('articles-with-tags').delete }
+
+      should "increment a counter" do
+        Tire.index('articles-with-tags') { update( 'article', '1', :script => "ctx._source.views += 1" ) and refresh }
+
+        document = Tire.search('articles-with-tags') { query { string 'title:one' } }.results.first
+        assert_equal 1, document.views, document.inspect
+
+        Tire.index('articles-with-tags') { update( 'article', '2', :script => "ctx._source.views += 1" ) and refresh }
+
+        document = Tire.search('articles-with-tags') { query { string 'title:two' } }.results.first
+        assert_equal 11, document.views, document.inspect
+      end
+
+      should "add a tag to document" do
+        Tire.index('articles-with-tags') do
+          update 'article', '1', :script => "ctx._source.tags += tag",
+                                 :params => { :tag => 'new' }
+          refresh
+        end
+
+        document = Tire.search('articles-with-tags') { query { string 'title:one' } }.results.first
+        assert_equal ['foo', 'new'], document.tags, document.inspect
+      end
+
+      should "remove a tag from document" do
+        Tire.index('articles-with-tags') do
+          update 'article', '1', :script => "ctx._source.tags = tags",
+                                 :params => { :tags => [] }
+          refresh
+        end
+
+        document = Tire.index('articles-with-tags').retrieve 'article', '1'
+        assert_equal [], document.tags, document.inspect
+      end
+
+      should "remove the entire document if specific condition is met" do
+        Tire.index('articles-with-tags') do
+          # Remove document when it contains tag 'foobar'
+          update 'article', '3', :script => "ctx._source.tags.contains(tag) ? ctx.op = 'delete' : 'none'",
+                                 :params => { :tag => 'foobar' }
+          refresh
+        end
+
+        assert_nil Tire.index('articles-with-tags').retrieve 'article', '3'
+      end
+
+      should "pass the operation parameters to the API" do
+        Tire.index('articles-with-tags').update 'article', '2', { :script => "ctx._source.tags += tag",
+                                                                  :params => { :tag => 'new' }
+                                                                },
+                                                                {
+                                                                  :refresh => true
+                                                                }
+
+        document = Tire.search('articles-with-tags') { query { string 'title:two' } }.results.first
+        assert_equal 3, document.tags.size, document.inspect
+      end
+
+      should "update the document with a partial one" do
+        Tire.index('articles-with-tags') do
+          update( 'article', '1', :doc => { :title => 'One UPDATED' } )
+          refresh
+        end
+
+        document = Tire.search('articles-with-tags') { query { string 'title:one' } }.results.first
+        assert_equal 'One UPDATED', document.title, document.inspect
+      end
+
+      should "access variables from the outer scope" do
+        $t = self
+
+        class Updater
+          @tags = ['foo', 'bar', 'baz']
+
+          def self.perform_update!
+            $t.assert_not_nil @tags
+
+            Tire.index('articles-with-tags') do |index|
+              $t.assert_not_nil @tags
+
+              index.update 'article', '3', :script => "ctx._source.tags = tags",
+                                     :params => { :tags => @tags }
+              index.refresh
+            end
+          end
+        end
+
+        Updater.perform_update!
+
+        document = Tire.search('articles-with-tags') { query { string 'title:three' } }.results.first
+        assert_equal 3, document.tags.size, document.inspect
+      end
+
+    end
+
+  end
+
+end