elasticsearch 8.18.0 → 9.0.1

data/spec/integration/helpers/bulk_helper_spec.rb

@@ -1,206 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# 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 'elasticsearch/helpers/bulk_helper'
-require 'tempfile'
-
-context 'Elasticsearch client helpers' do
-  context 'Bulk helper' do
-    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(:docs) do
-      [
-        { scientific_name: 'Lama guanicoe', name:'Guanaco' },
-        { scientific_name: 'Tayassu pecari', name:'White-lipped peccary' },
-        { scientific_name: 'Snycerus caffer', name:'Buffalo, african' },
-        { scientific_name: 'Coluber constrictor', name:'Snake, racer' },
-        { scientific_name: 'Thalasseus maximus', name:'Royal tern' },
-        { scientific_name: 'Centrocercus urophasianus', name:'Hen, sage' },
-        { scientific_name: 'Sitta canadensis', name:'Nuthatch, red-breasted' },
-        { scientific_name: 'Aegypius tracheliotus', name:'Vulture, lappet-faced' },
-        { scientific_name: 'Bucephala clangula', name:'Common goldeneye' },
-        { scientific_name: 'Felis pardalis', name:'Ocelot' }
-      ]
-    end
-
-    after do
-      client.indices.delete(index: index, ignore: 404)
-      client.indices.delete(index: index_slice, ignore: 404)
-    end
-
-    it 'Ingests documents' do
-      response = bulk_helper.ingest(docs)
-      expect(response).to be_an_instance_of Elasticsearch::API::Response
-      expect(response.status).to eq(200)
-      expect(response['items'].map { |a| a['index']['status'] }.uniq.first).to eq 201
-    end
-
-    it 'Updates documents' do
-      docs = [
-        { scientific_name: 'Otocyon megalotos', name: 'Bat-eared fox' },
-        { scientific_name: 'Herpestes javanicus', name: 'Small Indian mongoose' }
-      ]
-      bulk_helper.ingest(docs)
-      # Get the ingested documents, add id and modify them to update them:
-      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! }
-      response = bulk_helper.update(docs)
-      expect(response.status).to eq(200)
-      expect(response['items'].map { |i| i['update']['result'] }.uniq.first).to eq('updated')
-    end
-
-    it 'Deletes documents' do
-      response = bulk_helper.ingest(docs)
-      ids = response.body['items'].map { |a| a['index']['_id'] }
-      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)
-    end
-
-    it 'Ingests documents and yields response and docs' do
-      slice = 2
-      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)
-      expect(response['hits']['hits'].map { |a| a['_source'].transform_keys(&:to_sym) }).to eq docs
-    end
-
-    context 'JSON File helper' do
-      let(:file) { Tempfile.new('test-data.json') }
-      let(:json) do
-        json = <<~JSON
-        [
-          {
-            "character_name": "Anallese Lonie",
-           "species": "mouse",
-           "catchphrase": "Seamless regional definition",
-           "favorite_food": "pizza"
-          },
-          {
-            "character_name": "Janey Davidovsky",
-           "species": "cat",
-           "catchphrase": "Down-sized responsive pricing structure",
-           "favorite_food": "pizza"
-          },
-          {
-            "character_name": "Morse Mountford",
-           "species": "cat",
-           "catchphrase": "Ameliorated modular data-warehouse",
-           "favorite_food": "carrots"
-          },
-          {
-            "character_name": "Saundra Kauble",
-           "species": "dog",
-           "catchphrase": "Synchronised 24/7 support",
-           "favorite_food": "carrots"
-          },
-          {
-            "character_name": "Kain Viggars",
-           "species": "cat",
-           "catchphrase": "Open-architected asymmetric circuit",
-           "favorite_food": "carrots"
-          }
-        ]
-        JSON
-      end
-
-      before do
-        file.write(json)
-        file.rewind
-      end
-
-      after do
-        file.close
-        file.unlink
-      end
-
-      it 'Ingests a JSON file' do
-        response = bulk_helper.ingest_json(file)
-
-        expect(response).to be_an_instance_of Elasticsearch::API::Response
-        expect(response.status).to eq(200)
-      end
-
-      context 'with data not in root of JSON file' do
-        let(:json) do
-          json = <<~JSON
-          {
-            "field": "value",
-            "status": 200,
-            "data": {
-              "items": [
-                {
-                  "character_name": "Anallese Lonie",
-                  "species": "mouse",
-                  "catchphrase": "Seamless regional definition",
-                  "favorite_food": "pizza"
-                },
-                {
-                  "character_name": "Janey Davidovsky",
-                  "species": "cat",
-                  "catchphrase": "Down-sized responsive pricing structure",
-                  "favorite_food": "pizza"
-                },
-                {
-                  "character_name": "Morse Mountford",
-                  "species": "cat",
-                  "catchphrase": "Ameliorated modular data-warehouse",
-                  "favorite_food": "carrots"
-                },
-                {
-                  "character_name": "Saundra Kauble",
-                  "species": "dog",
-                  "catchphrase": "Synchronised 24/7 support",
-                  "favorite_food": "carrots"
-                },
-                {
-                  "character_name": "Kain Viggars",
-                  "species": "cat",
-                  "catchphrase": "Open-architected asymmetric circuit",
-                  "favorite_food": "carrots"
-                }
-              ]
-            }
-          }
-          JSON
-        end
-
-        it 'Ingests a JSON file passing keys as Array' do
-          response = bulk_helper.ingest_json(file, { keys: ['data', 'items'] })
-          expect(response).to be_an_instance_of Elasticsearch::API::Response
-          expect(response.status).to eq(200)
-          expect(response['items'].map { |a| a['index']['status'] }.uniq.first).to eq 201
-        end
-
-        it 'Ingests a JSON file passing keys as String' do
-          response = bulk_helper.ingest_json(file, { keys: 'data,items' })
-          expect(response).to be_an_instance_of Elasticsearch::API::Response
-          expect(response.status).to eq(200)
-          expect(response['items'].map { |a| a['index']['status'] }.uniq.first).to eq 201
-        end
-      end
-    end
-  end
-end

