2017-07-16 13:10:15 -04:00
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2016-08-06 13:15:15 -04:00
|
|
|
require "test_helper"
|
|
|
|
require "stubs/test_connection"
|
|
|
|
require "stubs/room"
|
2015-10-20 18:17:18 -04:00
|
|
|
|
|
|
|
class ActionCable::Channel::RejectionTest < ActiveSupport::TestCase
|
|
|
|
class SecretChannel < ActionCable::Channel::Base
|
|
|
|
def subscribed
|
2015-11-05 10:37:15 -05:00
|
|
|
reject if params[:id] > 0
|
2015-10-20 18:17:18 -04:00
|
|
|
end
|
2016-02-18 12:31:04 -05:00
|
|
|
|
|
|
|
def secret_action
|
|
|
|
end
|
2015-10-20 18:17:18 -04:00
|
|
|
end
|
|
|
|
|
|
|
|
setup do
|
|
|
|
@user = User.new "lifo"
|
|
|
|
@connection = TestConnection.new(@user)
|
|
|
|
end
|
|
|
|
|
|
|
|
test "subscription rejection" do
|
|
|
|
@connection.expects(:subscriptions).returns mock().tap { |m| m.expects(:remove_subscription).with instance_of(SecretChannel) }
|
2016-08-06 13:44:11 -04:00
|
|
|
@channel = SecretChannel.new @connection, "{id: 1}", id: 1
|
2016-09-20 19:57:10 -04:00
|
|
|
@channel.subscribe_to_channel
|
2015-10-20 18:17:18 -04:00
|
|
|
|
2016-03-11 18:32:02 -05:00
|
|
|
expected = { "identifier" => "{id: 1}", "type" => "reject_subscription" }
|
2015-10-20 18:17:18 -04:00
|
|
|
assert_equal expected, @connection.last_transmission
|
|
|
|
end
|
2016-02-18 12:31:04 -05:00
|
|
|
|
|
|
|
test "does not execute action if subscription is rejected" do
|
|
|
|
@connection.expects(:subscriptions).returns mock().tap { |m| m.expects(:remove_subscription).with instance_of(SecretChannel) }
|
|
|
|
@channel = SecretChannel.new @connection, "{id: 1}", id: 1
|
2016-09-20 19:57:10 -04:00
|
|
|
@channel.subscribe_to_channel
|
2016-02-18 12:31:04 -05:00
|
|
|
|
|
|
|
expected = { "identifier" => "{id: 1}", "type" => "reject_subscription" }
|
|
|
|
assert_equal expected, @connection.last_transmission
|
|
|
|
assert_equal 1, @connection.transmissions.size
|
|
|
|
|
|
|
|
@channel.perform_action("action" => :secret_action)
|
|
|
|
assert_equal 1, @connection.transmissions.size
|
|
|
|
end
|
2015-10-20 18:17:18 -04:00
|
|
|
end
|