statsd-instrument 2.5.1 → 2.6.0

data/test/assertions_test.rb

@@ -10,141 +10,120 @@ class AssertionsTest < Minitest::Test
   end
 
   def test_assert_no_statsd_calls
-    assert_no_assertion_triggered do
-      @test_case.assert_no_statsd_calls('counter') do
-        # noop
-      end
+    @test_case.assert_no_statsd_calls('counter') do
+      # noop
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_no_statsd_calls('counter') do
-        StatsD.increment('other')
-      end
+    @test_case.assert_no_statsd_calls('counter') do
+      StatsD.increment('other')
     end
 
-    assert_assertion_triggered("No StatsD calls for metric counter expected.") do
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_no_statsd_calls('counter') do
         StatsD.increment('counter')
       end
     end
+    assert_equal assertion.message, "No StatsD calls for metric counter expected."
 
-    assert_assertion_triggered("No StatsD calls for metric other expected.") do
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_no_statsd_calls do
         StatsD.increment('other')
       end
     end
+    assert_equal assertion.message, "No StatsD calls for metric other expected."
 
-    assert_assertion_triggered("No StatsD calls for metric other, another expected.") do
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_no_statsd_calls do
         StatsD.increment('other')
         StatsD.increment('another')
       end
     end
+    assert_equal assertion.message, "No StatsD calls for metric other, another expected."
   end
 
   def test_assert_statsd_call
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter') do
-        StatsD.increment('counter')
-      end
+    @test_case.assert_statsd_increment('counter') do
+      StatsD.increment('counter')
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter') do
-        StatsD.increment('counter')
-        StatsD.increment('other')
-      end
+    @test_case.assert_statsd_increment('counter') do
+      StatsD.increment('counter')
+      StatsD.increment('other')
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter') do
         StatsD.increment('other')
       end
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter') do
         StatsD.gauge('counter', 42)
       end
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter') do
         StatsD.increment('counter')
         StatsD.increment('counter')
       end
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', times: 2) do
-        StatsD.increment('counter')
-        StatsD.increment('counter')
-      end
+    @test_case.assert_statsd_increment('counter', times: 2) do
+      StatsD.increment('counter')
+      StatsD.increment('counter')
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', times: 2, tags: ['foo:1']) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 1 })
-      end
+    @test_case.assert_statsd_increment('counter', times: 2, tags: ['foo:1']) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 1 })
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', times: 2, tags: ['foo:1']) do
         StatsD.increment('counter', tags: { foo: 1 })
         StatsD.increment('counter', tags: { foo: 2 })
       end
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a', 'b']) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: ['a', 'b'])
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a', 'b']) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: ['a', 'b'])
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a', 'b'], ignore_tags: ['b']) do
         StatsD.increment('counter', sample_rate: 0.5, tags: ['a'])
       end
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a'], ignore_tags: ['b']) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: ['a', 'b'])
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a'], ignore_tags: ['b']) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: ['a', 'b'])
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a'], ignore_tags: ['b']) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: ['a'])
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a'], ignore_tags: ['b']) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: ['a'])
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: { b: 2 }) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: { b: 2 }) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: { b: 2 }) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 3 })
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: { b: 2 }) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 3 })
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1, b: 3 }, ignore_tags: ['b']) do
         StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
       end
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: ['b']) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }, ignore_tags: ['b']) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: ['a', 'b']) do
         StatsD.increment('counter', sample_rate: 0.2, tags: ['c'])
       end
@@ -152,13 +131,11 @@ class AssertionsTest < Minitest::Test
   end
 
   def test_tags_will_match_subsets
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }) do
-        StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
-      end
+    @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1 }) do
+      StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2 })
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', sample_rate: 0.5, tags: { a: 1, b: 3 }) do
         StatsD.increment('counter', sample_rate: 0.5, tags: { a: 1, b: 2, c: 4 })
       end
@@ -166,66 +143,58 @@ class AssertionsTest < Minitest::Test
   end
 
   def test_tags_friendly_error
-    @test_case.assert_statsd_increment('counter', tags: { class: "AnotherJob" }) do
-      StatsD.increment('counter', tags: { class: "MyJob" })
+    assertion = assert_raises(Minitest::Assertion) do
+      @test_case.assert_statsd_increment('counter', tags: { class: "AnotherJob" }) do
+        StatsD.increment('counter', tags: { class: "MyJob" })
+      end
     end
-  rescue MiniTest::Assertion => assertion
-    assert_match(/Captured metrics with the same key/, assertion.message)
-    assert_match(/MyJob/, assertion.message)
+
+    assert_includes assertion.message, "Captured metrics with the same key"
+    assert_includes assertion.message, "MyJob"
   end
 
   def test_multiple_metrics_are_not_order_dependent
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
     end
 
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
     end
 
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
     end
 
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
     end
 
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-        StatsD.increment('counter', tags: { foo: 1 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_2_metric, foo_1_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
+      StatsD.increment('counter', tags: { foo: 1 })
     end
   end
 
   def test_assert_multiple_statsd_calls
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
       foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
       @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
@@ -234,7 +203,7 @@ class AssertionsTest < Minitest::Test
       end
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
       foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
       @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
@@ -245,60 +214,52 @@ class AssertionsTest < Minitest::Test
       end
     end
 
-    assert_no_assertion_triggered do
-      foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
-      foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
-      @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 1 })
-        StatsD.increment('counter', tags: { foo: 2 })
-      end
+    foo_1_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 2, tags: ['foo:1'])
+    foo_2_metric = StatsD::Instrument::MetricExpectation.new(type: :c, name: 'counter', times: 1, tags: ['foo:2'])
+    @test_case.assert_statsd_calls([foo_1_metric, foo_2_metric]) do
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 1 })
+      StatsD.increment('counter', tags: { foo: 2 })
     end
   end
 
   def test_assert_statsd_call_with_tags
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', tags: ['a:b', 'c:d']) do
-        StatsD.increment('counter', tags: { a: 'b', c: 'd' })
-      end
+    @test_case.assert_statsd_increment('counter', tags: ['a:b', 'c:d']) do
+      StatsD.increment('counter', tags: { a: 'b', c: 'd' })
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter', tags: { a: 'b', c: 'd' }) do
-        StatsD.increment('counter', tags: ['a:b', 'c:d'])
-      end
+    @test_case.assert_statsd_increment('counter', tags: { a: 'b', c: 'd' }) do
+      StatsD.increment('counter', tags: ['a:b', 'c:d'])
     end
   end
 
   def test_assert_statsd_call_with_wrong_sample_rate_type
     skip("In Strict mode, the StatsD.increment call will raise") if StatsD::Instrument.strict_mode_enabled?
