1 files changed, 33 insertions, 0 deletions
diff --git a/actioncable/test/stubs/test_connection.rb b/actioncable/test/stubs/test_connection.rb
new file mode 100644
index 0000000000..885450dda6
--- /dev/null
+++ b/actioncable/test/stubs/test_connection.rb
@@ -0,0 +1,33 @@
+require 'stubs/user'
+
+class TestConnection
+  attr_reader :identifiers, :logger, :current_user, :server, :transmissions
+
+  delegate :pubsub, to: :server
+
+  def initialize(user = User.new("lifo"), coder: ActiveSupport::JSON, subscription_adapter: SuccessAdapter)
+    @coder = coder
+    @identifiers = [ :current_user ]
+
+    @current_user = user
+    @logger = ActiveSupport::TaggedLogging.new ActiveSupport::Logger.new(StringIO.new)
+    @server = TestServer.new(subscription_adapter: subscription_adapter)
+    @transmissions = []
+  end
+
+  def transmit(cable_message)
+    @transmissions << encode(cable_message)
+  end
+
+  def last_transmission
+    decode @transmissions.last if @transmissions.any?
+  end
+
+  def decode(websocket_message)
+    @coder.decode websocket_message
+  end
+
+  def encode(cable_message)
+    @coder.encode cable_message
+  end
+end