diff options
author | Loïc Hoguin <[email protected]> | 2012-04-12 23:22:52 +0200 |
---|---|---|
committer | Loïc Hoguin <[email protected]> | 2012-04-14 17:52:47 +0200 |
commit | 4156fa3ebb978a21b52abf94202bb3d7676f4c03 (patch) | |
tree | e05c4cac592afeace0f1722bd2111d0d88eb94d0 /test | |
parent | f971ec1ced3758834d6e98ab3c7fafddd8bf9da6 (diff) | |
download | ranch-4156fa3ebb978a21b52abf94202bb3d7676f4c03.tar.gz ranch-4156fa3ebb978a21b52abf94202bb3d7676f4c03.tar.bz2 ranch-4156fa3ebb978a21b52abf94202bb3d7676f4c03.zip |
Import the acceptor code from Cowboy
Modules were renamed. The 'cowboy_' prefix became 'ranch_'.
At the same time, ranch_ssl_transport became ranch_ssl,
and ranch_tcp_transport became ranch_tcp, because appending
'_transport' felt a bit redundant considering SSL and TCP
clearly are transports.
One test has been added to make sure everything is working.
Diffstat (limited to 'test')
-rw-r--r-- | test/acceptor_SUITE.erl | 52 | ||||
-rw-r--r-- | test/echo_protocol.erl | 19 |
2 files changed, 71 insertions, 0 deletions
diff --git a/test/acceptor_SUITE.erl b/test/acceptor_SUITE.erl new file mode 100644 index 0000000..cfb5cd8 --- /dev/null +++ b/test/acceptor_SUITE.erl @@ -0,0 +1,52 @@ +%% Copyright (c) 2011-2012, Loïc Hoguin <[email protected]> +%% +%% Permission to use, copy, modify, and/or distribute this software for any +%% purpose with or without fee is hereby granted, provided that the above +%% copyright notice and this permission notice appear in all copies. +%% +%% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES +%% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF +%% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR +%% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES +%% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN +%% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF +%% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. + +-module(acceptor_SUITE). + +-include_lib("common_test/include/ct.hrl"). + +-export([all/0, groups/0, init_per_suite/1, end_per_suite/1]). %% ct. +-export([tcp_echo/1]). %% tcp. + +%% ct. + +all() -> + [{group, tcp}]. + +groups() -> + Tests = [tcp_echo], + [{tcp, Tests}]. + +init_per_suite(Config) -> + application:start(ranch), + Config. + +end_per_suite(_) -> + application:stop(ranch), + ok. + +%% tcp. + +tcp_echo(_) -> + %% @todo Don't use a fixed port. start_listener should return the port used? + {ok, _} = ranch:start_listener(tcp_echo, 1, + ranch_tcp, [{port, 33333}], + echo_protocol, []), + {ok, Socket} = gen_tcp:connect("localhost", 33333, + [binary, {active, false}, {packet, raw}]), + ok = gen_tcp:send(Socket, <<"Ranch is working!">>), + {ok, <<"Ranch is working!">>} = gen_tcp:recv(Socket, 0, 1000), + ok = ranch:stop_listener(tcp_echo), + {error, closed} = gen_tcp:recv(Socket, 0, 1000), + ok. diff --git a/test/echo_protocol.erl b/test/echo_protocol.erl new file mode 100644 index 0000000..950cb06 --- /dev/null +++ b/test/echo_protocol.erl @@ -0,0 +1,19 @@ +-module(echo_protocol). +-export([start_link/4, 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) -> + case Transport:recv(Socket, 0, 5000) of + {ok, Data} -> + Transport:send(Socket, Data), + loop(Socket, Transport); + _ -> + ok = Transport:close(Socket) + end. |