elastomer-client 0.7.0 → 0.8.1

data/test/client/multi_percolate_test.rb

@@ -0,0 +1,130 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+describe Elastomer::Client::MultiPercolate do
+
+  before do
+    @name  = "elastomer-mpercolate-test"
+    @index = $client.index(@name)
+
+    unless @index.exists?
+      @index.create \
+        :settings => { "index.number_of_shards" => 1, "index.number_of_replicas" => 0 },
+        :mappings => {
+          :doc1 => {
+            :_source => { :enabled => true }, :_all => { :enabled => false },
+            :properties => {
+              :title  => { :type => "string", :analyzer => "standard" },
+              :author => { :type => "string", :index => "not_analyzed" }
+            }
+          },
+          :doc2 => {
+            :_source => { :enabled => true }, :_all => { :enabled => false },
+            :properties => {
+              :title  => { :type => "string", :analyzer => "standard" },
+              :author => { :type => "string", :index => "not_analyzed" }
+            }
+          }
+        }
+
+      wait_for_index(@name)
+    end
+
+    @docs = @index.docs
+  end
+
+  after do
+    @index.delete if @index.exists?
+  end
+
+  it "performs multi percolate queries" do
+    populate!
+
+    body = [
+      '{"percolate" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "pea53"}}',
+      '{"percolate" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "grantr"}}',
+      '{"count" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "grantr"}}',
+      nil
+    ]
+    body = body.join "\n"
+    h = $client.multi_percolate body
+    response1, response2, response3 = h["responses"]
+    assert_equal ["1", "2"], response1["matches"].map { |match| match["_id"] }.sort
+    assert_equal ["1", "3"], response2["matches"].map { |match| match["_id"] }.sort
+    assert_equal 2, response3["total"]
+  end
+
+  it "performs multi percolate queries with .mpercolate" do
+    populate!
+
+    body = [
+      '{"percolate" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "pea53"}}',
+      '{"percolate" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "grantr"}}',
+      '{"count" : {"index": "elastomer-mpercolate-test", "type": "doc2"}}',
+      '{"doc": {"author": "grantr"}}',
+      nil
+    ]
+    body = body.join "\n"
+    h = $client.mpercolate body
+    response1, response2, response3 = h["responses"]
+    assert_equal ["1", "2"], response1["matches"].map { |match| match["_id"] }.sort
+    assert_equal ["1", "3"], response2["matches"].map { |match| match["_id"] }.sort
+    assert_equal 2, response3["total"]
+  end
+
+  it "supports a nice block syntax" do
+    populate!
+
+    h = $client.multi_percolate(:index => @name, :type => "doc2") do |m|
+      m.percolate :author => "pea53"
+      m.percolate :author => "grantr"
+      m.count({ :author => "grantr" })
+    end
+
+    response1, response2, response3 = h["responses"]
+    assert_equal ["1", "2"], response1["matches"].map { |match| match["_id"] }.sort
+    assert_equal ["1", "3"], response2["matches"].map { |match| match["_id"] }.sort
+    assert_equal 2, response3["total"]
+  end
+
+  # rubocop:disable Metrics/MethodLength
+  def populate!
+    @docs.index \
+      :_id    => 1,
+      :_type  => "doc1",
+      :title  => "the author of gravatar",
+      :author => "mojombo"
+
+    @docs.index \
+      :_id    => 2,
+      :_type  => "doc1",
+      :title  => "the author of resque",
+      :author => "defunkt"
+
+    @docs.index \
+      :_id    => 1,
+      :_type  => "doc2",
+      :title  => "the author of logging",
+      :author => "pea53"
+
+    @docs.index \
+      :_id    => 2,
+      :_type  => "doc2",
+      :title  => "the author of rubber-band",
+      :author => "grantr"
+
+    percolator1 = @index.percolator "1"
+    percolator1.create :query => { :match_all => { } }
+    percolator2 = @index.percolator "2"
+    percolator2.create :query => { :match => { :author => "pea53" } }
+    percolator2 = @index.percolator "3"
+    percolator2.create :query => { :match => { :author => "grantr" } }
+
+    @index.refresh
+  end
+  # rubocop:enable Metrics/MethodLength
+end

