require 'elasticsearch/rails/tasks/import'

require 'elasticsearch/extensions/test/cluster/tasks'

namespace :elasticsearch do

desc 'Wipe the Ontology Index from ElasticSearch'

task :wipe => [:environment] do

Ontology.__elasticsearch__.delete_index!

end

task :run_cluster do

ENV['TEST_CLUSTER_NODES'] ||= '1'

env_timeout = ENV['TEST_CLUSTER_TIMEOUT']

timeout = env_timeout && !env_timeout.empty? ? env_timeout.to_i : 120

Signal.trap('INT') do

Elasticsearch::Extensions::Test::Cluster.stop timeout: timeout

end

Elasticsearch::Extensions::Test::Cluster.start timeout: timeout

loop {}

end

end