require_relative "../../../spec/es_spec_helper"
require "logstash/outputs/elasticsearch"

if ESHelper.es_version_satisfies?("<= 5.x")
  context "when using elasticsearch 5.x and before", :integration => true do
    shared_examples "a type based parent indexer" do
      let(:index) { 10.times.collect { rand(10).to_s }.join("") }
      let(:type) { 10.times.collect { rand(10).to_s }.join("") }
      let(:event_count) { 10000 + rand(500) }
      let(:parent) { "not_implemented" }
      let(:config) { "not_implemented" }
      let(:default_headers) {
        {"Content-Type" => "application/json"}
      }
      subject { LogStash::Outputs::ElasticSearch.new(config) }

      before do
        # Add mapping and a parent document
        index_url = "http://#{get_host_port()}/#{index}"
        mapping = { "mappings" => { "#{type}" => { "_parent" => { "type" => "#{type}_parent" } } } }
        Manticore.put("#{index_url}", {:body => mapping.to_json, :headers => default_headers}).call
        pdoc = { "foo" => "bar" }
        Manticore.put("#{index_url}/#{type}_parent/test", {:body => pdoc.to_json, :headers => default_headers}).call

        subject.register
        subject.multi_receive(event_count.times.map { LogStash::Event.new("link_to" => "test", "message" => "Hello World!", "type" => type) })
      end


      it "ships events" do
        index_url = "http://#{get_host_port()}/#{index}"

        Manticore.post("#{index_url}/_refresh").call

        # Wait until all events are available.
        Stud::try(10.times) do
          query = { "query" => { "has_parent" => { "type" => "#{type}_parent", "query" => { "match" => { "foo" => "bar" } } } } }
          response = Manticore.post("#{index_url}/_count", {:body => query.to_json, :headers => default_headers})
          data = response.body
          result = LogStash::Json.load(data)
          cur_count = result["count"]
          insist { cur_count } == event_count
        end
      end
    end

    describe "(http protocol) index events with static parent" do
      it_behaves_like 'a type based parent indexer' do
        let(:parent) { "test" }
        let(:config) {
          {
            "hosts" => get_host_port,
            "index" => index,
            "parent" => parent
          }
        }
      end
    end

    describe "(http_protocol) index events with fieldref in parent value" do
      it_behaves_like 'a type based parent indexer' do
        let(:config) {
          {
            "hosts" => get_host_port,
            "index" => index,
            "parent" => "%{link_to}"
          }
        }
      end
    end
  end
end

if ESHelper.es_version_satisfies?(">= 5.6")
  context "when using elasticsearch 5.6 and above", :integration => true do

    shared_examples "a join field based parent indexer" do
      let(:index) { 10.times.collect { rand(10).to_s }.join("") }
      let(:type) { 10.times.collect { rand(10).to_s }.join("") }
      let(:event_count) { 10000 + rand(500) }
      let(:parent) { "not_implemented" }
      let(:config) { "not_implemented" }
      let(:parent_id) { "test" }
      let(:join_field) { "join_field" }
      let(:parent_relation) { "parent_type" }
      let(:child_relation) { "child_type" }
      let(:default_headers) {
        {"Content-Type" => "application/json"}
      }
      subject { LogStash::Outputs::ElasticSearch.new(config) }

      before do
        # Add mapping and a parent document
        index_url = "http://#{get_host_port()}/#{index}"
        mapping = {
          "mappings" => {
            type => {
              "properties" => {
                join_field => {
                  "type" => "join",
                  "relations" => { parent_relation => child_relation }
                }
              }
            }
          }
        }
        if ESHelper.es_version_satisfies?('<6')
          mapping.merge!({
                 "settings" => {
                   "mapping.single_type" => true
                 }})
        end
        Manticore.put("#{index_url}", {:body => mapping.to_json, :headers => default_headers}).call
        pdoc = { "message" => "ohayo", join_field => parent_relation }
        Manticore.put("#{index_url}/#{type}/#{parent_id}", {:body => pdoc.to_json, :headers => default_headers}).call

        subject.register
        subject.multi_receive(event_count.times.map { LogStash::Event.new("link_to" => parent_id, "message" => "Hello World!", join_field => child_relation) })
      end


      it "ships events" do
        index_url = "http://#{get_host_port()}/#{index}"

        Manticore.post("#{index_url}/_refresh").call

        # Wait until all events are available.
        Stud::try(10.times) do
          query = { "query" => { "has_parent" => { "parent_type" => parent_relation, "query" => { "match_all" => { } } } } }
          response = Manticore.post("#{index_url}/_count", {:body => query.to_json, :headers => default_headers})
          data = response.body
          result = LogStash::Json.load(data)
          cur_count = result["count"]
          insist { cur_count } == event_count
        end
      end
    end

    describe "(http protocol) index events with static parent" do
      it_behaves_like 'a join field based parent indexer' do
        let(:config) {
          {
            "hosts" => get_host_port,
            "index" => index,
            "parent" => parent_id,
            "document_type" => type,
            "join_field" => join_field,
            "manage_template" => false
          }
        }
      end
    end

    describe "(http_protocol) index events with fieldref in parent value" do
      it_behaves_like 'a join field based parent indexer' do
        let(:config) {
          {
            "hosts" => get_host_port,
            "index" => index,
            "parent" => "%{link_to}",
            "document_type" => type,
            "join_field" => join_field,
            "manage_template" => false
          }
        }
      end
    end
  end
end