data/test/client/multi_search_test.rb

@@ -1,27 +1,27 @@
-require File.expand_path('../../test_helper', __FILE__)
+require File.expand_path("../../test_helper", __FILE__)
 
 describe Elastomer::Client::MultiSearch do
 
   before do
-    @name  = 'elastomer-msearch-test'
+    @name  = "elastomer-msearch-test"
     @index = $client.index(@name)
 
     unless @index.exists?
       @index.create \
-        :settings => { 'index.number_of_shards' => 1, 'index.number_of_replicas' => 0 },
+        :settings => { "index.number_of_shards" => 1, "index.number_of_replicas" => 0 },
         :mappings => {
           :doc1 => {
             :_source => { :enabled => true }, :_all => { :enabled => false },
             :properties => {
-              :title  => { :type => 'string', :analyzer => 'standard' },
-              :author => { :type => 'string', :index => 'not_analyzed' }
+              :title  => { :type => "string", :analyzer => "standard" },
+              :author => { :type => "string", :index => "not_analyzed" }
             }
           },
           :doc2 => {
             :_source => { :enabled => true }, :_all => { :enabled => false },
             :properties => {
-              :title  => { :type => 'string', :analyzer => 'standard' },
-              :author => { :type => 'string', :index => 'not_analyzed' }
+              :title  => { :type => "string", :analyzer => "standard" },
+              :author => { :type => "string", :index => "not_analyzed" }
             }
           }
         }
@@ -36,7 +36,7 @@ describe Elastomer::Client::MultiSearch do
     @index.delete if @index.exists?
   end
 
