tire-erez 0.5.4

data/test/integration/prefix_query_test.rb

@@ -0,0 +1,43 @@
+require 'test_helper'
+
+module Tire
+
+  class PrefixQueryTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Prefix queries" do
+
+      should "search by a prefix" do
+        s = Tire.search('articles-test') do
+          query do
+            # "on" => "One"
+            prefix :title, "on"
+          end
+        end
+
+        assert_equal 1, s.results.size
+        assert_equal ['One'], s.results.map(&:title)
+      end
+
+      should "allow to specify boost" do
+        s = Tire.search('articles-test') do
+          query do
+            boolean do
+              # "on" => "One", boost it
+              should { prefix :title, "on", :boost => 2.0 }
+              should { all }
+            end
+          end
+          sort { by :_score }
+        end
+        
+        assert_equal 5, s.results.size
+        assert_equal 'One', s.results.first.title
+      
+      end
+
+    end
+
+  end
+
+end

data/test/integration/query_return_version_test.rb

@@ -0,0 +1,70 @@
+require 'test_helper'
+
+module Tire
+
+  class DslVersionIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "DSL Version" do
+
+      setup do
+        Tire.index 'articles-test-ids' do
+          delete
+          create
+
+          store :id => 1, :title => 'One'
+          store :id => 2, :title => 'Two'
+
+          refresh
+        end
+      end
+
+      teardown { Tire.index('articles-test-ids').delete }
+
+      should "returns actual version (non-nil) value for records when 'version' is true" do
+        s = Tire.search('articles-test-ids') do
+          version true
+          query { string 'One' }
+        end
+
+        assert_equal 1, s.results.count
+
+        document = s.results.first
+        assert_equal 'One', document.title
+        assert_equal 1,     document._version.to_i
+
+      end
+
+      should "returns a nil version field when 'version' is false" do
+        s = Tire.search('articles-test-ids') do
+          version false
+          query { string 'One' }
+        end
+
+        assert_equal 1, s.results.count
+
+        document = s.results.first
+        assert_equal 'One', document.title
+        assert_nil   document._version
+
+      end
+
+      should "returns a nil version field when 'version' is not included" do
+        s = Tire.search('articles-test-ids') do
+          query { string 'One' }
+        end
+
+        assert_equal 1, s.results.count
+
+        document = s.results.first
+        assert_equal 'One', document.title
+        assert_nil   document._version
+
+      end
+
+    end
+
+  end
+
+end
+

data/test/integration/query_string_test.rb

@@ -0,0 +1,52 @@
+require 'test_helper'
+
+module Tire
+
+  class QueryStringIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Searching for query string" do
+
+      should "find article by title" do
+        q = 'title:one'
+        assert_equal 1, string_query(q).results.count
+        assert_equal 'One', string_query(q).results.first[:title]
+      end
+
+      should "find articles by title with boosting" do
+        q = 'title:one^100 OR title:two'
+        assert_equal 2, string_query(q).results.count
+        assert_equal 'One', string_query(q).results.first[:title]
+      end
+
+      should "find articles by tags" do
+        q = 'tags:ruby AND tags:python'
+        assert_equal 1, string_query(q).results.count
+        assert_equal 'Two', string_query(q).results.first[:title]
+      end
+
+      should "find any article with tags" do
+        q = 'tags:ruby OR tags:python OR tags:java'
+        assert_equal 4, string_query(q).results.count
+      end
+
+      should "pass options to query definition" do
+        s = Tire.search 'articles-test' do
+          query do
+            string 'ruby python', :default_operator => 'AND'
+          end
+        end
+        assert_equal 1, s.results.count
+      end
+
+    end
+
+    private
+
+    def string_query(q)
+      Tire.search('articles-test') { query { string q } }
+    end
+
+  end
+
+end

data/test/integration/range_queries_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+
+module Tire
+
+  class RangeQueriesIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Range queries" do
+
+      should "allow simple range queries" do
+        s = Tire.search('articles-test') do
+          query do
+            range :words, { :gte => 250 }
+          end
+        end
+
+        assert_equal 3, s.results.size
+        assert_equal ['Two', 'Three', 'Four'].sort, s.results.map(&:title).sort
+      end
+
+      should "allow combined range queries" do
+        s = Tire.search('articles-test') do
+          query do
+            range :words, { :gte => 250, :lt => 375 }
+          end
+        end
+
+        assert_equal 2, s.results.size
+        assert_equal ['Two', 'Four'].sort, s.results.map(&:title).sort
+      end
+
+    end
+
+  end
+
+end

data/test/integration/reindex_test.rb