-    assert_assertion_triggered "Unexpected sample rate type for metric counter, must be numeric" do
+
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter', tags: ['a', 'b']) do
         StatsD.increment('counter', sample_rate: 'abc', tags: ['a', 'b'])
       end
     end
+    assert_equal "Unexpected sample rate type for metric counter, must be numeric", assertion.message
   end
 
   def test_nested_assertions
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter1') do
-        @test_case.assert_statsd_increment('counter2') do
-          StatsD.increment('counter1')
-          StatsD.increment('counter2')
-        end
+    @test_case.assert_statsd_increment('counter1') do
+      @test_case.assert_statsd_increment('counter2') do
+        StatsD.increment('counter1')
+        StatsD.increment('counter2')
       end
     end
 
-    assert_no_assertion_triggered do
-      @test_case.assert_statsd_increment('counter1') do
-        StatsD.increment('counter1')
-        @test_case.assert_statsd_increment('counter2') do
-          StatsD.increment('counter2')
-        end
+    @test_case.assert_statsd_increment('counter1') do
+      StatsD.increment('counter1')
+      @test_case.assert_statsd_increment('counter2') do
+        StatsD.increment('counter2')
       end
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter1') do
         @test_case.assert_statsd_increment('counter2') do
           StatsD.increment('counter2')
@@ -306,7 +267,7 @@ class AssertionsTest < Minitest::Test
       end
     end
 
-    assert_assertion_triggered do
+    assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter1') do
         @test_case.assert_statsd_increment('counter2') do
           StatsD.increment('counter1')
@@ -316,62 +277,64 @@ class AssertionsTest < Minitest::Test
     end
   end
 
-  def test_assertion_with_exceptions
-    assert_no_assertion_triggered do
+  def test_assertion_block_with_expected_exceptions
+    @test_case.assert_statsd_increment('expected_happened') do
       @test_case.assert_raises(RuntimeError) do
-        @test_case.assert_statsd_increment('counter') do
-          StatsD.increment('counter')
-          raise "foo"
+        begin
+          raise "expected"
+        rescue
+          StatsD.increment('expected_happened')
+          raise
         end
       end
     end
 
-    assert_no_assertion_triggered do
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter') do
         @test_case.assert_raises(RuntimeError) do
-          StatsD.increment('counter')
-          raise "foo"
+          raise "expected"
         end
       end
     end
+    assert_includes assertion.message, "No StatsD calls for metric counter of type c were made"
+  end
 
-    assert_assertion_triggered do
+  def test_assertion_block_with_unexpected_exceptions
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_statsd_increment('counter') do
-        @test_case.assert_raises(RuntimeError) do
-          raise "foo"
-        end
+        StatsD.increment('counter')
+        raise "unexpected"
       end
     end
+    assert_includes assertion.message, "An exception occurred in the block provided to the StatsD assertion"
 
-    assert_assertion_triggered do
+    assertion = assert_raises(Minitest::Assertion) do
       @test_case.assert_raises(RuntimeError) do
         @test_case.assert_statsd_increment('counter') do
-          raise "foo"
+          StatsD.increment('counter')
+          raise "unexpected"
         end
       end
     end
-  end
-
-  private
+    assert_includes assertion.message, "An exception occurred in the block provided to the StatsD assertion"
 
-  def assert_no_assertion_triggered(&block)
-    block.call
-  rescue MiniTest::Assertion => assertion
-    flunk("No assertion trigger expected, but one was triggered with message #{assertion.message}.")
-  else
-    pass
+    assertion = assert_raises(Minitest::Assertion) do
+      @test_case.assert_raises(RuntimeError) do
+        @test_case.assert_no_statsd_calls do
+          raise "unexpected"
+        end
+      end
+    end
+    assert_includes assertion.message, "An exception occurred in the block provided to the StatsD assertion"
   end
 
-  def assert_assertion_triggered(message = nil, &block)
-    block.call
-  rescue MiniTest::Assertion => assertion
-    if message
-      assert_equal(message, assertion.message, "Assertion triggered, but message was not what was expected.")
-    else
-      pass
-    end
-    assertion
-  else
-    flunk("No assertion was triggered, but one was expected.")
+  def test_assertion_block_with_other_assertion_failures
+    # If another assertion failure happens inside the block, that failrue should have priority
+    assertion = assert_raises(Minitest::Assertion) do
+      @test_case.assert_statsd_increment('counter') do
+        @test_case.flunk('other assertion failure')
+      end
+    end
+    assert_equal "other assertion failure", assertion.message
   end
 end

data/test/capture_sink_test.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+require 'test_helper'
+
+require 'statsd/instrument/client'
+
+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