elasticsearch-dsl 0.0.0 → 0.1.0

data/test/integration/search_options_test.rb

@@ -0,0 +1,28 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class SearchOptionsIntegrationTest < ::Elasticsearch::Test::IntegrationTestCase
+      include Elasticsearch::DSL::Search
+
+      context "Search options" do
+        setup do
+          @client.indices.create index: 'test'
+          @client.index index: 'test', type: 'd', id: '1', body: { title: 'Test' }
+          @client.index index: 'test', type: 'd', id: '2', body: { title: 'Rest' }
+          @client.indices.refresh index: 'test'
+        end
+
+        should "explain the match" do
+          response = @client.search index: 'test', body: search {
+            query   { match title: 'test' }
+            explain true
+          }.to_hash
+
+          assert_equal 1, response['hits']['total']
+          assert_not_nil  response['hits']['hits'][0]['_explanation']
+        end
+      end
+    end
+  end
+end

data/test/integration/search_query_test.rb

@@ -0,0 +1,55 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class QueryIntegrationTest < ::Elasticsearch::Test::IntegrationTestCase
+      include Elasticsearch::DSL::Search
+
+      context "Queries integration" do
+        startup do
+          Elasticsearch::Extensions::Test::Cluster.start(nodes: 1) if ENV['SERVER'] and not Elasticsearch::Extensions::Test::Cluster.running?
+        end
+
+        setup do
+          @client.indices.create index: 'test'
+          @client.index index: 'test', type: 'd', id: '1', body: { title: 'Test', tags: ['one'] }
+          @client.index index: 'test', type: 'd', id: '2', body: { title: 'Rest', tags: ['one', 'two'] }
+          @client.indices.refresh index: 'test'
+        end
+
+        context "for match query" do
+          should "find the document" do
+            response = @client.search index: 'test', body: search { query { match title: 'test' } }.to_hash
+
+            assert_equal 1, response['hits']['total']
+          end
+        end
+
+        context "for query_string query" do
+          should "find the document" do
+            response = @client.search index: 'test', body: search { query { query_string { query 'te*' } } }.to_hash
+
+            assert_equal 1, response['hits']['total']
+          end
+        end
+
+        context "for the bool query" do
+          should "find the document" do
+            response = @client.search index: 'test', body: search {
+                query do
+                  bool do
+                    must   { terms tags: ['one'] }
+                    should { match title: 'Test' }
+                  end
+                end
+              }.to_hash
+
+            assert_equal 2, response['hits']['total']
+            assert_equal 'Test', response['hits']['hits'][0]['_source']['title']
+          end
+        end
+
+      end
+    end
+  end
+end

data/test/integration/search_size_from_test.rb

@@ -0,0 +1,41 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class SizeIntegrationTest < ::Elasticsearch::Test::IntegrationTestCase
+      include Elasticsearch::DSL::Search
+
+        context "Search results pagination" do
+          setup do
+            25.times { |i| @client.index index: 'test', type: 'd', id: i, body: { title: "Test #{sprintf('%03d', i)}" } }
+            @client.indices.refresh index: 'test'
+          end
+
+          should "find the correct number of documents" do
+            response = @client.search index: 'test', body: search {
+              query { match title: 'test' }
+              size 15
+            }.to_hash
+
+            assert_equal 25, response['hits']['total']
+            assert_equal 15, response['hits']['hits'].size
+          end
+
+          should "move the offset" do
+            response = @client.search index: 'test', body: search {
+              query { match(:title) { query 'test' and type 'phrase_prefix' } }
+              size 5
+              from 5
+              sort { by :title }
+            }.to_hash
+
+            assert_equal 25, response['hits']['total']
+            assert_equal 5,  response['hits']['hits'].size
+            assert_equal 'Test 005', response['hits']['hits'][0]['_source']['title']
+            assert_equal 'Test 009', response['hits']['hits'][4]['_source']['title']
+          end
+        end
+
+    end
+  end
+end

data/test/integration/search_sort_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class SortingIntegrationTest < ::Elasticsearch::Test::IntegrationTestCase
+      include Elasticsearch::DSL::Search
+
+      context "Sorting integration" do
+        startup do
+          Elasticsearch::Extensions::Test::Cluster.start(nodes: 1) if ENV['SERVER'] and not Elasticsearch::Extensions::Test::Cluster.running?
+        end
+
+        setup do
+          @client.indices.create index: 'test'
+          @client.index index: 'test', type: 'd', id: '1', body: { tags: ['one'], clicks: 5 }
+          @client.index index: 'test', type: 'd', id: '2', body: { tags: ['one', 'two'], clicks: 15 }
+          @client.index index: 'test', type: 'd', id: '3', body: { tags: ['one', 'three'], clicks: 20 }
+          @client.indices.refresh index: 'test'
+        end
+
+        context "sorting by clicks" do
+          should "return documents in order" do
+            response = @client.search index: 'test', body: search {
+              sort do
+                by :clicks, order: 'desc'
+              end
+            }.to_hash
+
+            assert_same_elements ['3', '2', '1'], response['hits']['hits'].map { |d| d['_id'] }
+          end
+        end
+
+      end
+    end
+  end
+end

