diff options
| author | Loïc Hoguin <[email protected]> | 2012-08-14 12:29:36 +0200 |
|---|---|---|
| committer | Loïc Hoguin <[email protected]> | 2012-08-15 14:30:14 +0200 |
| commit | d8981b586223af4bffb813e1e76c172828b0810a (patch) | |
| tree | dd4f99943f51af98756e26677ea74599d7122fc9 /test/active_echo_protocol.erl | |
| parent | bfa353f8e75e06363bea565cc50f8de52090ae95 (diff) | |
| download | ranch-d8981b586223af4bffb813e1e76c172828b0810a.tar.gz ranch-d8981b586223af4bffb813e1e76c172828b0810a.tar.bz2 ranch-d8981b586223af4bffb813e1e76c172828b0810a.zip | |
Add tests for {active, once} for both TCP and SSL
Diffstat (limited to 'test/active_echo_protocol.erl')
| -rw-r--r-- | test/active_echo_protocol.erl | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/test/active_echo_protocol.erl b/test/active_echo_protocol.erl new file mode 100644 index 0000000..ac53d12 --- /dev/null +++ b/test/active_echo_protocol.erl @@ -0,0 +1,25 @@ +-module(active_echo_protocol). + +-export([start_link/4]). +-export([init/4]). + +start_link(ListenerPid, Socket, Transport, Opts) -> + Pid = spawn_link(?MODULE, init, [ListenerPid, Socket, Transport, Opts]), + {ok, Pid}. + +init(ListenerPid, Socket, Transport, _Opts = []) -> + ok = ranch:accept_ack(ListenerPid), + loop(Socket, Transport). + +loop(Socket, Transport) -> + {OK, Closed, Error} = Transport:messages(), + Transport:setopts(Socket, [{active, once}]), + receive + {OK, Socket, Data} -> + Transport:send(Socket, Data), + loop(Socket, Transport); + {Closed, Socket} -> + ok; + {Error, Socket, _} -> + ok = Transport:close(Socket) + end. |