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/subscription_adapter/common.rb

116 lines
2.9 KiB
Ruby
Raw Normal View History

require "test_helper"
require "concurrent"
2016-01-24 05:43:40 -05:00
require "active_support/core_ext/hash/indifferent_access"
require "pathname"
2016-01-24 05:43:40 -05:00
module CommonSubscriptionAdapterTest
WAIT_WHEN_EXPECTING_EVENT = 3
WAIT_WHEN_NOT_EXPECTING_EVENT = 0.2
def setup
server = ActionCable::Server::Base.new
server.config.cable = cable_config.with_indifferent_access
server.config.logger = Logger.new(StringIO.new).tap { |l| l.level = Logger::UNKNOWN }
2016-01-24 05:43:40 -05:00
adapter_klass = server.config.pubsub_adapter
@rx_adapter = adapter_klass.new(server)
@tx_adapter = adapter_klass.new(server)
end
def teardown
[@rx_adapter, @tx_adapter].uniq.compact.each(&:shutdown)
2016-01-24 05:43:40 -05:00
end
def subscribe_as_queue(channel, adapter = @rx_adapter)
queue = Queue.new
callback = -> data { queue << data }
subscribed = Concurrent::Event.new
adapter.subscribe(channel, callback, Proc.new { subscribed.set })
subscribed.wait(WAIT_WHEN_EXPECTING_EVENT)
assert subscribed.set?
yield queue
sleep WAIT_WHEN_NOT_EXPECTING_EVENT
assert_empty queue
ensure
adapter.unsubscribe(channel, callback) if subscribed.set?
end
def test_subscribe_and_unsubscribe
subscribe_as_queue("channel") do |queue|
2016-01-24 05:43:40 -05:00
end
end
def test_basic_broadcast
subscribe_as_queue("channel") do |queue|
@tx_adapter.broadcast("channel", "hello world")
2016-01-24 05:43:40 -05:00
assert_equal "hello world", queue.pop
2016-01-24 05:43:40 -05:00
end
end
def test_broadcast_after_unsubscribe
keep_queue = nil
subscribe_as_queue("channel") do |queue|
2016-01-24 05:43:40 -05:00
keep_queue = queue
@tx_adapter.broadcast("channel", "hello world")
2016-01-24 05:43:40 -05:00
assert_equal "hello world", queue.pop
2016-01-24 05:43:40 -05:00
end
@tx_adapter.broadcast("channel", "hello void")
2016-01-24 05:43:40 -05:00
sleep WAIT_WHEN_NOT_EXPECTING_EVENT
assert_empty keep_queue
end
def test_multiple_broadcast
subscribe_as_queue("channel") do |queue|
@tx_adapter.broadcast("channel", "bananas")
@tx_adapter.broadcast("channel", "apples")
2016-01-24 05:43:40 -05:00
received = []
2.times { received << queue.pop }
assert_equal ["apples", "bananas"], received.sort
2016-01-24 05:43:40 -05:00
end
end
def test_identical_subscriptions
subscribe_as_queue("channel") do |queue|
subscribe_as_queue("channel") do |queue_2|
@tx_adapter.broadcast("channel", "hello")
2016-01-24 05:43:40 -05:00
assert_equal "hello", queue_2.pop
2016-01-24 05:43:40 -05:00
end
assert_equal "hello", queue.pop
2016-01-24 05:43:40 -05:00
end
end
def test_simultaneous_subscriptions
subscribe_as_queue("channel") do |queue|
subscribe_as_queue("other channel") do |queue_2|
@tx_adapter.broadcast("channel", "apples")
@tx_adapter.broadcast("other channel", "oranges")
2016-01-24 05:43:40 -05:00
assert_equal "apples", queue.pop
assert_equal "oranges", queue_2.pop
2016-01-24 05:43:40 -05:00
end
end
end
def test_channel_filtered_broadcast
subscribe_as_queue("channel") do |queue|
@tx_adapter.broadcast("other channel", "one")
@tx_adapter.broadcast("channel", "two")
2016-01-24 05:43:40 -05:00
assert_equal "two", queue.pop
2016-01-24 05:43:40 -05:00
end
end
end