@@ -0,0 +1,56 @@
+require 'test_helper'
+
+module Tire
+
+  class ReindexIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Reindex" do
+      setup do
+        Tire.index('reindex-test-new').delete
+
+        documents = (1..100).map { |i| { id: i, type: 'test', title: "Document #{i}" } }
+
+        Tire.index 'reindex-test' do
+          delete
+          create :settings => { :number_of_shards => 1, :number_of_replicas => 0 }
+          import documents
+          refresh
+        end
+      end
+
+      teardown do
+        Index.new('reindex-test').delete
+        Index.new('reindex-test-new').delete
+      end
+
+      should "reindex the index into a new index with different settings" do
+        Tire.index('reindex-test').reindex 'reindex-test-new', settings: { number_of_shards: 3 }
+
+        Tire.index('reindex-test-new').refresh
+        assert_equal 100, Tire.search('reindex-test-new').results.total
+        assert_equal '3', Tire.index('reindex-test-new').settings['index.number_of_shards']
+      end
+
+      should "reindex a portion of an index into a new index" do
+        Tire.index('reindex-test').reindex('reindex-test-new') { query { string '10*' } }
+        
+        Tire.index('reindex-test-new').refresh
+        assert_equal 2, Tire.search('reindex-test-new').results.total
+      end
+
+      should "transform documents with a passed lambda" do
+        Tire.index('reindex-test').reindex 'reindex-test-new', transform: lambda { |document|
+                                                                            document[:title] += 'UPDATED'
+                                                                            document
+                                                                          }
+
+        Tire.index('reindex-test-new').refresh
+        assert_match /UPDATED/, Tire.search('reindex-test-new').results.first.title
+      end
+
+    end
+
+  end
+
+end

data/test/integration/results_test.rb

@@ -0,0 +1,58 @@
+require 'test_helper'
+
+module Tire
+
+  class ResultsIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Query results" do
+
+      should "allow easy access to returned documents" do
+        q = 'title:one'
+        s = Tire.search('articles-test') { query { string q } }
+        assert_equal 'One',  s.results.first.title
+        assert_equal 'ruby', s.results.first.tags[0]
+      end
+
+      should "allow easy access to returned documents with limited fields" do
+        q = 'title:one'
+        s = Tire.search('articles-test') { query { string q }.fields :title }
+        assert_equal 'One', s.results.first.title
+        assert_nil s.results.first.tags
+      end
+
+      should "allow to retrieve multiple fields" do
+        q = 'title:one'
+        s = Tire.search('articles-test') do
+          query { string q }
+          fields 'title', 'tags'
+        end
+        assert_equal 'One',  s.results.first.title
+        assert_equal 'ruby', s.results.first.tags[0]
+        assert_nil s.results.first.published_on
+      end
+
+      should "iterate results with hits" do
+        s = Tire.search('articles-test') { query { string 'title:one' } }
+
+        s.results.each_with_hit do |result, hit|
+          assert_instance_of Tire::Results::Item, result
+          assert_instance_of Hash, hit
+
+          assert_equal 'One', result.title
+          assert_equal 'One', hit['_source']['title']
+          assert_not_nil hit['_score']
+        end
+      end
+
+      should "be serialized to JSON" do
+        s = Tire.search('articles-test') { query { string 'title:one' } }
+
+        assert_not_nil s.results.as_json(only: 'title').first['title']
+        assert_nil     s.results.as_json(only: 'title').first['published_on']
+      end
+
+    end
+  end
+
+end

data/test/integration/scan_test.rb

@@ -0,0 +1,56 @@
+require 'test_helper'
+
+module Tire
+
+  class ScanIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Scan" do
+      setup do
+        documents = (1..100).map { |i| { id: i, type: 'test', title: "Document #{i}" } }
+
+        Tire.index 'scantest' do
+          delete
+          create :settings => { :number_of_shards => 1, :number_of_replicas => 0 }
+          import documents
+          refresh
+        end
+      end
+
+      teardown { Index.new('scantest').delete }
+
+      should "iterate over batches of documents" do
+        count = 0
+
+        s = Tire.scan 'scantest'
+        s.each { |results| count += 1 }
+
+        assert_equal 10, count
+      end
+
+      should "iterate over individual documents" do
+        count = 0
+
+        s = Tire.scan 'scantest'
+        s.each_document { |results| count += 1 }
+
+        assert_equal 100, count
+      end
+
+      should "limit the returned results by query" do
+        count = 0
+
+        s = Tire.scan('scantest') { query { string '10*' } }
+        s.each do |results|
+          count += 1
+          assert_equal ['Document 10', 'Document 100'], results.map(&:title)
+        end
+
+        assert_equal 1, count
+      end
+
+    end
+
+  end
+
+end

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,52 @@
+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
+
+      should "sort by multiple fields" do
+        q = '*'
+        s = Tire.search('articles-test') do
+          query { string q }
+          sort  do
+            by :words, :desc
+            by :title, :asc
+          end
+        end
+
+        # p s.results.to_a.map { |i| [i.title, i.sort] }
+        assert_equal 'Three', s.results[0][:title]
+        assert_equal 'Four',  s.results[1][:title]
+        assert_equal 'Two',   s.results[2][:title]
+      end
+
+    end
+
+  end
+
+end