1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/activesupport/test/notifications/instrumenter_test.rb

67 lines
1.8 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
require_relative "../abstract_unit"
require "active_support/notifications/instrumenter"
module ActiveSupport
module Notifications
class InstrumenterTest < ActiveSupport::TestCase
class TestNotifier
attr_reader :starts, :finishes
def initialize
@starts = []
@finishes = []
end
def start(*args); @starts << args; end
def finish(*args); @finishes << args; end
end
attr_reader :instrumenter, :notifier, :payload
def setup
super
@notifier = TestNotifier.new
@instrumenter = Instrumenter.new @notifier
@payload = { foo: Object.new }
end
def test_instrument
called = false
instrumenter.instrument("foo", payload) {
called = true
}
assert called
end
def test_instrument_yields_the_payload_for_further_modification
assert_equal 2, instrumenter.instrument("awesome") { |p| p[:result] = 1 + 1 }
assert_equal 1, notifier.finishes.size
name, _, payload = notifier.finishes.first
assert_equal "awesome", name
assert_equal Hash[result: 2], payload
end
def test_instrument_works_without_a_block
instrumenter.instrument("no.block", payload)
assert_equal 1, notifier.finishes.size
assert_equal "no.block", notifier.finishes.first.first
end
def test_start
instrumenter.start("foo", payload)
assert_equal [["foo", instrumenter.id, payload]], notifier.starts
2018-01-25 18:16:57 -05:00
assert_empty notifier.finishes
end
def test_finish
instrumenter.finish("foo", payload)
assert_equal [["foo", instrumenter.id, payload]], notifier.finishes
2018-01-25 18:16:57 -05:00
assert_empty notifier.starts
end
end
end
end