mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
046e32656e
ActionCable does some things behind the scenes that expects these "broadcasting"s or "channel"s to be strings. However it's not immediately obvious that the value must be a string. So adding this conversion ensures things work as expected.
90 lines
2.9 KiB
Ruby
90 lines
2.9 KiB
Ruby
require 'test_helper'
|
|
require 'stubs/test_connection'
|
|
require 'stubs/room'
|
|
|
|
class ActionCable::Channel::StreamTest < ActionCable::TestCase
|
|
class ChatChannel < ActionCable::Channel::Base
|
|
def subscribed
|
|
if params[:id]
|
|
@room = Room.new params[:id]
|
|
stream_from "test_room_#{@room.id}"
|
|
end
|
|
end
|
|
|
|
def send_confirmation
|
|
transmit_subscription_confirmation
|
|
end
|
|
end
|
|
|
|
class SymbolChannel < ActionCable::Channel::Base
|
|
def subscribed
|
|
stream_from :channel
|
|
end
|
|
end
|
|
|
|
test "streaming start and stop" do
|
|
run_in_eventmachine do
|
|
connection = TestConnection.new
|
|
connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("test_room_1", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }
|
|
channel = ChatChannel.new connection, "{id: 1}", { id: 1 }
|
|
|
|
connection.expects(:pubsub).returns mock().tap { |m| m.expects(:unsubscribe) }
|
|
channel.unsubscribe_from_channel
|
|
end
|
|
end
|
|
|
|
test "stream from non-string channel" do
|
|
run_in_eventmachine do
|
|
connection = TestConnection.new
|
|
connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("channel", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }
|
|
channel = SymbolChannel.new connection, ""
|
|
|
|
connection.expects(:pubsub).returns mock().tap { |m| m.expects(:unsubscribe) }
|
|
channel.unsubscribe_from_channel
|
|
end
|
|
end
|
|
|
|
test "stream_for" do
|
|
run_in_eventmachine do
|
|
connection = TestConnection.new
|
|
connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("action_cable:channel:stream_test:chat:Room#1-Campfire", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }
|
|
|
|
channel = ChatChannel.new connection, ""
|
|
channel.stream_for Room.new(1)
|
|
end
|
|
end
|
|
|
|
test "stream_from subscription confirmation" do
|
|
run_in_eventmachine do
|
|
connection = TestConnection.new
|
|
|
|
ChatChannel.new connection, "{id: 1}", { id: 1 }
|
|
assert_nil connection.last_transmission
|
|
|
|
wait_for_async
|
|
|
|
expected = ActiveSupport::JSON.encode "identifier" => "{id: 1}", "type" => "confirm_subscription"
|
|
connection.transmit(expected)
|
|
|
|
assert_equal expected, connection.last_transmission, "Did not receive subscription confirmation within 0.1s"
|
|
end
|
|
end
|
|
|
|
test "subscription confirmation should only be sent out once" do
|
|
run_in_eventmachine do
|
|
connection = TestConnection.new
|
|
|
|
channel = ChatChannel.new connection, "test_channel"
|
|
channel.send_confirmation
|
|
channel.send_confirmation
|
|
|
|
wait_for_async
|
|
|
|
expected = ActiveSupport::JSON.encode "identifier" => "test_channel", "type" => "confirm_subscription"
|
|
assert_equal expected, connection.last_transmission, "Did not receive subscription confirmation"
|
|
|
|
assert_equal 1, connection.transmissions.size
|
|
end
|
|
end
|
|
|
|
end
|