require 'helper'
require 'fluent/plugin/out_record_serializer'
require 'yaml'

class RecordSerializerOutputTest < Test::Unit::TestCase
  def setup
    Fluent::Test.setup
  end

  TAG = 'test.serialize'
  FORMAT = 'json'
  FIELD_NAME = 'payload'

  CONFIG = %[
    tag #{TAG}
  ]

  YAML_CONFIG = %[
    tag #{TAG}
    format yaml
  ]

  def create_driver(conf=CONFIG)
    Fluent::Test::OutputTestDriver.new(Fluent::RecordSerializerOutput).configure(conf)
  end

  def test_configure
    d = create_driver
    assert_equal FORMAT, d.instance.format
    assert_equal FIELD_NAME, d.instance.field_name
  end

  def test_json
    record = {'spam' => 1, 'ham' => 'egg',}
    d = create_driver

    d.run do
      d.emit(record)
    end

    assert_equal [{'tag' => TAG, FIELD_NAME => record.to_json}], d.records
  end

  def test_yaml
    record = {'spam' => 1, 'ham' => 'egg',}
    d = create_driver YAML_CONFIG

    d.run do
      d.emit(record)
    end

    assert_equal [{'tag' => TAG, FIELD_NAME => record.to_yaml}], d.records
  end
end