Confirm Websocket pong frames are received by handlers

1 files changed, 23 insertions, 0 deletions

diff --git a/test/handlers/ws_ping_h.erl b/test/handlers/ws_ping_h.erl
new file mode 100644
index 0000000..a5848fe
--- /dev/null
+++ b/test/handlers/ws_ping_h.erl
@@ -0,0 +1,23 @@
+%% This module sends an empty ping to the client and
+%% waits for a pong before sending a text frame. It
+%% is used to confirm server-initiated pings work.
+
+-module(ws_ping_h).
+-behavior(cowboy_websocket).
+
+-export([init/2]).
+-export([websocket_init/1]).
+-export([websocket_handle/2]).
+-export([websocket_info/2]).
+
+init(Req, _) ->
+	{cowboy_websocket, Req, undefined}.
+
+websocket_init(State) ->
+	{[{ping, <<>>}], State}.
+
+websocket_handle(pong, State) ->
+	{[{text, <<"OK!!">>}], State}.
+
+websocket_info(_, State) ->
+	{[], State}.