Merge pull request #23723 from mwear/action_cable_notifications

Add ActiveSupport::Notification to Channel::Base#perform_action

1 files changed, 69 insertions, 0 deletions

diff --git a/actioncable/test/channel/log_subscriber_test.rb b/actioncable/test/channel/log_subscriber_test.rb
new file mode 100644
index 0000000000..18acdc9fbd
--- /dev/null
+++ b/actioncable/test/channel/log_subscriber_test.rb
@@ -0,0 +1,69 @@
+require 'test_helper'
+require 'stubs/test_connection'
+require 'active_support/log_subscriber/test_helper'
+require 'action_cable/channel/log_subscriber'
+
+class ActionCable::Channel::LogSubscriberTest < ActiveSupport::TestCase
+  include ActiveSupport::LogSubscriber::TestHelper
+
+  class ChatChannel < ActionCable::Channel::Base
+    attr_reader :last_action
+
+    def speak(data)
+      @last_action = [ :speak, data ]
+    end
+
+    def get_latest
+      transmit data: 'latest'
+    end
+  end
+
+  def setup
+    super
+    @connection = TestConnection.new
+    @channel = ChatChannel.new @connection, "{id: 1}", { id: 1 }
+    ActionCable::Channel::LogSubscriber.attach_to :action_cable
+  end
+
+  def test_perform_action
+    data = {'action' => :speak, 'content' => 'hello'}
+    @channel.perform_action(data)
+    wait
+
+    assert_equal(1, logs.size)
+    assert_match(/Completed #{channel_class}#speak in \d+ms/, logs.first)
+  end
+
+  def test_transmit
+    @channel.perform_action('action' => :get_latest)
+    wait
+
+    assert_equal(2, logs.size)
+    assert_match(/^#{channel_class} transmitting/, logs.first)
+  end
+
+  def test_transmit_subscription_confirmation
+    @channel.stubs(:subscription_confirmation_sent?).returns(false)
+    @channel.send(:transmit_subscription_confirmation)
+    wait
+
+    assert_equal(1, logs.size)
+    assert_equal("#{channel_class} is transmitting the subscription confirmation", logs.first)
+  end
+
+  def test_transmit_subscription_rejection
+    @channel.send(:transmit_subscription_rejection)
+    wait
+
+    assert_equal(1, logs.size)
+    assert_equal("#{channel_class} is transmitting the subscription rejection", logs.first)
+  end
+
+  def channel_class
+    "ActionCable::Channel::LogSubscriberTest::ChatChannel"
+  end
+
+  def logs
+    @logs ||= @logger.logged(:info)
+  end
+end