# encoding: utf-8
require "logstash/devutils/rspec/spec_helper"
require "logstash/outputs/lmlogs"
require "logstash/event"

describe LogStash::Outputs::LMLogs do
  let(:sample_event) { LogStash::Event.new("message" => "hello this is log","event" => {"sequence" => 0,"original" => "simple logstash msg from me  running with proxy settings"})}
  let(:client) { @lmlogs.client }
  let(:sample_lm_logs_event){{"message" => "hello this is log 1", "_lm.resourceId" => {"test.property" => "host1"}, "timestamp" => "2021-03-22T04:28:55.907121106Z"}}

  before do
    @lmlogs = LogStash::Outputs::LMLogs.new(
      "portal_name" => "localhost",
      "access_id" => "access_id",
      "access_key" => "access_key",
      "batch_size" => 3,
      "lm_property" => "test.property"
      )
    @lmlogs.register
    allow(@lmlogs).to receive(:client).and_return(client)
    allow(client).to receive(:post).and_call_original
  end

  before do
    allow(@lmlogs).to receive(:client).and_return(client)
  end

  it "Forwards an event" do
      expect(client).to receive(:post).once.and_call_original
      @lmlogs.multi_receive([sample_event])
  end

  # TODO fix failing test case
  # it "Batches multiple events and extracts metadata" do
  #   event1 = LogStash::Event.new("message" => "hello this is log 1", "host" => "host1")
  #   event2 = LogStash::Event.new("message" => "hello this is log 2", "host" => "host2")
  #   event3 = LogStash::Event.new("message" => "hello this is log 3", "host" => "host3")
  #   expect(client).to receive(:post).once.with("https://localhost.logicmonitor.com/rest/log/ingest",hash_including(body: 
  #       [{ "host" => "host1", "message" => "hello this is log 1","@version":"1", "_lm.resourceId" => {"test.property" => "host1"}, "timestamp" => event1.timestamp.to_s,},
  #       {"host" => "host2","message" => "hello this is log 2","@version":"1", "_lm.resourceId" => {"test.property" => "host2"}, "timestamp" => event2.timestamp.to_s},
  #       {"host" => "host3","message" => "hello this is log 3","@version":"1", "_lm.resourceId" => {"test.property" => "host3"}, "timestamp" => event3.timestamp.to_s}
  #       ].to_json
  #       )).and_call_original
  #   @lmlogs.multi_receive([event1, event2, event3])
  # end

  it "Batches data of size batch_size" do
    expect(client).to receive(:post).exactly(2).times.and_call_original
    @lmlogs.multi_receive([sample_event, sample_event, sample_event, sample_event, sample_event])
  end

  it "max payload exceeded" do

    document = [sample_lm_logs_event,sample_lm_logs_event]

    lm_logs_event = {"message" => "hello this is log 3", "_lm.resourceId" => {"test.property" => "host3"}, "timestamp" => "2021-03-22T04:28:55.909421106Z"}
    document_expected = [lm_logs_event]
    expect(client).to receive(:post).once.with("https://localhost.logicmonitor.com/rest/log/ingest",hash_including(:body => LogStash::Json.dump(
        [sample_lm_logs_event,sample_lm_logs_event]
        ))).and_call_original

    document_result = @lmlogs.isValidPayloadSize(document,lm_logs_event,document.to_json.bytesize)
    expect(document_result).to eq(document_expected)
  end
  it "max payload in limit" do

    document = [sample_lm_logs_event]

    lm_logs_event = {"message" => "hello this is log 2", "_lm.resourceId" => {"test.property" => "host3"}, "timestamp" => "2021-03-22T04:28:55.909421106Z"}

    document_expected = [sample_lm_logs_event,lm_logs_event]
    expect(client).to receive(:post).exactly(0).times.and_call_original

    document_result = @lmlogs.isValidPayloadSize(document,lm_logs_event,document.to_json.bytesize + lm_logs_event.to_json.bytesize)
    expect(document_result).to eq(document_expected)
  end

end