load_balanced_tire 0.1

data/test/integration/persistent_model_test.rb

@@ -0,0 +1,130 @@
+require 'test_helper'
+
+module Tire
+
+  class PersistentModelIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    def setup
+      super
+      PersistentArticle.index.delete
+    end
+
+    def teardown
+      super
+      PersistentArticle.index.delete
+      PersistentArticleWithDefaults.index.delete
+    end
+
+    context "PersistentModel" do
+      should "search with simple query" do
+        PersistentArticle.create :id => 1, :title => 'One'
+        PersistentArticle.index.refresh
+
+        results = PersistentArticle.search 'one'
+        assert_equal 'One', results.first.title
+      end
+
+      should "search with a block" do
+        PersistentArticle.create :id => 1, :title => 'One'
+        PersistentArticle.index.refresh
+
+        results = PersistentArticle.search(:sort => 'title') { query { string 'one' } }
+        assert_equal 'One', results.first.title
+      end
+
+      should "return instances of model" do
+        PersistentArticle.create :id => 1, :title => 'One'
+        PersistentArticle.index.refresh
+
+        results = PersistentArticle.search 'one'
+        assert_instance_of PersistentArticle, results.first
+      end
+
+      should "save documents into index and find them by IDs" do
+        one = PersistentArticle.create :id => 1, :title => 'One'
+        two = PersistentArticle.create :id => 2, :title => 'Two'
+
+        PersistentArticle.index.refresh
+
+        results = PersistentArticle.find [1, 2]
+
+        assert_equal 2, results.size
+      end
+
+      should "be persisted" do
+        one = PersistentArticle.create :id => 1, :title => 'One'
+        PersistentArticle.index.refresh
+
+        a = PersistentArticle.all.first
+        assert a.persisted?, a.inspect
+
+        b = PersistentArticle.first
+        assert b.persisted?, b.inspect
+
+        c = PersistentArticle.search { query { string 'one' } }.first
+        assert c.persisted?, c.inspect
+      end
+
+      should "return default values for properties without value" do
+        PersistentArticleWithDefaults.create :id => 1, :title => 'One'
+        PersistentArticleWithDefaults.index.refresh
+
+        results = PersistentArticleWithDefaults.all
+
+        assert_equal [], results.first.tags
+      end
+
+      context "with pagination" do
+
+        setup do
+          1.upto(9) { |number| PersistentArticle.create :title => "Test#{number}" }
+          PersistentArticle.index.refresh
+        end
+
+        should "find first page with five results" do
+          results = PersistentArticle.search( :per_page => 5, :page => 1 ) { query { all } }
+          assert_equal 5, results.size
+
+          # WillPaginate
+          #
+          assert_equal 2, results.total_pages
+          assert_equal 1, results.current_page
+          assert_equal nil, results.previous_page
+          assert_equal 2, results.next_page
+
+          # Kaminari
+          #
+          assert_equal 5, results.limit_value
+          assert_equal 9, results.total_count
+          assert_equal 2, results.num_pages
+          assert_equal 0, results.offset_value
+        end
+
+      end
+
+      context "with namespaced models" do
+        setup do
+          MyNamespace::PersistentArticleInNamespace.create :title => 'Test'
+          MyNamespace::PersistentArticleInNamespace.index.refresh
+        end
+
+        teardown do
+          MyNamespace::PersistentArticleInNamespace.index.delete
+        end
+
+        should "find the document in the index" do
+          results = MyNamespace::PersistentArticleInNamespace.search 'test'
+
+          assert       results.any?, "No results returned: #{results.inspect}"
+          assert_equal 1, results.count
+
+          assert_instance_of MyNamespace::PersistentArticleInNamespace, results.first
+        end
+
+      end
+
+    end
+
+  end
+end

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,46 @@
+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
+
+    end
+
+  end
+
+end

data/test/integration/results_test.rb

@@ -0,0 +1,39 @@
+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
+
+    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