mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
3ae8eb13cb
The `WorkerTest`'s `Receiver` is imporsonating an `ActionCable::Connection::Base`, but just delegates the logger to `ActionCable.logger`. This creates a mismatch as the connection requires the logger to be a `TaggedLoggerProxy`'ied logger, while the server doesn't. Thus to ensure an exception isn't raised when the worker tries to call `tag` other tests have to assign a proxied logger to their test server. Instead of forcing change on other tests, have Receiver adhere to the connection contract and use a `TaggedLoggerProxy`. As a consequence remove more setup from the tests.
54 lines
1.2 KiB
Ruby
54 lines
1.2 KiB
Ruby
require 'test_helper'
|
|
|
|
class WorkerTest < ActiveSupport::TestCase
|
|
class Receiver
|
|
attr_accessor :last_action
|
|
|
|
def run
|
|
@last_action = :run
|
|
end
|
|
|
|
def process(message)
|
|
@last_action = [ :process, message ]
|
|
end
|
|
|
|
def connection
|
|
self
|
|
end
|
|
|
|
def logger
|
|
# Impersonating a connection requires a TaggedLoggerProxy'ied logger.
|
|
inner_logger = Logger.new(StringIO.new).tap { |l| l.level = Logger::UNKNOWN }
|
|
ActionCable::Connection::TaggedLoggerProxy.new(inner_logger, tags: [])
|
|
end
|
|
end
|
|
|
|
setup do
|
|
@worker = ActionCable::Server::Worker.new
|
|
@receiver = Receiver.new
|
|
end
|
|
|
|
teardown do
|
|
@receiver.last_action = nil
|
|
end
|
|
|
|
test "invoke" do
|
|
@worker.invoke @receiver, :run
|
|
assert_equal :run, @receiver.last_action
|
|
end
|
|
|
|
test "invoke with arguments" do
|
|
@worker.invoke @receiver, :process, "Hello"
|
|
assert_equal [ :process, "Hello" ], @receiver.last_action
|
|
end
|
|
|
|
test "running periodic timers with a proc" do
|
|
@worker.run_periodic_timer @receiver, @receiver.method(:run)
|
|
assert_equal :run, @receiver.last_action
|
|
end
|
|
|
|
test "running periodic timers with a method" do
|
|
@worker.run_periodic_timer @receiver, :run
|
|
assert_equal :run, @receiver.last_action
|
|
end
|
|
end
|