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

Add Missing ActiveSupport::Rescuable to ActionCable::Channel

[timthez, Ilia Kasianenko]
This commit is contained in:
Ilia Kasianenko 2018-12-12 15:42:34 +02:00
parent c4f37cc8c5
commit 87f407db3e
2 changed files with 21 additions and 1 deletions

View file

@ -1,6 +1,7 @@
# frozen_string_literal: true
require "set"
require "active_support/rescuable"
module ActionCable
module Channel
@ -99,6 +100,7 @@ module ActionCable
include Streams
include Naming
include Broadcasting
include ActiveSupport::Rescuable
attr_reader :params, :connection, :identifier
delegate :logger, to: :connection
@ -267,6 +269,8 @@ module ActionCable
else
public_send action
end
rescue Exception => exception
rescue_with_handler(exception) || raise
end
def action_signature(action, data)

View file

@ -26,6 +26,9 @@ class ActionCable::Channel::BaseTest < ActionCable::TestCase
after_subscribe :toggle_subscribed
after_unsubscribe :toggle_subscribed
class SomeCustomError < StandardError; end
rescue_from SomeCustomError, with: :error_handler
def initialize(*)
@subscribed = false
super
@ -68,10 +71,18 @@ class ActionCable::Channel::BaseTest < ActionCable::TestCase
@last_action = [ :receive ]
end
def error_action
raise SomeCustomError
end
private
def rm_rf
@last_action = [ :rm_rf ]
end
def error_handler
@last_action = [ :error_action ]
end
end
setup do
@ -168,7 +179,7 @@ class ActionCable::Channel::BaseTest < ActionCable::TestCase
end
test "actions available on Channel" do
available_actions = %w(room last_action subscribed unsubscribed toggle_subscribed leave speak subscribed? get_latest receive chatters topic).to_set
available_actions = %w(room last_action subscribed unsubscribed toggle_subscribed leave speak subscribed? get_latest receive chatters topic error_action).to_set
assert_equal available_actions, ChatChannel.action_methods
end
@ -256,6 +267,11 @@ class ActionCable::Channel::BaseTest < ActionCable::TestCase
end
end
test "behaves like rescuable" do
@channel.perform_action "action" => :error_action
assert_equal [ :error_action ], @channel.last_action
end
private
def assert_logged(message)
old_logger = @connection.logger