require "helper"
require "fluent/plugin/filter_kubernetes_log_level.rb"

class KubernetesLogLevelFilterTest < Test::Unit::TestCase
  setup do
    Fluent::Test.setup
    @expected_warning = [{
      'level'      => 'warning',
      'kubernetes' => {
        'labels' => {
          'logging-level' => 'warning'
        }
      }
    }]

    @expected_error = [{
      'level'  => 'error',
      'kubernetes' => {
        'labels' => {
          'logging-level' => 'warning'
        }
      }
    }]

    @expected_info = []

    @expected_static_warning_default = [{
      'level' => 'warning'
    }]

    @expected_warning_serilog = [{
      'level'      => 'Warning',
      'kubernetes' => {
        'labels' => {
          'logging-level' => 'Warning'
        }
      }
    }]

  end

  CONFIG = %[
    log_level_label logging-level
    log_level_key level
    default_logging_level warning
  ]

 # private

  def create_driver(conf = CONFIG)
    Fluent::Test::Driver::Filter.new(Fluent::Plugin::KubernetesLogLevelFilter).configure(conf)
  end

  def filter(msg, time = event_time("2017-07-12 19:20:21 UTC"))
    d = create_driver
    d.run { d.feed('kubernetes', time, msg) }
    d.filtered_records
  end

  def test_default_configuration
    d = create_driver
    assert_equal 'logging-level', d.instance.config['log_level_label']
    assert_equal 'level', d.instance.config['log_level_key']
    assert_equal 'warning', d.instance.config['default_logging_level']
  end

  def test_log_level_higher_then_threshold
    assert_equal @expected_error, filter({"level"=>"error", "kubernetes"=>{"labels"=>{"logging-level"=>"warning"}}})
  end 

  def test_log_level_equal_to_threshold
    assert_equal @expected_warning, filter({"level"=>"warning", "kubernetes"=>{"labels"=>{"logging-level"=>"warning"}}})
  end 

  def test_log_level_lower_then_threshold
    assert_equal @expected_info, filter({"level"=>"info", "kubernetes"=>{"labels"=>{"logging-level"=>"warning"}}})
  end 

  def test_static_log
    assert_equal @expected_static_warning_default, filter({"level"=>"warning"})
  end

  def test_serilog_structure
    assert_equal @expected_warning_serilog, filter({"level"=>"Warning", "kubernetes"=>{"labels"=>{"logging-level"=>"Warning"}}})
  end
end