require 'helper'

class NotifierOutputStateTest < Test::Unit::TestCase
  TEST_DEF_DEFAULTS = {
    :tag => 'n', :tag_warn => nil, :tag_crit => nil,
    :interval_1st => 60, :repetitions_1st => 5,
    :interval_2nd => 300, :repetitions_2nd => 5,
    :interval_3rd => 1800
  }
  TEST_DEF_CONF1 = {
    'tag' => 'notify',
    'pattern' => 'name1', 'target_keys' => 'field1,field2',
    'check' => 'numeric_upward', 'warn_threshold' => '1', 'crit_threshold' => '2',
  }
  TEST_DEF_CONF2 = {
    'tag_warn' => 'warn', 'tag_crit' => 'crit',
    'pattern' => 'name2', 'target_key_pattern' => '^field\d$',
    'check' => 'string_find', 'warn_regexp' => 'WARN', 'crit_regexp' => 'CRIT',
    'interval_1st' => '5', 'repetitions_1st' => '1',
    'interval_2nd' => '6', 'repetitions_2nd' => '2',
    'interval_3rd' => '7'
  }

  def test_init
    s = Fluent::NotifierOutput::State.new({
        :pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'
      })
    assert_equal 'name1', s.pattern
    assert_equal 'test.tag', s.target_tag
    assert_equal 'field1', s.target_key
    assert_equal 'warn', s.level
    assert_equal 0, s.stage
    assert_equal 1, s.counter
    assert (s.first_notified <= Fluent::Engine.now)
    assert (s.last_notified <= Fluent::Engine.now)
  end

  def test_suppress?
    s = Fluent::NotifierOutput::State.new({
        :pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'
      })
    d = Fluent::NotifierOutput::Definition.new(TEST_DEF_CONF1, TEST_DEF_DEFAULTS)
    s.last_notified = Fluent::Engine.now - TEST_DEF_DEFAULTS[:interval_1st] + 5
    assert_equal true, s.suppress?(d, {:pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'})
    s.last_notified = Fluent::Engine.now - TEST_DEF_DEFAULTS[:interval_1st] - 5
    assert_equal false, s.suppress?(d, {:pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'})

    s = Fluent::NotifierOutput::State.new({
        :pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'
      })
    d = Fluent::NotifierOutput::Definition.new(TEST_DEF_CONF1, TEST_DEF_DEFAULTS)
    assert_equal true, s.suppress?(d, {:pattern => 'name1', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'crit'})
  end

  def test_update_notified
    s = Fluent::NotifierOutput::State.new({
        :pattern => 'name2', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'
      })
    d = Fluent::NotifierOutput::Definition.new(TEST_DEF_CONF2, TEST_DEF_DEFAULTS)

    assert_equal 0, s.stage
    assert_equal 1, s.counter

    s.update_notified(d, {:pattern => 'name2', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'})
    assert_equal 1, s.stage
    assert_equal 0, s.counter
    
    s.update_notified(d, {:pattern => 'name2', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'warn'})
    assert_equal 1, s.stage
    assert_equal 1, s.counter

    s.update_notified(d, {:pattern => 'name2', :target_tag => 'test.tag', :target_key => 'field1', 'level' => 'crit'})
    assert_equal 0, s.stage
    assert_equal 1, s.counter
  end
end