7 files changed, 184 insertions, 0 deletions
diff --git a/test/ws_SUITE_data/ws_echo.erl b/test/ws_SUITE_data/ws_echo.erl
new file mode 100644
index 0000000..d4a5f07
--- /dev/null
+++ b/test/ws_SUITE_data/ws_echo.erl
@@ -0,0 +1,27 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_echo).
+-behaviour(cowboy_websocket_handler).
+-export([init/3]).
+-export([websocket_init/3, websocket_handle/3,
+	websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, _Opts) ->
+	Req2 = cowboy_req:compact(Req),
+	{ok, Req2, undefined}.
+
+websocket_handle({text, Data}, Req, State) ->
+	{reply, {text, Data}, Req, State};
+websocket_handle({binary, Data}, Req, State) ->
+	{reply, {binary, Data}, Req, State};
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info(_Info, Req, State) ->
+	{ok, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
diff --git a/test/ws_SUITE_data/ws_echo_timer.erl b/test/ws_SUITE_data/ws_echo_timer.erl
new file mode 100644
index 0000000..666a26d
--- /dev/null
+++ b/test/ws_SUITE_data/ws_echo_timer.erl
@@ -0,0 +1,31 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_echo_timer).
+-behaviour(cowboy_websocket_handler).
+-export([init/3]).
+-export([websocket_init/3, websocket_handle/3,
+	websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, _Opts) ->
+	erlang:start_timer(1000, self(), <<"websocket_init">>),
+	Req2 = cowboy_req:compact(Req),
+	{ok, Req2, undefined}.
+
+websocket_handle({text, Data}, Req, State) ->
+	{reply, {text, Data}, Req, State};
+websocket_handle({binary, Data}, Req, State) ->
+	{reply, {binary, Data}, Req, State};
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info({timeout, _Ref, Msg}, Req, State) ->
+	erlang:start_timer(1000, self(), <<"websocket_handle">>),
+	{reply, {text, Msg}, Req, State};
+websocket_info(_Info, Req, State) ->
+	{ok, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
diff --git a/test/ws_SUITE_data/ws_init_shutdown.erl b/test/ws_SUITE_data/ws_init_shutdown.erl
new file mode 100644
index 0000000..2b1dd99
--- /dev/null
+++ b/test/ws_SUITE_data/ws_init_shutdown.erl
@@ -0,0 +1,23 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_init_shutdown).
+-behaviour(cowboy_websocket_handler).
+-export([init/3]).
+-export([websocket_init/3, websocket_handle/3,
+	websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, _Opts) ->
+	{ok, Req2} = cowboy_req:reply(403, Req),
+	{shutdown, Req2}.
+
+websocket_handle(_Frame, _Req, _State) ->
+	exit(badarg).
+
+websocket_info(_Info, _Req, _State) ->
+	exit(badarg).
+
+websocket_terminate(_Reason, _Req, _State) ->
+	exit(badarg).
diff --git a/test/ws_SUITE_data/ws_send_many.erl b/test/ws_SUITE_data/ws_send_many.erl
new file mode 100644
index 0000000..2ed4772
--- /dev/null
+++ b/test/ws_SUITE_data/ws_send_many.erl
@@ -0,0 +1,27 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_send_many).
+-behaviour(cowboy_websocket_handler).
+
+-export([init/3]).
+-export([websocket_init/3]).
+-export([websocket_handle/3]).
+-export([websocket_info/3]).
+-export([websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, Sequence) ->
+	Req2 = cowboy_req:compact(Req),
+	erlang:send_after(10, self(), send_many),
+	{ok, Req2, Sequence}.
+
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info(send_many, Req, State = [{sequence, Sequence}]) ->
+	{reply, Sequence, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
diff --git a/test/ws_SUITE_data/ws_timeout_cancel.erl b/test/ws_SUITE_data/ws_timeout_cancel.erl
new file mode 100644
index 0000000..9c7b72b
--- /dev/null
+++ b/test/ws_SUITE_data/ws_timeout_cancel.erl
@@ -0,0 +1,26 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_timeout_cancel).
+-behaviour(cowboy_websocket_handler).
+-export([init/3]).
+-export([websocket_init/3, websocket_handle/3,
+	websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, _Opts) ->
+	erlang:start_timer(500, self(), should_not_cancel_timer),
+	{ok, Req, undefined, 1000}.
+
+websocket_handle({text, Data}, Req, State) ->
+	{reply, {text, Data}, Req, State};
+websocket_handle({binary, Data}, Req, State) ->
+	{reply, {binary, Data}, Req, State}.
+
+websocket_info(_Info, Req, State) ->
+	erlang:start_timer(500, self(), should_not_cancel_timer),
+	{ok, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
diff --git a/test/ws_SUITE_data/ws_timeout_hibernate.erl b/test/ws_SUITE_data/ws_timeout_hibernate.erl
new file mode 100644
index 0000000..cc91e26
--- /dev/null
+++ b/test/ws_SUITE_data/ws_timeout_hibernate.erl
@@ -0,0 +1,22 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_timeout_hibernate).
+-behaviour(cowboy_websocket_handler).
+-export([init/3]).
+-export([websocket_init/3, websocket_handle/3,
+	websocket_info/3, websocket_terminate/3]).
+
+init(_Any, _Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket}.
+
+websocket_init(_TransportName, Req, _Opts) ->
+	{ok, Req, undefined, 1000, hibernate}.
+
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State, hibernate}.
+
+websocket_info(_Info, Req, State) ->
+	{ok, Req, State, hibernate}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.
diff --git a/test/ws_SUITE_data/ws_upgrade_with_opts.erl b/test/ws_SUITE_data/ws_upgrade_with_opts.erl
new file mode 100644
index 0000000..b4f82fa
--- /dev/null
+++ b/test/ws_SUITE_data/ws_upgrade_with_opts.erl
@@ -0,0 +1,28 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(ws_upgrade_with_opts).
+-behaviour(cowboy_websocket_handler).
+
+-export([init/3]).
+-export([websocket_init/3]).
+-export([websocket_handle/3]).
+-export([websocket_info/3]).
+-export([websocket_terminate/3]).
+
+init(_Any, Req, _Opts) ->
+	{upgrade, protocol, cowboy_websocket, Req, <<"success">>}.
+
+websocket_init(_TransportName, Req, Response) ->
+	Req2 = cowboy_req:compact(Req),
+	erlang:send_after(10, self(), send_response),
+	{ok, Req2, Response}.
+
+websocket_handle(_Frame, Req, State) ->
+	{ok, Req, State}.
+
+websocket_info(send_response, Req, State = Response)
+		when is_binary(Response) ->
+	{reply, {text, Response}, Req, State}.
+
+websocket_terminate(_Reason, _Req, _State) ->
+	ok.