Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Browse files

Add tests for {active, once} for both TCP and SSL

  • Loading branch information...
commit d8981b586223af4bffb813e1e76c172828b0810a 1 parent bfa353f
@essen essen authored
View
35 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, []),
View
25 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.
View
4 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]),
View
1  test/notify_and_wait_protocol.erl
@@ -1,4 +1,5 @@
-module(notify_and_wait_protocol).
+
-export([start_link/4]).
-export([init/2]).
View
1  test/remove_conn_and_wait_protocol.erl
@@ -1,4 +1,5 @@
-module(remove_conn_and_wait_protocol).
+
-export([start_link/4]).
-export([init/2]).

0 comments on commit d8981b5

Please sign in to comment.
Something went wrong with that request. Please try again.