forked from elastic/elasticsearch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
es_spec_helper.rb
48 lines (40 loc) · 1.05 KB
/
es_spec_helper.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
require "logstash/devutils/rspec/spec_helper"
require 'manticore'
require 'elasticsearch'
module ESHelper
def get_host_port
"127.0.0.1:9200"
end
def get_client
Elasticsearch::Client.new(:hosts => [get_host_port])
end
def get_doc_type
if ESHelper.es_version_satisfies?(">=7")
"_doc"
else
"doc"
end
end
def get_mapping_name
if ESHelper.es_version_satisfies?(">=7")
"_doc"
else
"_default_"
end
end
def self.es_version
RSpec.configuration.filter[:es_version] || ENV['ES_VERSION']
end
def self.es_version_satisfies?(*requirement)
es_version = RSpec.configuration.filter[:es_version] || ENV['ES_VERSION']
if es_version.nil?
puts "Info: ES_VERSION environment or 'es_version' tag wasn't set. Returning false to all `es_version_satisfies?` call."
return false
end
es_release_version = Gem::Version.new(es_version).release
Gem::Requirement.new(requirement).satisfied_by?(es_release_version)
end
end
RSpec.configure do |config|
config.include ESHelper
end