mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
acfdcf5568
* Move specific requires close to where they're needed. * Use the private active_support/rails dep to wrap up common needs like eager autoload and module delegation. * Use a single Rails engine rather than an engine and a railtie. * Prefer `AS::JSON.encode` to `Object#to_json`.
116 lines
3.1 KiB
Ruby
116 lines
3.1 KiB
Ruby
require 'test_helper'
|
|
|
|
class ActionCable::Connection::SubscriptionsTest < ActionCable::TestCase
|
|
class Connection < ActionCable::Connection::Base
|
|
attr_reader :websocket
|
|
|
|
def send_async(method, *args)
|
|
# Bypass Celluloid
|
|
send method, *args
|
|
end
|
|
end
|
|
|
|
class ChatChannel < ActionCable::Channel::Base
|
|
attr_reader :room, :lines
|
|
|
|
def subscribed
|
|
@room = Room.new params[:id]
|
|
@lines = []
|
|
end
|
|
|
|
def speak(data)
|
|
@lines << data
|
|
end
|
|
end
|
|
|
|
setup do
|
|
@server = TestServer.new
|
|
@server.stubs(:channel_classes).returns(ChatChannel.name => ChatChannel)
|
|
|
|
@chat_identifier = ActiveSupport::JSON.encode(id: 1, channel: 'ActionCable::Connection::SubscriptionsTest::ChatChannel')
|
|
end
|
|
|
|
test "subscribe command" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
channel = subscribe_to_chat_channel
|
|
|
|
assert_kind_of ChatChannel, channel
|
|
assert_equal 1, channel.room.id
|
|
end
|
|
end
|
|
|
|
test "subscribe command without an identifier" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
|
|
@subscriptions.execute_command 'command' => 'subscribe'
|
|
assert @subscriptions.identifiers.empty?
|
|
end
|
|
end
|
|
|
|
test "unsubscribe command" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
subscribe_to_chat_channel
|
|
|
|
channel = subscribe_to_chat_channel
|
|
channel.expects(:unsubscribe_from_channel)
|
|
|
|
@subscriptions.execute_command 'command' => 'unsubscribe', 'identifier' => @chat_identifier
|
|
assert @subscriptions.identifiers.empty?
|
|
end
|
|
end
|
|
|
|
test "unsubscribe command without an identifier" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
|
|
@subscriptions.execute_command 'command' => 'unsubscribe'
|
|
assert @subscriptions.identifiers.empty?
|
|
end
|
|
end
|
|
|
|
test "message command" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
channel = subscribe_to_chat_channel
|
|
|
|
data = { 'content' => 'Hello World!', 'action' => 'speak' }
|
|
@subscriptions.execute_command 'command' => 'message', 'identifier' => @chat_identifier, 'data' => ActiveSupport::JSON.encode(data)
|
|
|
|
assert_equal [ data ], channel.lines
|
|
end
|
|
end
|
|
|
|
test "unsubscrib from all" do
|
|
run_in_eventmachine do
|
|
setup_connection
|
|
|
|
channel1 = subscribe_to_chat_channel
|
|
|
|
channel2_id = ActiveSupport::JSON.encode(id: 2, channel: 'ActionCable::Connection::SubscriptionsTest::ChatChannel')
|
|
channel2 = subscribe_to_chat_channel(channel2_id)
|
|
|
|
channel1.expects(:unsubscribe_from_channel)
|
|
channel2.expects(:unsubscribe_from_channel)
|
|
|
|
@subscriptions.unsubscribe_from_all
|
|
end
|
|
end
|
|
|
|
private
|
|
def subscribe_to_chat_channel(identifier = @chat_identifier)
|
|
@subscriptions.execute_command 'command' => 'subscribe', 'identifier' => identifier
|
|
assert_equal identifier, @subscriptions.identifiers.last
|
|
|
|
@subscriptions.send :find, 'identifier' => identifier
|
|
end
|
|
|
|
def setup_connection
|
|
env = Rack::MockRequest.env_for "/test", 'HTTP_CONNECTION' => 'upgrade', 'HTTP_UPGRADE' => 'websocket'
|
|
@connection = Connection.new(@server, env)
|
|
|
|
@subscriptions = ActionCable::Connection::Subscriptions.new(@connection)
|
|
end
|
|
end
|