tire 0.1.0

data/lib/tire/version.rb

@@ -0,0 +1,3 @@
+module Tire
+  VERSION = "0.1.0"
+end

data/test/fixtures/articles/1.json

@@ -0,0 +1 @@
+{"title" : "One", "tags" : ["ruby"], "published_on" : "2011-01-01"}

data/test/fixtures/articles/2.json

@@ -0,0 +1 @@
+{"title" : "Two", "tags" : ["ruby", "python"], "published_on" : "2011-01-02"}

data/test/fixtures/articles/3.json

@@ -0,0 +1 @@
+{"title" : "Three", "tags" : ["java"], "published_on" : "2011-01-02"}

data/test/fixtures/articles/4.json

@@ -0,0 +1 @@
+{"title" : "Four", "tags" : ["erlang"], "published_on" : "2011-01-03"}

data/test/fixtures/articles/5.json

@@ -0,0 +1 @@
+{"title" : "Five", "tags" : ["javascript", "java"], "published_on" : "2011-01-04"}

data/test/integration/active_model_searchable_test.rb

@@ -0,0 +1,80 @@
+require 'test_helper'
+
+module Tire
+
+  class ActiveModelSearchableIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    def setup
+      super
+      SupermodelArticle.delete_all
+      @model = SupermodelArticle.new :title => 'Test'
+    end
+
+    def teardown
+      super
+      SupermodelArticle.delete_all
+    end
+
+    context "ActiveModel" do
+
+      setup    do
+        Tire.index('supermodel_articles').delete
+        load File.expand_path('../../models/supermodel_article.rb', __FILE__)
+      end
+      teardown { Tire.index('supermodel_articles').delete }
+
+      should "configure mapping" do
+        assert_equal 'czech', SupermodelArticle.mapping[:title][:analyzer]
+        assert_equal 15,      SupermodelArticle.mapping[:title][:boost]
+
+        assert_equal 'czech', SupermodelArticle.index.mapping['supermodel_article']['properties']['title']['analyzer']
+      end
+
+      should "save document into index on save and find it with score" do
+        a = SupermodelArticle.new :title => 'Test'
+        a.save
+        id = a.id
+
+        a.index.refresh
+        sleep(1.5)
+
+        results = SupermodelArticle.search 'test'
+
+        assert_equal 1, results.count
+        assert_instance_of SupermodelArticle, results.first
+        assert_equal       'Test', results.first.title
+        assert_not_nil     results.first.score
+        assert_equal       id, results.first.id
+      end
+
+      should "remove document from index on destroy" do
+        a = SupermodelArticle.new :title => 'Test'
+        a.save
+        a.destroy
+
+        a.index.refresh
+        sleep(1.25)
+
+        results = SupermodelArticle.search 'test'
+        
+        assert_equal 0, results.count
+      end
+
+      should "retrieve sorted documents by IDs returned from search" do
+        SupermodelArticle.create! :title => 'foo'
+        SupermodelArticle.create! :title => 'bar'
+
+        SupermodelArticle.index.refresh
+        results = SupermodelArticle.search 'foo OR bar^100'
+
+        assert_equal 2, results.count
+
+        assert_equal 'bar', results.first.title
+      end
+
+    end
+
+  end
+
+end

data/test/integration/active_record_searchable_test.rb

