aboutsummaryrefslogblamecommitdiffstats
path: root/actioncable/test/channel/stream_test.rb
blob: 526ea92e4f482f2fd38f1d5ecaffb80c8fa4c3fa (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11

                               
                    
 
                                                              

                                                



                                           
       



                                        

     





                                                  
                                    
                          
                                     
                                                                                                                                                                     
                                                                
 
                                                                                    

                                      
     
 










                                                                                                                                                                 
                      
                          
                                     
                                                                                                                                                                                                               
 
                                              

                                    
     

                                                 
                          
                                     
 
                                                      

                                             
                    
 

                                                                                                       
 
                                                                                                                  


       
                                                                  
                          
                                     




                                                          
                    

                                                                                                            
                                                                                                      

                                                   


       
   
require 'test_helper'
require 'stubs/test_connection'
require 'stubs/room'

class ActionCable::Channel::StreamTest < ActionCable::TestCase
  class ChatChannel < ActionCable::Channel::Base
    def subscribed
      if params[:id]
        @room = Room.new params[:id]
        stream_from "test_room_#{@room.id}"
      end
    end

    def send_confirmation
      transmit_subscription_confirmation
    end
  end

  class SymbolChannel < ActionCable::Channel::Base
    def subscribed
      stream_from :channel
    end
  end

  test "streaming start and stop" do
    run_in_eventmachine do
      connection = TestConnection.new
      connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("test_room_1", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }
      channel = ChatChannel.new connection, "{id: 1}", { id: 1 }

      connection.expects(:pubsub).returns mock().tap { |m| m.expects(:unsubscribe) }
      channel.unsubscribe_from_channel
    end
  end

  test "stream from non-string channel" do
    run_in_eventmachine do
      connection = TestConnection.new
      connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("channel", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }
      channel = SymbolChannel.new connection, ""

      connection.expects(:pubsub).returns mock().tap { |m| m.expects(:unsubscribe) }
      channel.unsubscribe_from_channel
    end
  end

  test "stream_for" do
    run_in_eventmachine do
      connection = TestConnection.new
      connection.expects(:pubsub).returns mock().tap { |m| m.expects(:subscribe).with("action_cable:channel:stream_test:chat:Room#1-Campfire", kind_of(Proc), kind_of(Proc)).returns stub_everything(:pubsub) }

      channel = ChatChannel.new connection, ""
      channel.stream_for Room.new(1)
    end
  end

  test "stream_from subscription confirmation" do
    run_in_eventmachine do
      connection = TestConnection.new

      ChatChannel.new connection, "{id: 1}", { id: 1 }
      assert_nil connection.last_transmission

      wait_for_async

      expected = ActiveSupport::JSON.encode "identifier" => "{id: 1}", "type" => "confirm_subscription"
      connection.transmit(expected)

      assert_equal expected, connection.last_transmission, "Did not receive subscription confirmation within 0.1s"
    end
  end

  test "subscription confirmation should only be sent out once" do
    run_in_eventmachine do
      connection = TestConnection.new

      channel = ChatChannel.new connection, "test_channel"
      channel.send_confirmation
      channel.send_confirmation

      wait_for_async

      expected = ActiveSupport::JSON.encode "identifier" => "test_channel", "type" => "confirm_subscription"
      assert_equal expected, connection.last_transmission, "Did not receive subscription confirmation"

      assert_equal 1, connection.transmissions.size
    end
  end

end