1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00
rails--rails/actioncable/test/connection/identifier_test.rb
2016-03-02 12:17:15 +10:30

77 lines
2.1 KiB
Ruby

require 'test_helper'
require 'stubs/test_server'
require 'stubs/user'
class ActionCable::Connection::IdentifierTest < ActionCable::TestCase
class Connection < ActionCable::Connection::Base
identified_by :current_user
attr_reader :websocket
public :process_internal_message
def connect
self.current_user = User.new "lifo"
end
end
test "connection identifier" do
run_in_eventmachine do
open_connection_with_stubbed_pubsub
assert_equal "User#lifo", @connection.connection_identifier
end
end
test "should subscribe to internal channel on open and unsubscribe on close" do
run_in_eventmachine do
pubsub = mock('pubsub_adapter')
pubsub.expects(:subscribe).with('action_cable/User#lifo', kind_of(Proc))
pubsub.expects(:unsubscribe).with('action_cable/User#lifo', kind_of(Proc))
server = TestServer.new
server.stubs(:pubsub).returns(pubsub)
open_connection server: server
close_connection
end
end
test "processing disconnect message" do
run_in_eventmachine do
open_connection_with_stubbed_pubsub
@connection.websocket.expects(:close)
message = ActiveSupport::JSON.encode('type' => 'disconnect')
@connection.process_internal_message message
end
end
test "processing invalid message" do
run_in_eventmachine do
open_connection_with_stubbed_pubsub
@connection.websocket.expects(:close).never
message = ActiveSupport::JSON.encode('type' => 'unknown')
@connection.process_internal_message message
end
end
protected
def open_connection_with_stubbed_pubsub
server = TestServer.new
server.stubs(:adapter).returns(stub_everything('adapter'))
open_connection server: server
end
def open_connection(server:)
env = Rack::MockRequest.env_for "/test", 'HTTP_HOST' => 'localhost', 'HTTP_CONNECTION' => 'upgrade', 'HTTP_UPGRADE' => 'websocket'
@connection = Connection.new(server, env)
@connection.process
@connection.send :handle_open
end
def close_connection
@connection.send :handle_close
end
end