diff options
author | Loïc Hoguin <[email protected]> | 2013-04-22 15:55:22 +0200 |
---|---|---|
committer | Loïc Hoguin <[email protected]> | 2013-04-22 17:03:54 +0200 |
commit | 2e916777233d8d82b93cbca9b754a023a6ca63c2 (patch) | |
tree | bc3de082804c1c9f220cdbba542e545082b2e793 /test/ws_SUITE_data | |
parent | cf0e00589409eb8caf423e3efa6777540951d424 (diff) | |
download | cowboy-2e916777233d8d82b93cbca9b754a023a6ca63c2.tar.gz cowboy-2e916777233d8d82b93cbca9b754a023a6ca63c2.tar.bz2 cowboy-2e916777233d8d82b93cbca9b754a023a6ca63c2.zip |
Reorganize the websocket test suite
We now have the suite specific modules in the data folder.
Compilation is performed by the Makefile instead of ct_run.
Diffstat (limited to 'test/ws_SUITE_data')
-rw-r--r-- | test/ws_SUITE_data/ws_echo.erl | 27 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_echo_timer.erl | 31 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_init_shutdown.erl | 23 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_send_many.erl | 27 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_timeout_cancel.erl | 26 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_timeout_hibernate.erl | 22 | ||||
-rw-r--r-- | test/ws_SUITE_data/ws_upgrade_with_opts.erl | 28 |
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. |