# frozen_string_literal: true

require 'test_helper'

class CaptureSinkTest < Minitest::Test
  def test_capture_sink_captures_datagram_instances
    capture_sink = StatsD::Instrument::CaptureSink.new(parent: [])
    capture_sink << 'foo:1|c'

    assert_equal(1, capture_sink.datagrams.length)
    assert_kind_of(StatsD::Instrument::Datagram, capture_sink.datagrams.first)
    assert_equal('foo:1|c', capture_sink.datagrams.first.source)
  end

  def test_capture_sink_sends_datagrams_to_parent
    parent = []
    capture_sink = StatsD::Instrument::CaptureSink.new(parent: parent)
    capture_sink << 'foo:1|c' << 'bar:1|c'

    assert_equal(['foo:1|c', 'bar:1|c'], parent)
  end

  def test_nesting_capture_sink_instances
    null_sink = StatsD::Instrument::NullSink.new
    outer_capture_sink = StatsD::Instrument::CaptureSink.new(parent: null_sink)
    inner_capture_sink = StatsD::Instrument::CaptureSink.new(parent: outer_capture_sink)

    outer_capture_sink << 'foo:1|c'
    inner_capture_sink << 'bar:1|c'

    assert_equal(['foo:1|c', 'bar:1|c'], outer_capture_sink.datagrams.map(&:source))
    assert_equal(['bar:1|c'], inner_capture_sink.datagrams.map(&:source))
  end

  def test_using_a_different_datagram_class
    sink = StatsD::Instrument::CaptureSink.new(parent: [], datagram_class: String)
    sink << 'foo:1|c'

    assert(sink.datagrams.all? { |datagram| datagram.is_a?(String) })
    assert_equal(['foo:1|c'], sink.datagrams)
  end
end