86 lines
2.0 KiB
Ruby
86 lines
2.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require "test_helper"
|
|
require "stubs/test_connection"
|
|
require "stubs/room"
|
|
require "active_support/time"
|
|
|
|
class ActionCable::Channel::PeriodicTimersTest < ActionCable::TestCase
|
|
class ChatChannel < ActionCable::Channel::Base
|
|
# Method name arg
|
|
periodically :send_updates, every: 1
|
|
|
|
# Proc arg
|
|
periodically -> { ping }, every: 2
|
|
|
|
# Block arg
|
|
periodically every: 3 do
|
|
ping
|
|
end
|
|
|
|
private
|
|
def ping
|
|
end
|
|
end
|
|
|
|
setup do
|
|
@connection = TestConnection.new
|
|
end
|
|
|
|
test "periodic timers definition" do
|
|
timers = ChatChannel.periodic_timers
|
|
|
|
assert_equal 3, timers.size
|
|
|
|
timers.each_with_index do |timer, i|
|
|
assert_kind_of Proc, timer[0]
|
|
assert_equal i + 1, timer[1][:every]
|
|
end
|
|
end
|
|
|
|
test "disallow negative and zero periods" do
|
|
[ 0, 0.0, 0.seconds, -1, -1.seconds, "foo", :foo, Object.new ].each do |invalid|
|
|
e = assert_raise ArgumentError do
|
|
ChatChannel.periodically :send_updates, every: invalid
|
|
end
|
|
assert_match(/Expected every:/, e.message)
|
|
end
|
|
end
|
|
|
|
test "disallow block and arg together" do
|
|
e = assert_raise ArgumentError do
|
|
ChatChannel.periodically(:send_updates, every: 1) { ping }
|
|
end
|
|
assert_match(/not both/, e.message)
|
|
end
|
|
|
|
test "disallow unknown args" do
|
|
[ "send_updates", Object.new, nil ].each do |invalid|
|
|
e = assert_raise ArgumentError do
|
|
ChatChannel.periodically invalid, every: 1
|
|
end
|
|
assert_match(/Expected a Symbol/, e.message)
|
|
end
|
|
end
|
|
|
|
test "timer start and stop" do
|
|
mock = Minitest::Mock.new
|
|
3.times { mock.expect(:shutdown, nil) }
|
|
|
|
assert_called(
|
|
@connection.server.event_loop,
|
|
:timer,
|
|
times: 3,
|
|
returns: mock
|
|
) do
|
|
channel = ChatChannel.new @connection, "{id: 1}", id: 1
|
|
|
|
channel.subscribe_to_channel
|
|
channel.unsubscribe_from_channel
|
|
assert_equal [], channel.send(:active_periodic_timers)
|
|
end
|
|
|
|
assert mock.verify
|
|
end
|
|
end
|