forward ping to @driver - with test

celluloid · Feb 21, 2016 · de30b45 · de30b45
1 parent dbfaf49
commit de30b45
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/lib/reel/websocket.rb b/lib/reel/websocket.rb
@@ -10,6 +10,7 @@ class WebSocket
 
     attr_reader :socket
     def_delegators :@socket, :addr, :peeraddr
+    def_delegators :@driver, :ping
 
     def initialize(info, connection)
       driver_env = DriverEnvironment.new(info, connection.socket)      

diff --git a/spec/reel/websocket_spec.rb b/spec/reel/websocket_spec.rb
@@ -173,6 +173,20 @@ def initialize(websocket)
     end
   end
 
+  it "pings clients" do
+    with_websocket_pair do |client, websocket|
+      websocket.ping "<3"
+
+      pinger = double "pinger"
+      expect(pinger).to receive(:ping) { |msg| expect(msg).to eq("<3") }
+
+      parser = WebSocket::Parser.new
+      parser.on_ping { |msg| pinger.ping msg }
+
+      parser << client.readpartial(4096)
+    end
+  end
+
   def with_websocket_pair
     with_socket_pair do |client, peer|
       connection = Reel::Connection.new(peer)