discourse/spec/support/diagnostics_helper.rb

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

29 lines
605 B
Ruby
Raw Normal View History

# frozen_string_literal: true
2013-09-05 07:22:15 -04:00
module MessageBus::DiagnosticsHelper
def publish(channel, data, opts = nil)
id = super(channel, data, opts)
if @tracking && (@channel.nil? || @channel == channel)
2013-09-05 07:22:15 -04:00
m = MessageBus::Message.new(-1, id, channel, data)
m.user_ids = opts[:user_ids] if opts
m.group_ids = opts[:group_ids] if opts
@tracking << m
end
id
end
def track_publish(channel = nil)
@channel = channel
2013-09-05 07:22:15 -04:00
@tracking = tracking = []
yield
tracking
ensure
@tracking = nil
2013-09-05 07:22:15 -04:00
end
2013-09-05 07:22:15 -04:00
end
module MessageBus
extend MessageBus::DiagnosticsHelper
end