require 'simplecov'

if ENV['COVERAGE']
  SimpleCov.start
end

require 'bundler/setup'
require 'algolia'
require 'minitest/autorun'
require 'minitest/hooks'
require 'algolia/integration/mocks/mock_requester'

APPLICATION_ID_1   = ENV['ALGOLIA_APPLICATION_ID_1']
ADMIN_KEY_1        = ENV['ALGOLIA_ADMIN_KEY_1']
SEARCH_KEY_1       = ENV['ALGOLIA_SEARCH_KEY_1']
APPLICATION_ID_2   = ENV['ALGOLIA_APPLICATION_ID_2']
ADMIN_KEY_2        = ENV['ALGOLIA_ADMIN_KEY_2']
MCM_APPLICATION_ID = ENV['ALGOLIA_APPLICATION_ID_MCM']
MCM_ADMIN_KEY      = ENV['ALGOLIA_ADMIN_KEY_MCM']
USER_AGENT         = 'test-ruby'

class Minitest::Test
  attr_reader :search_client

  include Minitest::Hooks
  include Helpers
  @@search_config = Algolia::Search::Config.new(application_id: APPLICATION_ID_1, api_key: ADMIN_KEY_1, user_agent: USER_AGENT)
  @@search_client = Algolia::Search::Client.new(@@search_config)
end

def assert_requests(requester, requests)
  refute_empty requests
  refute_nil requester

  actual_requests = requester.requests
  assert_equal requests.size, actual_requests.size

  requests.each_with_index do |expected_request, i|
    request = actual_requests[i]

    assert_equal(expected_request[:body], request[:body])
    assert_equal(expected_request[:method], request[:method])
    assert_equal(expected_request[:path], request[:path])
  end
end

def check_environment_variables
  raise Algolia::AlgoliaError, 'ALGOLIA_APPLICATION_ID_1 must be defined' if ENV['ALGOLIA_APPLICATION_ID_1'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_ADMIN_KEY_1 must be defined' if ENV['ALGOLIA_ADMIN_KEY_1'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_SEARCH_KEY_1 must be defined' if ENV['ALGOLIA_SEARCH_KEY_1'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_APPLICATION_ID_2 must be defined' if ENV['ALGOLIA_APPLICATION_ID_2'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_ADMIN_KEY_2 must be defined' if ENV['ALGOLIA_ADMIN_KEY_2'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_APPLICATION_ID_MCM must be defined' if ENV['ALGOLIA_APPLICATION_ID_MCM'].to_s.strip.empty?
  raise Algolia::AlgoliaError, 'ALGOLIA_ADMIN_KEY_MCM must be defined' if ENV['ALGOLIA_ADMIN_KEY_MCM'].to_s.strip.empty?
end

def get_test_index_name(name)
  date = DateTime.now.strftime('%Y-%m-%d_%H_%M_%S')
  user = ENV['USER'] || 'unknown'

  instance = ENV['CI'].to_s == 'true' ? ENV['CIRCLE_BUILD_NUM'] : user

  format('ruby_%<date>s_%<instance>s_%<name>s', date: date, instance: instance, name: name)
end

def get_mcm_user_name(user_id)
  date = DateTime.now.strftime('%Y-%m-%d')
  user = ENV['USER'] || 'unknown'

  instance = ENV['CI'].to_s == 'true' ? ENV['CIRCLE_BUILD_NUM'] : user

  format('ruby-%<date>s-%<instance>s-%<user_id>s', date: date, instance: instance, user_id: user_id)
end

def generate_object(object_id = nil)
  object = { property: 'property' }
  if object_id
    object[:objectID] = object_id
  end

  object
end

def create_employee_records
  [
    { company: 'Algolia', name: 'Julien Lemoine', objectID: 'julien-lemoine' },
    { company: 'Algolia', name: 'Nicolas Dessaigne', objectID: 'nicolas-dessaigne' },
    { company: 'Amazon', name: 'Jeff Bezos' },
    { company: 'Apple', name: 'Steve Jobs' },
    { company: 'Apple', name: 'Steve Wozniak' },
    { company: 'Arista Networks', name: 'Jayshree Ullal' },
    { company: 'Google', name: 'Larry Page' },
    { company: 'Google', name: 'Rob Pike' },
    { company: 'Google', name: 'Serguey Brin' },
    { company: 'Microsoft', name: 'Bill Gates' },
    { company: 'SpaceX', name: 'Elon Musk' },
    { company: 'Tesla', name: 'Elon Musk' },
    { company: 'Yahoo', name: 'Marissa Mayer' }
  ]
end

def rule_without_metadata(rule)
  rule.delete(:_metadata)
  rule
end

def retry_test(delay = 0.1, max_retries = 30)
  (1...max_retries).each do |i|
    begin
      return yield
    rescue Algolia::AlgoliaHttpError
      sleep delay * i
    end
  end

  raise StandardError, 'reached the maximum number of retries'
end