data/test/integration/search_suggest_test.rb

@@ -0,0 +1,62 @@
+# encoding: UTF-8
+
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    class SuggestIntegrationTest < ::Elasticsearch::Test::IntegrationTestCase
+      include Elasticsearch::DSL::Search
+
+      context "Suggest integration" do
+        startup do
+          Elasticsearch::Extensions::Test::Cluster.start(nodes: 1) if ENV['SERVER'] and not Elasticsearch::Extensions::Test::Cluster.running?
+        end
+
+        setup do
+          @client.indices.create index: 'test', body: {
+            mappings: {
+              d: {
+                properties: {
+                  title: { type: 'string' },
+                  title_suggest: {
+                    type: 'completion',
+                    payloads: true
+                  }
+                }
+              }
+            }
+          }
+
+          @client.index index: 'test', type: 'd', id: '1', body: {
+            title: 'One',
+            title_suggest: { input: ['one', 'uno', 'jedna'] }, output: 'ONE', payload: { id: '1' } }
+          @client.index index: 'test', type: 'd', id: '2', body: {
+            title: 'Two',
+            title_suggest: { input: ['two', 'due', 'dvě'] }, output: 'TWO', payload: { id: '2' } }
+          @client.index index: 'test', type: 'd', id: '3', body: {
+            title: 'Three',
+            title_suggest: { input: ['three', 'tres', 'tři'] }, output: 'THREE', payload: { id: '3' } }
+          @client.indices.refresh index: 'test'
+        end
+
+        should "return suggestions" do
+          s = search do
+            suggest :title, text: 't', completion: { field: 'title_suggest' }
+          end
+
+          response = @client.search index: 'test', body: s.to_hash
+          assert_equal 4, response['suggest']['title'][0]['options'].size
+        end
+
+        should "return a single suggestion" do
+          s = search do
+            suggest :title, text: 'th', completion: { field: 'title_suggest' }
+          end
+
+          response = @client.search index: 'test', body: s.to_hash
+          assert_equal 1, response['suggest']['title'][0]['options'].size
+        end
+      end
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,62 @@
+JRUBY = defined?(JRUBY_VERSION)
+
+if ENV['COVERAGE'] || ENV['CI']
+  require 'simplecov'
+  SimpleCov.start { add_filter "/test|test_" }
+end
+
+at_exit { Elasticsearch::Test::IntegrationTestCase.__run_at_exit_hooks }
+
+require 'test/unit'
+require 'shoulda-context'
+require 'mocha/setup'
+
+require 'minitest/reporters'
+Minitest::Reporters.use! Minitest::Reporters::SpecReporter.new
+
+require 'elasticsearch'
+require 'elasticsearch/extensions/test/cluster'
+require 'elasticsearch/extensions/test/startup_shutdown'
+
+require 'elasticsearch/dsl'
+
+module Elasticsearch
+  module Test
+    class IntegrationTestCase < ::Test::Unit::TestCase
+      include Elasticsearch::Extensions::Test
+      extend  StartupShutdown
+
+      startup do
+        Cluster.start(nodes: 1) if ENV['SERVER'] \
+                                && ! Elasticsearch::Extensions::Test::Cluster.running?
+      end
+
+      shutdown do
+        Cluster.stop if ENV['SERVER'] \
+                     && started?      \
+                     && Elasticsearch::Extensions::Test::Cluster.running?
+      end
+
+      def setup
+        @port = (ENV['TEST_CLUSTER_PORT'] || 9250).to_i
+
+        @logger =  Logger.new(STDERR)
+        @logger.formatter = proc do |severity, datetime, progname, msg|
+          color = case severity
+            when /INFO/ then :green
+            when /ERROR|WARN|FATAL/ then :red
+            when /DEBUG/ then :cyan
+            else :white
+          end
+          ANSI.ansi(severity[0] + ' ', color, :faint) + ANSI.ansi(msg, :white, :faint) + "\n"
+        end
+
+        @client = Elasticsearch::Client.new host: "localhost:#{@port}", logger: @logger
+      end
+
+      def teardown
+        @client.indices.delete index: '_all'
+      end
+    end
+  end
+end

