aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/acceptor_SUITE.erl35
-rw-r--r--test/active_echo_protocol.erl25
-rw-r--r--test/echo_protocol.erl4
-rw-r--r--test/notify_and_wait_protocol.erl1
-rw-r--r--test/remove_conn_and_wait_protocol.erl1
5 files changed, 65 insertions, 1 deletions
diff --git a/test/acceptor_SUITE.erl b/test/acceptor_SUITE.erl
index 1c0931b..8966d57 100644
--- a/test/acceptor_SUITE.erl
+++ b/test/acceptor_SUITE.erl
@@ -26,9 +26,11 @@
%% ssl.
-export([ssl_accept_error/1]).
+-export([ssl_active_echo/1]).
-export([ssl_echo/1]).
%% tcp.
+-export([tcp_active_echo/1]).
-export([tcp_echo/1]).
-export([tcp_max_connections/1]).
-export([tcp_max_connections_and_beyond/1]).
@@ -41,12 +43,14 @@ all() ->
groups() ->
[{tcp, [
+ tcp_active_echo,
tcp_echo,
tcp_max_connections,
tcp_max_connections_and_beyond,
tcp_upgrade
]}, {ssl, [
ssl_accept_error,
+ ssl_active_echo,
ssl_echo
]}].
@@ -91,6 +95,23 @@ ssl_accept_error(Config) ->
receive after 500 -> ok end,
true = is_process_alive(AcceptorPid).
+ssl_active_echo(Config) ->
+ {ok, _} = ranch:start_listener(ssl_active_echo, 1,
+ ranch_ssl, [{port, 0},
+ {certfile, ?config(data_dir, Config) ++ "cert.pem"}],
+ active_echo_protocol, []),
+ Port = ranch:get_port(ssl_active_echo),
+ {ok, Socket} = ssl:connect("localhost", Port,
+ [binary, {active, false}, {packet, raw},
+ {certfile, ?config(data_dir, Config) ++ "cert.pem"}]),
+ ok = ssl:send(Socket, <<"SSL Ranch is working!">>),
+ {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000),
+ ok = ranch:stop_listener(ssl_active_echo),
+ {error, closed} = ssl:recv(Socket, 0, 1000),
+ %% Make sure the listener stopped.
+ {'EXIT', _} = begin catch ranch:get_port(ssl_active_echo) end,
+ ok.
+
ssl_echo(Config) ->
{ok, _} = ranch:start_listener(ssl_echo, 1,
ranch_ssl, [{port, 0},
@@ -110,6 +131,20 @@ ssl_echo(Config) ->
%% tcp.
+tcp_active_echo(_) ->
+ {ok, _} = ranch:start_listener(tcp_active_echo, 1,
+ ranch_tcp, [{port, 0}], active_echo_protocol, []),
+ Port = ranch:get_port(tcp_active_echo),
+ {ok, Socket} = gen_tcp:connect("localhost", Port,
+ [binary, {active, false}, {packet, raw}]),
+ ok = gen_tcp:send(Socket, <<"TCP Ranch is working!">>),
+ {ok, <<"TCP Ranch is working!">>} = gen_tcp:recv(Socket, 21, 1000),
+ ok = ranch:stop_listener(tcp_active_echo),
+ {error, closed} = gen_tcp:recv(Socket, 0, 1000),
+ %% Make sure the listener stopped.
+ {'EXIT', _} = begin catch ranch:get_port(tcp_active_echo) end,
+ ok.
+
tcp_echo(_) ->
{ok, _} = ranch:start_listener(tcp_echo, 1,
ranch_tcp, [{port, 0}], echo_protocol, []),
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.
diff --git a/test/echo_protocol.erl b/test/echo_protocol.erl
index 950cb06..6839239 100644
--- a/test/echo_protocol.erl
+++ b/test/echo_protocol.erl
@@ -1,5 +1,7 @@
-module(echo_protocol).
--export([start_link/4, init/4]).
+
+-export([start_link/4]).
+-export([init/4]).
start_link(ListenerPid, Socket, Transport, Opts) ->
Pid = spawn_link(?MODULE, init, [ListenerPid, Socket, Transport, Opts]),
diff --git a/test/notify_and_wait_protocol.erl b/test/notify_and_wait_protocol.erl
index 27542f2..c959732 100644
--- a/test/notify_and_wait_protocol.erl
+++ b/test/notify_and_wait_protocol.erl
@@ -1,4 +1,5 @@
-module(notify_and_wait_protocol).
+
-export([start_link/4]).
-export([init/2]).
diff --git a/test/remove_conn_and_wait_protocol.erl b/test/remove_conn_and_wait_protocol.erl
index 8fffca7..011bd5d 100644
--- a/test/remove_conn_and_wait_protocol.erl
+++ b/test/remove_conn_and_wait_protocol.erl
@@ -1,4 +1,5 @@
-module(remove_conn_and_wait_protocol).
+
-export([start_link/4]).
-export([init/2]).