# frozen_string_literal: true

require 'test_helper'

class DogStatsDDatagramBuilderTest < Minitest::Test
  def setup
    @datagram_builder = StatsD::Instrument::DogStatsDDatagramBuilder.new
  end

  def test_raises_on_unsupported_metrics
    assert_raises(NotImplementedError) { @datagram_builder.kv('foo', 10, nil, nil) }
  end

  def test_simple_service_check
    datagram = @datagram_builder._sc('service', :ok)
    assert_equal '_sc|service|0', datagram
    parsed_datagram = StatsD::Instrument::DogStatsDDatagramBuilder.datagram_class.new(datagram)
    assert_equal :_sc, parsed_datagram.type
    assert_equal 'service', parsed_datagram.name
    assert_equal 0, parsed_datagram.value
  end

  def test_complex_service_check
    datagram = @datagram_builder._sc('service', :warning, timestamp: Time.parse('2019-09-30T04:22:12Z'),
      hostname: 'localhost', tags: { foo: 'bar|baz' }, message: 'blah')
    assert_equal "_sc|service|1|h:localhost|d:1569817332|#foo:barbaz|m:blah", datagram

    parsed_datagram = StatsD::Instrument::DogStatsDDatagramBuilder.datagram_class.new(datagram)
    assert_equal :_sc, parsed_datagram.type
    assert_equal 'service', parsed_datagram.name
    assert_equal 1, parsed_datagram.value
    assert_equal 'localhost', parsed_datagram.hostname
    assert_equal Time.parse('2019-09-30T04:22:12Z'), parsed_datagram.timestamp
    assert_equal ["foo:barbaz"], parsed_datagram.tags
    assert_equal 'blah', parsed_datagram.message
  end

  def test_simple_event
    datagram = @datagram_builder._e('hello', "world")
    assert_equal '_e{5,5}:hello|world', datagram

    parsed_datagram = StatsD::Instrument::DogStatsDDatagramBuilder.datagram_class.new(datagram)
    assert_equal :_e, parsed_datagram.type
    assert_equal 'hello', parsed_datagram.name
    assert_equal 'world', parsed_datagram.value
  end

  def test_complex_event
    datagram = @datagram_builder._e("testing", "with\nnewline", timestamp: Time.parse('2019-09-30T04:22:12Z'),
      hostname: 'localhost', aggregation_key: 'my-key', priority: 'low', source_type_name: 'source',
      alert_type: 'success', tags: { foo: 'bar|baz' })
    assert_equal '_e{7,13}:testing|with\\nnewline|h:localhost|d:1569817332|k:my-key|' \
      'p:low|s:source|t:success|#foo:barbaz', datagram

    parsed_datagram = StatsD::Instrument::DogStatsDDatagramBuilder.datagram_class.new(datagram)
    assert_equal :_e, parsed_datagram.type
    assert_equal 'testing', parsed_datagram.name
    assert_equal "with\nnewline", parsed_datagram.value
    assert_equal 'localhost', parsed_datagram.hostname
    assert_equal Time.parse('2019-09-30T04:22:12Z'), parsed_datagram.timestamp
    assert_equal ["foo:barbaz"], parsed_datagram.tags
    assert_equal "my-key", parsed_datagram.aggregation_key
    assert_equal "low", parsed_datagram.priority
    assert_equal "source", parsed_datagram.source_type_name
    assert_equal "success", parsed_datagram.alert_type
  end
end