1 files changed, 28 insertions, 0 deletions
diff --git a/actioncable/test/test_helper.rb b/actioncable/test/test_helper.rb
new file mode 100644
index 0000000000..8ddbd4e764
--- /dev/null
+++ b/actioncable/test/test_helper.rb
@@ -0,0 +1,28 @@
+require File.expand_path('../../../load_paths', __FILE__)
+
+require 'action_cable'
+require 'active_support/testing/autorun'
+
+
+require 'puma'
+
+require 'mocha/setup'
+
+require 'rack/mock'
+
+# Require all the stubs and models
+Dir[File.dirname(__FILE__) + '/stubs/*.rb'].each {|file| require file }
+
+class ActionCable::TestCase < ActiveSupport::TestCase
+  def wait_for_async
+    e = Concurrent.global_io_executor
+    until e.completed_task_count == e.scheduled_task_count
+      sleep 0.1
+    end
+  end
+
+  def run_in_eventmachine
+    yield
+    wait_for_async
+  end
+end