Fix stream tests

1 files changed, 2 insertions, 2 deletions

diff --git a/test/channel/stream_test.rb b/test/channel/stream_test.rb
index 5914b39be0..4e0248d7b4 100644
--- a/test/channel/stream_test.rb
+++ b/test/channel/stream_test.rb
@@ -18,7 +18,7 @@ class ActionCable::Channel::StreamTest < ActionCable::TestCase
 
   test "streaming start and stop" do
     run_in_eventmachine do
-      @connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("test_room_1") }
+      @connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("test_room_1").returns stub_everything(:pubsub) }
       channel = ChatChannel.new @connection, "{id: 1}", { id: 1 }
 
       @connection.expects(:pubsub).returns mock().tap { |m| m.expects(:unsubscribe_proc) }
@@ -29,7 +29,7 @@ class ActionCable::Channel::StreamTest < ActionCable::TestCase
   test "stream_for" do
     run_in_eventmachine do
       EM.next_tick do
-        @connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("action_cable:channel:stream_test:chat:Room#1-Campfire") }
+        @connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("action_cable:channel:stream_test:chat:Room#1-Campfire").returns stub_everything(:pubsub) }
       end
 
       channel = ChatChannel.new @connection, ""