# 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 "#{File.expand_path(File.dirname('..'))}/api-spec-testing/test_file"
require "#{File.expand_path(File.dirname('..'))}/api-spec-testing/rspec_matchers"
require "#{File.expand_path(File.dirname('..'))}/api-spec-testing/wipe_cluster"
include Elasticsearch::RestAPIYAMLTests

PROJECT_PATH = File.join(File.dirname(__FILE__), '..')
host = ENV['TEST_ES_SERVER'] || 'http://localhost:9200'
raise URI::InvalidURIError unless host =~ /\A#{URI::DEFAULT_PARSER.make_regexp}\z/

test_suite = ENV['TEST_SUITE'] || 'free'
password = ENV['ELASTIC_PASSWORD'] || 'changeme'
uri = URI.parse(host)

if test_suite == 'platinum'
  raw_certificate = File.read(File.join(PROJECT_PATH, '../.ci/certs/testnode.crt'))
  certificate = OpenSSL::X509::Certificate.new(raw_certificate)
  raw_key = File.read(File.join(PROJECT_PATH, '../.ci/certs/testnode.key'))
  key = OpenSSL::PKey::RSA.new(raw_key)
  ca_file = File.expand_path(File.join(PROJECT_PATH, '/.ci/certs/ca.crt'))
  host = "https://elastic:#{password}@#{uri.host}:#{uri.port}".freeze
  transport_options = { ssl: { verify: false, client_cert: certificate, client_key: key, ca_file: ca_file } }
else
  host = "http://elastic:#{password}@#{uri.host}:#{uri.port}".freeze
  transport_options = {}
end

ADMIN_CLIENT = Elasticsearch::Client.new(host: host, transport_options: transport_options)

DEFAULT_CLIENT = if ENV['QUIET'] == 'true'
                   Elasticsearch::Client.new(host: host, transport_options: transport_options)
                 else
                   Elasticsearch::Client.new(
                     host: host,
                     tracer: Logger.new($stdout),
                     transport_options: transport_options
                   )
                 end

unless ADMIN_CLIENT.ping
  raise('Couldn\'t connect to Elasticsearch cluster')
  exit
end

Elasticsearch::RestAPIYAMLTests::WipeCluster.create_xpack_rest_user(ADMIN_CLIENT) if test_suite == 'platinum'

YAML_FILES_DIRECTORY = "#{PROJECT_PATH}/../tmp/rest-api-spec/test/#{test_suite}".freeze

SINGLE_TEST = if ENV['SINGLE_TEST'] && !ENV['SINGLE_TEST'].empty?
                test_target = ENV['SINGLE_TEST']

                if test_target.match?(/\.yml$/)
                  ["#{PROJECT_PATH}/../tmp/rest-api-spec/test/#{test_suite}/#{test_target}"]
                else
                  Dir.glob(
                    ["#{PROJECT_PATH}/../tmp/rest-api-spec/test/#{test_suite}/#{test_target}/*.yml"]
                  )
                end
              end

# Skipped tests
file = File.expand_path("#{__dir__}/skipped_tests_#{test_suite}.yml")
skipped_tests = YAML.load_file(file)

# The directory of rest api YAML files.
REST_API_YAML_FILES = if ENV['RUN_SKIPPED_TESTS'] # only run the skipped tests if true
                        SKIPPED_TESTS = []
                        skipped_tests.map { |test| "#{YAML_FILES_DIRECTORY}/#{test[:file]}" }
                      else
                        # If not, define the skipped tests constant and try the single test or all
                        # the tests
                        SKIPPED_TESTS = skipped_tests
                        SINGLE_TEST || Dir.glob("#{YAML_FILES_DIRECTORY}/**/*.yml")
                      end

# The features to skip
REST_API_YAML_SKIP_FEATURES = ['warnings', 'node_selector'].freeze