57 lines
1.6 KiB
Ruby
57 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "test_helper"
|
|
require "minitest/mock"
|
|
require "stubs/test_connection"
|
|
require "stubs/room"
|
|
|
|
class ActionCable::Channel::RejectionTest < ActionCable::TestCase
|
|
class SecretChannel < ActionCable::Channel::Base
|
|
def subscribed
|
|
reject if params[:id] > 0
|
|
end
|
|
|
|
def secret_action
|
|
end
|
|
end
|
|
|
|
setup do
|
|
@user = User.new "lifo"
|
|
@connection = TestConnection.new(@user)
|
|
end
|
|
|
|
test "subscription rejection" do
|
|
subscriptions = Minitest::Mock.new
|
|
subscriptions.expect(:remove_subscription, SecretChannel, [SecretChannel])
|
|
|
|
@connection.stub(:subscriptions, subscriptions) do
|
|
@channel = SecretChannel.new @connection, "{id: 1}", id: 1
|
|
@channel.subscribe_to_channel
|
|
|
|
expected = { "identifier" => "{id: 1}", "type" => "reject_subscription" }
|
|
assert_equal expected, @connection.last_transmission
|
|
end
|
|
|
|
assert subscriptions.verify
|
|
end
|
|
|
|
test "does not execute action if subscription is rejected" do
|
|
subscriptions = Minitest::Mock.new
|
|
subscriptions.expect(:remove_subscription, SecretChannel, [SecretChannel])
|
|
|
|
@connection.stub(:subscriptions, subscriptions) do
|
|
@channel = SecretChannel.new @connection, "{id: 1}", id: 1
|
|
@channel.subscribe_to_channel
|
|
|
|
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
|
|
|
|
assert subscriptions.verify
|
|
end
|
|
end
|