data/spec/integration/helpers/esql_helper_spec.rb

@@ -1,118 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# 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 'elasticsearch/helpers/esql_helper'
-require 'ipaddr'
-
-context 'Elasticsearch client helpers' do
-  let(:index) { 'esql_helper_test' }
-  let(:body) { { size: 12, query: { match_all: {} } } }
-  let(:esql_helper) { Elasticsearch::Helpers::ESQLHelper }
-  let(:query) do
-    <<~ESQL
-        FROM #{index}
-        | EVAL duration_ms = ROUND(event.duration / 1000000.0, 1)
-    ESQL
-  end
-
-  before do
-    client.indices.create(
-      index: index,
-      body: {
-        mappings: {
-          properties: { 'client.ip' => { type: 'ip' }, message: { type: 'keyword' } }
-        }
-      }
-    )
-    client.bulk(
-      index: index,
-      body: [
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T12:15:03.360Z', 'client.ip' => '172.21.2.162', message: 'Connected to 10.1.0.3', 'event.duration' => 3450233},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T12:27:28.948Z', 'client.ip' => '172.21.2.113', message: 'Connected to 10.1.0.2', 'event.duration' => 2764889},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T13:33:34.937Z', 'client.ip' => '172.21.0.5', message: 'Disconnected', 'event.duration' => 1232382},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T13:51:54.732Z', 'client.ip' => '172.21.3.15', message: 'Connection error', 'event.duration' => 725448},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T13:52:55.015Z', 'client.ip' => '172.21.3.15', message: 'Connection error', 'event.duration' => 8268153},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T13:53:55.832Z', 'client.ip' => '172.21.3.15', message: 'Connection error', 'event.duration' => 5033755},
-        {'index': {}},
-        {'@timestamp' => '2023-10-23T13:55:01.543Z', 'client.ip' => '172.21.3.15', message: 'Connected to 10.1.0.1', 'event.duration' => 1756467}
-      ],
-      refresh: true
-    )
-  end
-
-  after do
-    client.indices.delete(index: index)
-  end
-
-  it 'returns an ESQL response as a relational key/value object' do
-    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|
-      expect(r['@timestamp']).to be_a String
-      expect(r['client.ip']).to be_a String
-      expect(r['message']).to be_a String
-      expect(r['event.duration']).to be_a Integer
-    end
-  end
-
-  it 'parses iterated objects when procs are passed in' do
-    parser = {
-      '@timestamp' => Proc.new { |t| DateTime.parse(t) },
-      '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.each do |r|
-      expect(r['@timestamp']).to be_a DateTime
-      expect(r['client.ip']).to be_a IPAddr
-      expect(r['message']).to be_a String
-      expect(r['event.duration']).to be_a String
-    end
-  end
-
-  it 'parser does not error when value is nil, leaves nil' do
-    client.index(
-      index: index,
-      body: {
-        '@timestamp' => nil,
-        'client.ip' => nil,
-        message: 'Connected to 10.1.0.1',
-        'event.duration' => 1756465
-      },
-      refresh: true
-    )
-    parser = {
-      '@timestamp' => Proc.new { |t| DateTime.parse(t) },
-      '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.each do |r|
-      expect [DateTime, NilClass].include?(r['@timestamp'].class)
-      expect [IPAddr, NilClass].include?(r['client.ip'].class)
-      expect(r['message']).to be_a String
-      expect(r['event.duration']).to be_a String
-    end
-  end
-end

data/spec/integration/helpers/helpers_spec_helper.rb

@@ -1,29 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-require 'spec_helper'
-
-ELASTICSEARCH_URL = ENV['TEST_ES_SERVER'] || "http://localhost:#{(ENV['PORT'] || 9200)}"
-raise URI::InvalidURIError unless ELASTICSEARCH_URL =~ /\A#{URI::DEFAULT_PARSER.make_regexp}\z/
-
-def client
-  @client ||= Elasticsearch::Client.new(
-    host: ELASTICSEARCH_URL,
-    user: 'elastic',
-    password: 'changeme'
-  )
-end

data/spec/integration/helpers/scroll_helper_spec.rb

@@ -1,83 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# 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 '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) }
-
-    before do
-      documents = [
-        { index: { _index: index, data: {name: "Leviathan Wakes", "author": "James S.A. Corey", "release_date": "2011-06-02", "page_count": 561} } },
-        { index: { _index: index, data: {name: "Hyperion", "author": "Dan Simmons", "release_date": "1989-05-26", "page_count": 482} } },
-        { index: { _index: index, data: {name: "Dune", "author": "Frank Herbert", "release_date": "1965-06-01", "page_count": 604} } },
-        { index: { _index: index, data: {name: "Dune Messiah", "author": "Frank Herbert", "release_date": "1969-10-15", "page_count": 331} } },
-        { index: { _index: index, data: {name: "Children of Dune", "author": "Frank Herbert", "release_date": "1976-04-21", "page_count": 408} } },
-        { index: { _index: index, data: {name: "God Emperor of Dune", "author": "Frank Herbert", "release_date": "1981-05-28", "page_count": 454} } },
-        { index: { _index: index, data: {name: "Consider Phlebas", "author": "Iain M. Banks", "release_date": "1987-04-23", "page_count": 471} } },
-        { index: { _index: index, data: {name: "Pandora's Star", "author": "Peter F. Hamilton", "release_date": "2004-03-02", "page_count": 768} } },
-        { index: { _index: index, data: {name: "Revelation Space", "author": "Alastair Reynolds", "release_date": "2000-03-15", "page_count": 585} } },
-        { index: { _index: index, data: {name: "A Fire Upon the Deep", "author": "Vernor Vinge", "release_date": "1992-06-01", "page_count": 613} } },
-        { index: { _index: index, data: {name: "Ender's Game", "author": "Orson Scott Card", "release_date": "1985-06-01", "page_count": 324} } },
-        { index: { _index: index, data: {name: "1984", "author": "George Orwell", "release_date": "1985-06-01", "page_count": 328} } },
-        { index: { _index: index, data: {name: "Fahrenheit 451", "author": "Ray Bradbury", "release_date": "1953-10-15", "page_count": 227} } },
-        { index: { _index: index, data: {name: "Brave New World", "author": "Aldous Huxley", "release_date": "1932-06-01", "page_count": 268} } },
-        { index: { _index: index, data: {name: "Foundation", "author": "Isaac Asimov", "release_date": "1951-06-01", "page_count": 224} } },
-        { index: { _index: index, data: {name: "The Giver", "author": "Lois Lowry", "release_date": "1993-04-26", "page_count": 208} } },
-        { index: { _index: index, data: {name: "Slaughterhouse-Five", "author": "Kurt Vonnegut", "release_date": "1969-06-01", "page_count": 275} } },
-        { index: { _index: index, data: {name: "The Hitchhiker's Guide to the Galaxy", "author": "Douglas Adams", "release_date": "1979-10-12", "page_count": 180} } },
-        { index: { _index: index, data: {name: "Snow Crash", "author": "Neal Stephenson", "release_date": "1992-06-01", "page_count": 470} } },
-        { index: { _index: index, data: {name: "Neuromancer", "author": "William Gibson", "release_date": "1984-07-01", "page_count": 271} } },
-        { index: { _index: index, data: {name: "The Handmaid's Tale", "author": "Margaret Atwood", "release_date": "1985-06-01", "page_count": 311} } },
-        { index: { _index: index, data: {name: "Starship Troopers", "author": "Robert A. Heinlein", "release_date": "1959-12-01", "page_count": 335} } },
-        { 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')
-    end
-
-    after do
-      client.indices.delete(index: index)
-    end
-
-    it 'instantiates a scroll helper' do
-      expect(scroll_helper).to be_an_instance_of Elasticsearch::Helpers::ScrollHelper
-    end
-
-    it 'searches an index' do
-      my_documents = []
-      while !(documents = scroll_helper.results).empty?
-        my_documents << documents
-      end
-
-      expect(my_documents.flatten.size).to eq 24
-    end
-
-    it 'uses enumerable' do
-      count = 0
-      scroll_helper.each { count += 1 }
-      expect(count).to eq 24
-      expect(scroll_helper).to respond_to(:count)
-      expect(scroll_helper).to respond_to(:reject)
-      expect(scroll_helper).to respond_to(:uniq)
-      expect(scroll_helper.map { |a| a['_id'] }.uniq.count).to eq 24
-    end
-  end
-end

data/spec/integration/opentelemetry_spec.rb

@@ -1,55 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-if ENV['TEST_WITH_OTEL'] == 'true'
-  ELASTICSEARCH_URL = ENV['TEST_ES_SERVER'] || "http://localhost:#{(ENV['PORT'] || 9200)}"
-  raise URI::InvalidURIError unless ELASTICSEARCH_URL =~ /\A#{URI::DEFAULT_PARSER.make_regexp}\z/
-
-  require 'spec_helper'
-
-  context 'OpenTelemetry' do
-    let(:exporter) { EXPORTER }
-    before { exporter.reset }
-    after { exporter.reset }
-    let(:span) { exporter.finished_spans[0] }
-
-    let(:client) do
-      Elasticsearch::Client.new(
-        host: ELASTICSEARCH_URL,
-        user: 'elastic',
-        password: 'changeme'
-      )
-    end
-
-    after do
-      client.delete(index: 'myindex', id: 1); rescue
-    end
-
-    context 'when a request is instrumented' do
-      it 'sets the span name to the endpoint id' do
-        client.search(body: { query: { match: { a: 1 } } })
-        expect(span.name).to eq 'search'
-      end
-
-      it 'sets the path parts' do
-        client.index(index: 'myindex', id: 1, body: { title: 'Test' })
-        expect(span.attributes['db.elasticsearch.path_parts.index']).to eq 'myindex'
-        expect(span.attributes['db.elasticsearch.path_parts.id']).to eq 1
-      end
-    end
-  end
-end

data/spec/spec_helper.rb

@@ -1,43 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-require 'elasticsearch'
-require 'rspec'
-
-RSpec.configure do |config|
-  config.formatter = :documentation
-end
-
-def meta_version
-  client.send(:client_meta_version, Elasticsearch::VERSION)
-end
-
-def jruby?
-  defined?(JRUBY_VERSION)
-end
-
-if ENV['TEST_WITH_OTEL'] == 'true'
-  require 'opentelemetry-sdk'
-  EXPORTER = OpenTelemetry::SDK::Trace::Export::InMemorySpanExporter.new
-  span_processor = OpenTelemetry::SDK::Trace::Export::SimpleSpanProcessor.new(EXPORTER)
-
-  OpenTelemetry::SDK.configure do |c|
-    c.error_handler = ->(exception:, message:) { raise(exception || message) }
-    c.logger = Logger.new($stderr, level: ENV.fetch('OTEL_LOG_LEVEL', 'fatal').to_sym)
-    c.add_span_processor span_processor
-  end
-end

data/spec/unit/api_key_spec.rb

@@ -1,138 +0,0 @@
-# Licensed to Elasticsearch B.V. under one or more contributor
-# license agreements. See the NOTICE file distributed with
-# this work for additional information regarding copyright
-# ownership. Elasticsearch B.V. licenses this file to you under
-# the Apache License, Version 2.0 (the "License"); you may
-# not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing,
-# software distributed under the License is distributed on an
-# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-# KIND, either express or implied.  See the License for the
-# specific language governing permissions and limitations
-# under the License.
-
-require 'spec_helper'
-require 'base64'
-
-describe Elasticsearch::Client do
-  context 'when using API Key' do
-    let(:authorization_header) do
-      client.transport.connections.first.connection.headers['Authorization']
-    end
-
-    context 'when an encoded api_key is provided' do
-      let(:client) do
-        described_class.new(api_key: 'an_api_key')
-      end
-
-      it 'Adds the ApiKey header to the connection' do
-        expect(authorization_header).to eq('ApiKey an_api_key')
-      end
-    end
-
-    context 'when an un-encoded api_key is provided' do
-      let(:client) do
-        described_class.new(api_key: { id: 'my_id', api_key: 'my_api_key' })
-      end
-
-      it 'Adds the ApiKey header to the connection' do
-        expect(authorization_header).to eq("ApiKey #{Base64.strict_encode64('my_id:my_api_key')}")
-      end
-    end
-
-    context 'when basic auth and api_key are provided' do
-      let(:client) do
-        described_class.new(
-          api_key: { id: 'my_id', api_key: 'my_api_key' },
-          host: 'http://elastic:password@localhost:9200'
-        )
-      end
-
-      it 'removes basic auth credentials' do
-        expect(authorization_header).not_to match(/^Basic/)
-        expect(authorization_header).to match(/^ApiKey/)
-      end
-    end
-
-    context 'when other headers were specified' do
-      let(:client) do
-        described_class.new(
-          api_key: 'elasticsearch_api_key',
-          transport_options: { headers: { 'x-test-header' => 'test' } }
-        )
-      end
-
-      it 'Adds the ApiKey header to the connection and keeps the header' do
-        header = client.transport.connections.first.connection.headers
-        expect(header['Authorization']).to eq('ApiKey elasticsearch_api_key')
-        expect(header['X-Test-Header']).to eq('test')
-      end
-    end
-
-    context 'when sending transport_options but no headers were specified' do
-      let(:client) do
-        described_class.new(
-          api_key: 'elasticsearch_api_key',
-          transport_options: { ssl: { verify: false } }
-        )
-      end
-
-      it 'Adds the ApiKey header to the connection and keeps the options' do
-        header = client.transport.connections.first.connection.headers
-        expect(header['Authorization']).to eq('ApiKey elasticsearch_api_key')
-        expect(client.transport.options[:transport_options]).to include({ ssl: { verify: false } })
-        expect(client.transport.options[:transport_options][:headers]).to include('Authorization' => 'ApiKey elasticsearch_api_key')
-      end
-    end
-
-    context 'when other headers and options were specified' do
-      let(:client) do
-        described_class.new(
-          api_key: 'elasticsearch_api_key',
-          transport_options: {
-            headers: { 'x-test-header' => 'test' },
-            ssl: { verify: false }
-          }
-        )
-      end
-
-      it 'Adds the ApiKey header to the connection and keeps the header' do
-        header = client.transport.connections.first.connection.headers
-        expect(header['X-Test-Header']).to eq('test')
-        expect(header['Authorization']).to eq('ApiKey elasticsearch_api_key')
-        expect(client.transport.options[:transport_options]).to include({ ssl: { verify: false } })
-        expect(client.transport.options[:transport_options][:headers]).to include('Authorization' => 'ApiKey elasticsearch_api_key')
-      end
-    end
-
-    context 'Metaheader' do
-      let(:adapter_code) { "nh=#{defined?(Net::HTTP::VERSION) ? Net::HTTP::VERSION : Net::HTTP::HTTPVersion}" }
-      let(:meta_header) do
-        if jruby?
-          "es=#{meta_version},rb=#{RUBY_VERSION},t=#{Elastic::Transport::VERSION},jv=#{ENV_JAVA['java.version']},jr=#{JRUBY_VERSION},fd=#{Faraday::VERSION},#{adapter_code}"
-        else
-          "es=#{meta_version},rb=#{RUBY_VERSION},t=#{Elastic::Transport::VERSION},fd=#{Faraday::VERSION},#{adapter_code}"
-        end
-      end
-
-      context 'when using API Key' do
-        let(:client) do
-          described_class.new(api_key: 'an_api_key')
-        end
-
-        let(:headers) do
-          client.transport.connections.first.connection.headers
-        end
-
-        it 'adds the ApiKey header to the connection' do
-          expect(authorization_header).to eq('ApiKey an_api_key')
-          expect(headers).to include('x-elastic-client-meta' => meta_header)
-        end
-      end
-    end
-  end
-end