-  it 'performs multisearches' do
+  it "performs multisearches" do
     populate!
 
     body = [
@@ -54,7 +54,7 @@ describe Elastomer::Client::MultiSearch do
     assert_equal "2", response2["hits"]["hits"][0]["_id"]
 
     body = [
-      '{}',
+      "{}",
       '{"query" : {"match": {"author" : "grantr"}}}',
       nil
     ]
@@ -65,7 +65,7 @@ describe Elastomer::Client::MultiSearch do
     assert_equal "2", response1["hits"]["hits"][0]["_id"]
   end
 
-  it 'performs multisearches with .msearch' do
+  it "performs multisearches with .msearch" do
     populate!
 
     body = [
@@ -83,7 +83,7 @@ describe Elastomer::Client::MultiSearch do
     assert_equal "2", response2["hits"]["hits"][0]["_id"]
 
     body = [
-      '{}',
+      "{}",
       '{"query" : {"match": {"author" : "grantr"}}}',
       nil
     ]
@@ -94,12 +94,12 @@ describe Elastomer::Client::MultiSearch do
     assert_equal "2", response1["hits"]["hits"][0]["_id"]
   end
 
-  it 'supports a nice block syntax' do
+  it "supports a nice block syntax" do
     populate!
 
     h = $client.multi_search do |m|
       m.search({:query => { :match_all => {}}}, :index => @name, :search_type => :count)
-      m.search({:query => { :match => { "title" => "author" }}}, :index => @name, :type => 'doc2')
+      m.search({:query => { :match => { "title" => "author" }}}, :index => @name, :type => "doc2")
     end
 
     response1, response2 = h["responses"]
@@ -109,7 +109,7 @@ describe Elastomer::Client::MultiSearch do
 
     h = @index.multi_search do |m|
       m.search({:query => { :match_all => {}}}, :search_type => :count)
-      m.search({:query => { :match => { "title" => "author" }}}, :type => 'doc2')
+      m.search({:query => { :match => { "title" => "author" }}}, :type => "doc2")
     end
 
     response1, response2 = h["responses"]
@@ -117,9 +117,9 @@ describe Elastomer::Client::MultiSearch do
     assert_equal 4, response1["hits"]["total"]
     assert_equal 2, response2["hits"]["total"]
 
-    h = @index.docs('doc1').multi_search do |m|
+    h = @index.docs("doc1").multi_search do |m|
       m.search({:query => { :match_all => {}}}, :search_type => :count)
-      m.search({:query => { :match => { "title" => "logging" }}}, :type => 'doc2')
+      m.search({:query => { :match => { "title" => "logging" }}}, :type => "doc2")
     end
 
     response1, response2 = h["responses"]
@@ -128,31 +128,33 @@ describe Elastomer::Client::MultiSearch do
     assert_equal 1, response2["hits"]["total"]
   end
 
+  # rubocop:disable Metrics/MethodLength
   def populate!
     @docs.index \
       :_id    => 1,
-      :_type  => 'doc1',
-      :title  => 'the author of gravatar',
-      :author => 'mojombo'
+      :_type  => "doc1",
+      :title  => "the author of gravatar",
+      :author => "mojombo"
 
     @docs.index \
       :_id    => 2,
-      :_type  => 'doc1',
-      :title  => 'the author of resque',
-      :author => 'defunkt'
+      :_type  => "doc1",
+      :title  => "the author of resque",
+      :author => "defunkt"
 
     @docs.index \
       :_id    => 1,
-      :_type  => 'doc2',
-      :title  => 'the author of logging',
-      :author => 'pea53'
+      :_type  => "doc2",
+      :title  => "the author of logging",
+      :author => "pea53"
 
     @docs.index \
       :_id    => 2,
-      :_type  => 'doc2',
-      :title  => 'the author of rubber-band',
-      :author => 'grantr'
+      :_type  => "doc2",
+      :title  => "the author of rubber-band",
+      :author => "grantr"
 
     @index.refresh
   end
+  # rubocop:enable Metrics/MethodLength
 end

data/test/client/nodes_test.rb

@@ -1,57 +1,58 @@
-require File.expand_path('../../test_helper', __FILE__)
+require File.expand_path("../../test_helper", __FILE__)
 
 describe Elastomer::Client::Nodes do
 
-  it 'gets info for the node(s)' do
+  it "gets info for the node(s)" do
     h = $client.nodes.info
-    assert h.key?('cluster_name'), 'the cluster name is returned'
-    assert_instance_of Hash, h['nodes'], 'the node list is returned'
+    assert h.key?("cluster_name"), "the cluster name is returned"
+    assert_instance_of Hash, h["nodes"], "the node list is returned"
   end
 
-  it 'gets stats for the node(s)' do
+  it "gets stats for the node(s)" do
     h = $client.nodes.stats
-    assert_instance_of Hash, h['nodes'], 'the node list is returned'
+    assert_instance_of Hash, h["nodes"], "the node list is returned"
 
-    node = h['nodes'].values.first
-    assert node.key?('indices'), 'indices stats are returned'
+    node = h["nodes"].values.first
+    assert node.key?("indices"), "indices stats are returned"
   end
 
   if es_version_1_x?
-    it 'fitlers node info' do
-      h = $client.nodes.info(:info => 'os')
-      node = h['nodes'].values.first
-      assert node.key?('os'), 'expected os info to be present'
-      assert !node.key?('jvm'), 'expected jvm info to be absent'
+    it "fitlers node info" do
+      h = $client.nodes.info(:info => "os")
+      node = h["nodes"].values.first
+      assert node.key?("os"), "expected os info to be present"
+      assert !node.key?("jvm"), "expected jvm info to be absent"
 
       h = $client.nodes.info(:info => %w[jvm process])
-      node = h['nodes'].values.first
-      assert node.key?('jvm'), 'expected jvm info to be present'
-      assert node.key?('process'), 'expected process info to be present'
-      assert !node.key?('network'), 'expected network info to be absent'
+      node = h["nodes"].values.first
+      assert node.key?("jvm"), "expected jvm info to be present"
+      assert node.key?("process"), "expected process info to be present"
+      assert !node.key?("network"), "expected network info to be absent"
     end
 
-    it 'filters node stats' do
-      h = $client.nodes.stats(:stats => 'http')
-      node = h['nodes'].values.first
-      assert node.key?('http'), 'expected http stats to be present'
-      assert !node.key?('indices'), 'expected indices stats to be absent'
+    it "filters node stats" do
+      h = $client.nodes.stats(:stats => "http")
+      node = h["nodes"].values.first
+      assert node.key?("http"), "expected http stats to be present"
+      assert !node.key?("indices"), "expected indices stats to be absent"
     end
   end
 
-  it 'gets the hot threads for the node(s)' do
+  it "gets the hot threads for the node(s)" do
     str = $client.nodes.hot_threads :read_timeout => 2
     assert_instance_of String, str
-    assert_match %r/cpu usage by thread/, str
+    assert !str.nil?, "expected response to not be nil"
+    assert !str.empty?, "expected response to not be empty"
   end
 
-  it 'can be scoped to a single node' do
-    h = $client.nodes('node-with-no-name').info
-    assert_empty h['nodes']
+  it "can be scoped to a single node" do
+    h = $client.nodes("node-with-no-name").info
+    assert_empty h["nodes"]
   end
 
-  it 'can be scoped to multiple nodes' do
+  it "can be scoped to multiple nodes" do
     h = $client.nodes(%w[node1 node2 node3]).info
-    assert_empty h['nodes']
+    assert_empty h["nodes"]
   end
 
 end

data/test/client/percolator_test.rb

@@ -0,0 +1,52 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+describe Elastomer::Client::Percolator do
+
+  before do
+    @index = $client.index "elastomer-percolator-test"
+    @index.delete if @index.exists?
+    @docs = @index.docs("docs")
+  end
+
+  after do
+    @index.delete if @index.exists?
+  end
+
+  describe "when an index exists" do
+    before do
+      @index.create(nil)
+      wait_for_index(@index_name)
+    end
+
+    it "creates a query" do
+      percolator = @index.percolator "1"
+      response = percolator.create :query => { :match_all => { } }
+      assert response["created"], "Couldn't create the percolator query"
+    end
+
+    it "gets a query" do
+      percolator = @index.percolator "1"
+      percolator.create :query => { :match_all => { } }
+      response = percolator.get
+      assert response["found"], "Couldn't find the percolator query"
+    end
+
+    it "deletes a query" do
+      percolator = @index.percolator "1"
+      percolator.create :query => { :match_all => { } }
+      response = percolator.delete
+      assert response["found"], "Couldn't find the percolator query"
+    end
+
+    it "checks for the existence of a query" do
+      percolator = @index.percolator "1"
+      refute percolator.exists?, "Percolator query exists"
+      percolator.create :query => { :match_all => { } }
+      assert percolator.exists?, "Percolator query does not exist"
+    end
+
+    it "cannot delete all percolators by providing a nil id" do
+      assert_raises(ArgumentError) { percolator = @index.percolator nil }
+    end
+  end
+end

data/test/client/repository_test.rb

@@ -1,4 +1,4 @@
-require File.expand_path('../../test_helper', __FILE__)
+require File.expand_path("../../test_helper", __FILE__)
 
 describe Elastomer::Client::Repository do
 
@@ -9,11 +9,11 @@ describe Elastomer::Client::Repository do
         skip "To enable snapshot tests, add a path.repo setting to your elasticsearch.yml file."
       end
 
-      @name = 'elastomer-repository-test'
+      @name = "elastomer-repository-test"
       @repo = $client.repository(@name)
     end
 
-    it 'determines if a repo exists' do
+    it "determines if a repo exists" do
       assert_equal false, @repo.exists?
       assert_equal false, @repo.exist?
       with_tmp_repo(@name) do
@@ -21,63 +21,63 @@ describe Elastomer::Client::Repository do
       end
     end
 
-    it 'creates repos' do
+    it "creates repos" do
       response = create_repo(@name)
       assert_equal true, response["acknowledged"]
       delete_repo(@name)
     end
 
-    it 'cannot create a repo without a name' do
+    it "cannot create a repo without a name" do
       lambda {
         create_repo(nil)
       }.must_raise ArgumentError
     end
 
-    it 'gets repos' do
+    it "gets repos" do
       with_tmp_repo do |repo|
         response = repo.get
         refute_nil response[repo.name]
       end
     end
 
-    it 'gets all repos' do
+    it "gets all repos" do
       with_tmp_repo do |repo|
         response = $client.repository.get
         refute_nil response[repo.name]
       end
     end
 
-    it 'gets repo status' do
+    it "gets repo status" do
       with_tmp_repo do |repo|
         response = repo.status
         assert_equal [], response["snapshots"]
       end
     end
 
-    it 'gets status of all repos' do
+    it "gets status of all repos" do
       response = $client.repository.status
       assert_equal [], response["snapshots"]
     end
 
-    it 'updates repos' do
+    it "updates repos" do
       with_tmp_repo do |repo|
-        settings = repo.get[repo.name]['settings']
-        response = repo.update(:type => 'fs', :settings => settings.merge('compress' => true))
+        settings = repo.get[repo.name]["settings"]
+        response = repo.update(:type => "fs", :settings => settings.merge("compress" => true))
         assert_equal true, response["acknowledged"]
         assert_equal "true", repo.get[repo.name]["settings"]["compress"]
       end
     end
 
-    it 'cannot update a repo without a name' do
+    it "cannot update a repo without a name" do
       with_tmp_repo do |repo|
         lambda {
-          settings = repo.get[repo.name]['settings']
-          $client.repository.update(:type => 'fs', :settings => settings.merge('compress' => true))
+          settings = repo.get[repo.name]["settings"]
+          $client.repository.update(:type => "fs", :settings => settings.merge("compress" => true))
         }.must_raise ArgumentError
       end
     end
 
-    it 'deletes repos' do
+    it "deletes repos" do
       with_tmp_repo do |repo|
         response = repo.delete
         assert_equal true, response["acknowledged"]
@@ -85,26 +85,26 @@ describe Elastomer::Client::Repository do
       end
     end
 
-    it 'cannot delete a repo without a name' do
+    it "cannot delete a repo without a name" do
       lambda {
         $client.repository.delete
       }.must_raise ArgumentError
     end
 
-    it 'gets snapshots' do
+    it "gets snapshots" do
       with_tmp_repo do |repo|
         response = repo.snapshots.get
         assert_equal [], response["snapshots"]
 
-        create_snapshot(repo, 'test-snapshot')
+        create_snapshot(repo, "test-snapshot")
         response = repo.snapshot.get
-        assert_equal ['test-snapshot'], response["snapshots"].collect { |info| info["snapshot"] }
+        assert_equal ["test-snapshot"], response["snapshots"].collect { |info| info["snapshot"] }
 
-        snapshot2 = create_snapshot(repo, 'test-snapshot2')
+        snapshot2 = create_snapshot(repo, "test-snapshot2")
         response  = repo.snapshots.get
         snapshot_names = response["snapshots"].collect { |info| info["snapshot"] }
-        assert_includes snapshot_names, 'test-snapshot'
-        assert_includes snapshot_names, 'test-snapshot2'
+        assert_includes snapshot_names, "test-snapshot"
+        assert_includes snapshot_names, "test-snapshot2"
       end
     end
   end