ssickles-tire 0.4.2

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

data/test/integration/sort_test.rb

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

data/test/integration/text_query_test.rb

@@ -0,0 +1,39 @@
+require 'test_helper'
+
+module Tire
+
+  class QueryStringIntegrationTest < Test::Unit::TestCase
+    include Test::Integration
+
+    context "Text query" do
+      setup do
+        Tire.index('articles-test') do
+          store :type => 'article', :title => '+1 !!!'
+          store :type => 'article', :title => 'Furry Kitten'
+          refresh
+        end
+      end
+
+      should "find article by title" do
+        results = Tire.search('articles-test') do
+          query { text :title, '+1' }
+        end.results
+
+        assert_equal 1,        results.count
+        assert_equal "+1 !!!", results.first[:title]
+      end
+
+      should "allow to pass options (fuzziness)" do
+        results = Tire.search('articles-test') do
+          query { text :title, 'fuzzy mitten', :fuzziness => 0.5, :operator => 'and' }
+        end.results
+
+        assert_equal 1,        results.count
+        assert_equal "Furry Kitten", results.first[:title]
+      end
+
+    end
+
+  end
+
+end

data/test/models/active_model_article.rb

@@ -0,0 +1,31 @@
+# Example ActiveModel class
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticle
+
+  extend  ActiveModel::Naming
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+
+  include Tire::Model::Search
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def id;         attributes['id'] || attributes['_id']; end
+  def id=(value); attributes['id'] = value;              end
+
+  def method_missing(name, *args, &block)
+    attributes[name.to_sym] || attributes[name.to_s] || super
+  end
+
+  def persisted?; true; end
+  def save;       true; end
+
+end

data/test/models/active_model_article_with_callbacks.rb

@@ -0,0 +1,49 @@
+# Example ActiveModel class with callbacks
+
+require 'rubygems'
+require 'active_model'
+
+class ActiveModelArticleWithCallbacks
+
+  include ActiveModel::AttributeMethods
+  include ActiveModel::Validations
+  include ActiveModel::Serialization
+  include ActiveModel::Serializers::JSON
+  include ActiveModel::Naming
+
+  extend  ActiveModel::Callbacks
+  define_model_callbacks :save, :destroy
+
+  include Tire::Model::Search
+  include Tire::Model::Callbacks
+
+  attr_reader :attributes
+
+  def initialize(attributes = {})
+    @attributes = attributes
+  end
+
+  def method_missing(id, *args, &block)
+    attributes[id.to_sym] || attributes[id.to_s] || super
+  end
+
+  def persisted?
+    true
+  end
+
+  def save
+    _run_save_callbacks do
+      STDERR.puts "[Saving ...]"
+    end
+  end
+
+  def destroy
+    _run_destroy_callbacks do
+      STDERR.puts "[Destroying ...]"
+      @destroyed = true
+    end
+  end
+
+  def destroyed?; !!@destroyed; end
+
+end