1 files changed, 54 insertions, 0 deletions
diff --git a/actioncable/test/worker_test.rb b/actioncable/test/worker_test.rb
new file mode 100644
index 0000000000..654f49821e
--- /dev/null
+++ b/actioncable/test/worker_test.rb
@@ -0,0 +1,54 @@
+require 'test_helper'
+
+class WorkerTest < ActiveSupport::TestCase
+  class Receiver
+    attr_accessor :last_action
+
+    def run
+      @last_action = :run
+    end
+
+    def process(message)
+      @last_action =  [ :process, message ]
+    end
+
+    def connection
+      self
+    end
+
+    def logger
+      # Impersonating a connection requires a TaggedLoggerProxy'ied logger.
+      inner_logger = Logger.new(StringIO.new).tap { |l| l.level = Logger::UNKNOWN }
+      ActionCable::Connection::TaggedLoggerProxy.new(inner_logger, tags: [])
+    end
+  end
+
+  setup do
+    @worker = ActionCable::Server::Worker.new
+    @receiver = Receiver.new
+  end
+
+  teardown do
+    @receiver.last_action = nil
+  end
+
+  test "invoke" do
+    @worker.invoke @receiver, :run
+    assert_equal :run, @receiver.last_action
+  end
+
+  test "invoke with arguments" do
+    @worker.invoke @receiver, :process, "Hello"
+    assert_equal [ :process, "Hello" ], @receiver.last_action
+  end
+
+  test "running periodic timers with a proc" do
+    @worker.run_periodic_timer @receiver, @receiver.method(:run)
+    assert_equal :run, @receiver.last_action
+  end
+
+  test "running periodic timers with a method" do
+    @worker.run_periodic_timer @receiver, :run
+    assert_equal :run, @receiver.last_action
+  end
+end