# encoding: utf-8

require "logstash/codecs/plain"
require "logstash/event"
require "insist"

describe LogStash::Codecs::Plain do
  context "#decode" do
    it "should return a valid event" do
      subject.decode("Testing decoding.") do |event|
        insist { event.is_a? LogStash::Event }
      end
    end

    context "using default UTF-8 charset" do

      it "should decode valid UTF-8 input" do
        ["foobar", "κόσμε"].each do |data|
          insist { data.encoding.name } == "UTF-8"
          insist { data.valid_encoding? } == true
          subject.decode(data) do |event|
            insist { event["message"] } == data
            insist { event["message"].encoding.name } == "UTF-8"
          end
        end
      end

      it "should escape invalid sequences" do
        ["foo \xED\xB9\x81\xC3", "bar \xAD"].each do |data|
          insist { data.encoding.name } == "UTF-8"
          insist { data.valid_encoding? } == false
          subject.decode(data) do |event|
            insist { event["message"] } == data.inspect[1..-2]
            insist { event["message"].encoding.name } == "UTF-8"
          end
        end
      end
    end


    context "with valid non UTF-8 source encoding" do

      subject{LogStash::Codecs::Plain.new("charset" => "ISO-8859-1")}

      it "should encode to UTF-8" do
        samples = [
          ["foobar", "foobar"],
          ["\xE0 Montr\xE9al", "à Montréal"],
        ]
        samples.map{|(a, b)| [a.force_encoding("ISO-8859-1"), b]}.each do |(a, b)|
          insist { a.encoding.name } == "ISO-8859-1"
          insist { b.encoding.name } == "UTF-8"
          insist { a.valid_encoding? } == true

          subject.decode(a) do |event|
            insist { event["message"] } == b
            insist { event["message"].encoding.name } == "UTF-8"
          end
        end
      end
    end

    context "with invalid non UTF-8 source encoding" do

      subject{LogStash::Codecs::Plain.new("charset" => "ASCII-8BIT")}

      it "should encode to UTF-8" do
        samples = [
          ["\xE0 Montr\xE9al", "� Montr�al"],
          ["\xCE\xBA\xCF\x8C\xCF\x83\xCE\xBC\xCE\xB5", "����������"],
        ]
        samples.map{|(a, b)| [a.force_encoding("ASCII-8BIT"), b]}.each do |(a, b)|
          insist { a.encoding.name } == "ASCII-8BIT"
          insist { b.encoding.name } == "UTF-8"
          insist { a.valid_encoding? } == true

          subject.decode(a) do |event|
            insist { event["message"] } == b
            insist { event["message"].encoding.name } == "UTF-8"
          end
        end
      end
    end
  end

  context "#encode" do
    it "should return a plain text encoding" do
      event = LogStash::Event.new
      event["message"] = "Hello World."
      subject.on_event do |data|
        insist { data } == event.to_s
      end
      subject.encode(event)
    end

    it "should respect the format setting" do
      format = "%{[hello]} %{[something][fancy]}"
      codec = LogStash::Codecs::Plain.new("format" => format)
      event = LogStash::Event.new("hello" => "world", "something" => { "fancy" => 123 })
      codec.on_event do |data|
        insist { data } == event.sprintf(format)
      end
      codec.encode(event)
    end
  end
end