Updates elasticsearch helper integration tests

Author: picandocodigo

elasticsearch/spec/integration/helpers/bulk_helper_spec.rb

@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-require_relative 'helpers_spec_helper'
+require_relative File.expand_path('../../spec_helper'. __dir__)
 require 'elasticsearch/helpers/bulk_helper'
 require 'tempfile'
 
@@ -23,7 +23,7 @@
     let(:index) { 'bulk_animals' }
     let(:index_slice) { 'bulk_animals_slice' }
     let(:params) { { refresh: 'wait_for' } }
-    let(:bulk_helper) { Elasticsearch::Helpers::BulkHelper.new(client, index, params) }
+    let(:bulk_helper) { Elasticsearch::Helpers::BulkHelper.new(CLIENT, index, params) }
     let(:docs) do
       [
         { scientific_name: 'Lama guanicoe', name:'Guanaco' },
@@ -40,8 +40,8 @@
     end
 
     after do
-      client.indices.delete(index: index, ignore: 404)
-      client.indices.delete(index: index_slice, ignore: 404)
+      CLIENT.indices.delete(index: index, ignore: 404)
+      CLIENT.indices.delete(index: index_slice, ignore: 404)
     end
 
     it 'Ingests documents' do
@@ -58,7 +58,7 @@
       ]
       bulk_helper.ingest(docs)
       # Get the ingested documents, add id and modify them to update them:
-      animals = client.search(index: index)['hits']['hits']
+      animals = CLIENT.search(index: index)['hits']['hits']
       # Add id to each doc
       docs = animals.map { |animal| animal['_source'].merge({'id' => animal['_id'] }) }
       docs.map { |doc| doc['scientific_name'].upcase! }
@@ -73,17 +73,17 @@
       response = bulk_helper.delete(ids)
       expect(response.status).to eq 200
       expect(response['items'].map { |item| item['delete']['result'] }.uniq.first).to eq('deleted')
-      expect(client.count(index: index)['count']).to eq(0)
+      expect(CLIENT.count(index: index)['count']).to eq(0)
     end
 
     it 'Ingests documents and yields response and docs' do
       slice = 2
-      bulk_helper = Elasticsearch::Helpers::BulkHelper.new(client, index_slice, params)
+      bulk_helper = Elasticsearch::Helpers::BulkHelper.new(CLIENT, index_slice, params)
       response = bulk_helper.ingest(docs, {slice: slice}) do |response, docs|
         expect(response).to be_an_instance_of Elasticsearch::API::Response
         expect(docs.count).to eq slice
       end
-      response = client.search(index: index_slice, size: 200)
+      response = CLIENT.search(index: index_slice, size: 200)
       expect(response['hits']['hits'].map { |a| a['_source'].transform_keys(&:to_sym) }).to eq docs
     end
 

elasticsearch/spec/integration/helpers/esql_helper_spec.rb

@@ -14,7 +14,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-require_relative 'helpers_spec_helper'
+require_relative File.expand_path('../../spec_helper'. __dir__)
 require 'elasticsearch/helpers/esql_helper'
 require 'ipaddr'
 
@@ -30,15 +30,15 @@
   end
 
   before do
-    client.indices.create(
+    CLIENT.indices.create(
       index: index,
       body: {
         mappings: {
           properties: { 'client.ip' => { type: 'ip' }, message: { type: 'keyword' } }
         }
       }
     )
-    client.bulk(
+    CLIENT.bulk(
       index: index,
       body: [
         {'index': {}},
@@ -61,11 +61,11 @@
   end
 
   after do
-    client.indices.delete(index: index)
+    CLIENT.indices.delete(index: index)
   end
 
   it 'returns an ESQL response as a relational key/value object' do
-    response = esql_helper.query(client, query)
+    response = esql_helper.query(CLIENT, query)
     expect(response.count).to eq 7
     expect(response.first.keys).to eq ['duration_ms', 'message', 'event.duration', 'client.ip', '@timestamp']
     response.each do |r|
@@ -82,7 +82,7 @@
       'client.ip' => Proc.new { |i| IPAddr.new(i) },
       'event.duration' => Proc.new { |d| d.to_s }
     }
-    response = esql_helper.query(client, query, parser: parser)
+    response = esql_helper.query(CLIENT, query, parser: parser)
     response.each do |r|
       expect(r['@timestamp']).to be_a DateTime
       expect(r['client.ip']).to be_a IPAddr
@@ -92,7 +92,7 @@
   end
 
   it 'parser does not error when value is nil, leaves nil' do
-    client.index(
+    CLIENT.index(
       index: index,
       body: {
         '@timestamp' => nil,
@@ -107,7 +107,7 @@
       'client.ip' => Proc.new { |i| IPAddr.new(i) },
       'event.duration' => Proc.new { |d| d.to_s }
     }
-    response = esql_helper.query(client, query, parser: parser)
+    response = esql_helper.query(CLIENT, query, parser: parser)
     response.each do |r|
       expect [DateTime, NilClass].include?(r['@timestamp'].class)
       expect [IPAddr, NilClass].include?(r['client.ip'].class)

elasticsearch/spec/integration/helpers/helpers_spec_helper.rb

@@ -14,14 +14,14 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-require_relative 'helpers_spec_helper'
+require_relative File.expand_path('../../spec_helper'. __dir__)
 require 'elasticsearch/helpers/scroll_helper'
 
 context 'Elasticsearch client helpers' do
   context 'ScrollHelper' do
     let(:index) { 'books' }
     let(:body) { { size: 12, query: { match_all: {} } } }
-    let(:scroll_helper) { Elasticsearch::Helpers::ScrollHelper.new(client, index, body) }
+    let(:scroll_helper) { Elasticsearch::Helpers::ScrollHelper.new(CLIENT, index, body) }
 
     before do
       documents = [
@@ -50,11 +50,11 @@
         { index: { _index: index, data: {name: "The Left Hand of Darkness", "author": "Ursula K. Le Guin", "release_date": "1969-06-01", "page_count": 304} } },
         { index: { _index: index, data: {name: "The Moon is a Harsh Mistress", "author": "Robert A. Heinlein", "release_date": "1966-04-01", "page_count": 288 } } }
       ]
-      client.bulk(body: documents, refresh: 'wait_for')
+      CLIENT.bulk(body: documents, refresh: 'wait_for')
     end
 
     after do
-      client.indices.delete(index: index)
+      CLIENT.indices.delete(index: index)
     end
 
     it 'instantiates a scroll helper' do