data/test/unit/aggregations/avg_test.rb

@@ -0,0 +1,24 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    module Aggregations
+      class AvgTest < ::Test::Unit::TestCase
+        include Elasticsearch::DSL::Search::Aggregations
+
+        context "Avg agg" do
+          subject { Avg.new }
+
+          should "be converted to a Hash" do
+            assert_equal({ avg: {} }, subject.to_hash)
+          end
+
+          should "take a Hash" do
+            subject = Avg.new foo: 'bar'
+            assert_equal({ avg: { foo: 'bar' } }, subject.to_hash)
+          end
+        end
+      end
+    end
+  end
+end

data/test/unit/aggregations/cardinality_test.rb

@@ -0,0 +1,40 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    module Aggregations
+      class CardinalityTest < ::Test::Unit::TestCase
+        include Elasticsearch::DSL::Search::Aggregations
+
+        context "Cardinality agg" do
+          subject { Cardinality.new }
+
+          should "be converted to a Hash" do
+            assert_equal({ cardinality: {} }, subject.to_hash)
+          end
+                          
+          should "have option methods" do
+            subject = Cardinality.new :foo
+            
+            subject.field 'bar'
+            subject.precision_threshold 'bar'
+            subject.rehash 'bar'
+            subject.script 'bar'
+            subject.params 'bar'
+          
+            assert_equal %w[ field params precision_threshold rehash script ],
+                         subject.to_hash[:cardinality][:foo].keys.map(&:to_s).sort
+            assert_equal 'bar', subject.to_hash[:cardinality][:foo][:field]
+          end
+          
+          should "take a block" do
+            subject = Cardinality.new :foo do
+              field 'bar'
+            end
+            assert_equal({cardinality: { foo: { field: 'bar' } }}, subject.to_hash)
+          end
+        end
+      end
+    end
+  end
+end

data/test/unit/aggregations/children_test.rb

@@ -0,0 +1,36 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    module Aggregations
+      class ChildrenTest < ::Test::Unit::TestCase
+        include Elasticsearch::DSL::Search::Aggregations
+
+        context "Children aggregation" do
+          subject { Children.new }
+
+          should "be converted to a Hash" do
+            assert_equal({ children: {} }, subject.to_hash)
+          end
+
+          should "have option methods" do
+            subject = Children.new
+
+            subject.type 'bar'
+
+            assert_equal %w[ type ],
+                         subject.to_hash[:children].keys.map(&:to_s).sort
+            assert_equal 'bar', subject.to_hash[:children][:type]
+          end
+
+          should "take a block" do
+            subject = Children.new do
+              type 'bar'
+            end
+            assert_equal({children: { type: 'bar' } }, subject.to_hash)
+          end
+        end
+      end
+    end
+  end
+end

data/test/unit/aggregations/date_histogram_test.rb

@@ -0,0 +1,49 @@
+require 'test_helper'
+
+module Elasticsearch
+  module Test
+    module Aggregations
+      class DateHistogramTest < ::Test::Unit::TestCase
+        include Elasticsearch::DSL::Search::Aggregations
+
+        context "DateHistogram aggregation" do
+          subject { DateHistogram.new }
+
+          should "be converted to a Hash" do
+            assert_equal({ date_histogram: {} }, subject.to_hash)
+          end
+
+          should "have option methods" do
+            subject = DateHistogram.new
+
+            subject.field 'bar'
+            subject.interval 'bar'
+            subject.pre_zone 'bar'
+            subject.post_zone 'bar'
+            subject.time_zone 'bar'
+            subject.pre_zone_adjust_large_interval 'bar'
+            subject.pre_offset 'bar'
+            subject.post_offset 'bar'
+            subject.format 'bar'
+            subject.min_doc_count 'bar'
+            subject.extended_bounds 'bar'
+            subject.order 'bar'
+
+            assert_equal %w[ extended_bounds field format interval min_doc_count order post_offset post_zone pre_offset pre_zone pre_zone_adjust_large_interval time_zone ],
+                         subject.to_hash[:date_histogram].keys.map(&:to_s).sort
+            assert_equal 'bar', subject.to_hash[:date_histogram][:field]
+          end
+
+          should "take a block" do
+            subject = DateHistogram.new do
+              field    'bar'
+              interval 'day'
+              format   'yyyy-MM-dd'
+            end
+            assert_equal({date_histogram: { field: 'bar', interval: 'day', format: 'yyyy-MM-dd' } }, subject.to_hash)
+          end
+        end
+      end
+    end
+  end
+end