@@ -0,0 +1,193 @@
+require 'test_helper'
+
+module Tire
+
+  class ActiveRecordSearchableIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    def setup
+      super
+      File.delete fixtures_path.join('articles.db') rescue nil
+
+      ActiveRecord::Base.establish_connection( :adapter => 'sqlite3', :database => fixtures_path.join('articles.db') )
+
+      ActiveRecord::Migration.verbose = false
+      ActiveRecord::Schema.define(:version => 1) do
+        create_table :active_record_articles do |t|
+          t.string   :title
+          t.datetime :created_at, :default => 'NOW()'
+        end
+      end
+    end
+
+    def teardown
+      super
+      File.delete fixtures_path.join('articles.db') rescue nil
+    end
+
+    context "ActiveRecord integration" do
+
+      setup    do
+        Tire.index('active_record_articles').delete
+        load File.expand_path('../../models/active_record_article.rb', __FILE__)
+      end
+      teardown { Tire.index('active_record_articles').delete }
+
+      should "configure mapping" do
+        assert_equal 'snowball', ActiveRecordArticle.mapping[:title][:analyzer]
+        assert_equal 10, ActiveRecordArticle.mapping[:title][:boost]
+
+        assert_equal 'snowball', ActiveRecordArticle.index.mapping['active_record_article']['properties']['title']['analyzer']
+      end
+
+      should "save document into index on save and find it" do
+        a = ActiveRecordArticle.new :title => 'Test'
+        a.save!
+        id = a.id
+
+        a.index.refresh
+        sleep(1.5) # Leave ES some breathing room here...
+
+        results = ActiveRecordArticle.search 'test'
+
+        assert_equal 1, results.count
+
+        assert_instance_of ActiveRecordArticle, results.first
+        assert_not_nil results.first.id
+        assert_equal   id, results.first.id
+        assert         results.first.persisted?, "Record should be persisted"
+        assert_not_nil results.first._score
+        assert_equal   'Test', results.first.title
+      end
+
+      should "remove document from index on destroy" do
+        a = ActiveRecordArticle.new :title => 'Test'
+        a.save!
+        a.destroy
+
+        a.index.refresh
+        results = ActiveRecordArticle.search 'test'
+        
+        assert_equal 0, results.count
+      end
+
+      should "return documents with scores" do
+        ActiveRecordArticle.create! :title => 'foo'
+        ActiveRecordArticle.create! :title => 'bar'
+
+        ActiveRecordArticle.index.refresh
+        results = ActiveRecordArticle.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| ActiveRecordArticle.create :title => "Test#{number}" }
+          ActiveRecordArticle.index.refresh
+        end
+
+        context "and parameter searches" do
+
+          should "find first page with five results" do
+            results = ActiveRecordArticle.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 0, 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 = ActiveRecordArticle.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 3, results.next_page
+
+            assert_equal 'Test6', results.first.title
+          end
+
+          should "find not find missing page" do
+            results = ActiveRecordArticle.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 4, 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 = ActiveRecordArticle.search nil, :per_page => 5, :page => 1 do |search|
+              search.query { |query| query.string @q }
+              search.sort  { 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 0, 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 = ActiveRecordArticle.search nil, :per_page => 5, :page => 2 do |search|
+              search.query { |query| query.string @q }
+              search.sort  { 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 3, results.next_page
+
+            assert_equal 'Test6', results.first.title
+          end
+
+          should "find not find missing page" do
+            results = ActiveRecordArticle.search nil, :per_page => 5, :page => 3 do |search|
+              search.query { |query| query.string @q }
+              search.sort  { 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 4, results.next_page
+
+            assert_nil results.first
+          end
+
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/test/integration/facets_test.rb

@@ -0,0 +1,65 @@
+require 'test_helper'
+require 'date'
+
+module Tire
+
+  class FacetsIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Facets" do
+
+      should "return results scoped to current query" do
+        q = 'tags:ruby'
+        s = Tire.search('articles-test') do
+          query { string q }
+          facet 'tags' do
+            terms :tags
+          end
+        end
+        facets = s.results.facets['tags']['terms']
+        assert_equal 2, facets.count
+        assert_equal 'ruby', facets.first['term']
+        assert_equal 2,      facets.first['count']
+      end
+
+      should "allow to specify global facets and query-scoped facets" do
+        q = 'tags:ruby'
+        s = Tire.search('articles-test') do
+          query { string q }
+          facet 'scoped-tags' do
+            terms :tags
+          end
+          facet 'global-tags', :global => true do
+            terms :tags
+          end
+        end
+
+        scoped_facets = s.results.facets['scoped-tags']['terms']
+        global_facets = s.results.facets['global-tags']['terms']
+
+        assert_equal 2, scoped_facets.count
+        assert_equal 5, global_facets.count
+      end
+
+      context "date histogram" do
+
+        should "return aggregated values for all results" do
+          s = Tire.search('articles-test') do
+            query { all }
+            facet 'published_on' do
+              date :published_on
+            end
+          end
+
+          facets = s.results.facets['published_on']['entries']
+          assert_equal 4, facets.size, facets.inspect
+          assert_equal 2, facets.entries[1]["count"], facets.inspect
+        end
+
+      end
+
+    end
+
+  end
+
+end

data/test/integration/filters_test.rb

@@ -0,0 +1,46 @@
+require 'test_helper'
+
+module Tire
+
+  class FiltersIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Filters" do
+
+      should "filter the results" do
+        s = Tire.search('articles-test') do
+          query { string 'title:T*' }
+          filter :terms, :tags => ['ruby']
+        end
+
+        assert_equal 1, s.results.count
+        assert_equal 'Two', s.results.first.title
+      end
+
+      should "filter the results with multiple filters" do
+        s = Tire.search('articles-test') do
+          query { string 'title:F*' }
+          filter :or, {:terms => {:tags => ['ruby']}},
+                      {:terms => {:tags => ['erlang']}}
+        end
+
+        assert_equal 1, s.results.count
+        assert_equal 'Four', s.results.first.title
+      end
+
+      should "not influence facets" do
+        s = Tire.search('articles-test') do
+          query { string 'title:T*' }
+          filter :terms, :tags => ['ruby']
+          facet('tags') { terms :tags }
+        end
+
+        assert_equal 1, s.results.count
+        assert_equal 3, s.results.facets['tags'].size
+      end
+
+    end
+
+  end
+
+end

data/test/integration/highlight_test.rb

@@ -0,0 +1,52 @@
+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 "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
+        sleep(0.5)
+
+        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