From f5f480c5750b76b662430b62ef85ebceea5aadc6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Mon, 17 Aug 2015 16:52:46 +0200 Subject: Beginnings of modernizing the test suite Merging because I have to do other work but this is still a nice enough improvement. --- test/acceptor_SUITE.erl | 205 ++++++++++++++---------------------------------- test/ranch_ct_hook.erl | 23 ++++++ test/sendfile_SUITE.erl | 100 ++++++++++------------- test/shutdown_SUITE.erl | 36 ++------- 4 files changed, 129 insertions(+), 235 deletions(-) create mode 100644 test/ranch_ct_hook.erl (limited to 'test') diff --git a/test/acceptor_SUITE.erl b/test/acceptor_SUITE.erl index 6325055..507c2dd 100644 --- a/test/acceptor_SUITE.erl +++ b/test/acceptor_SUITE.erl @@ -13,44 +13,9 @@ %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. -module(acceptor_SUITE). +-compile(export_all). --include_lib("common_test/include/ct.hrl"). - -%% ct. --export([all/0]). --export([groups/0]). --export([init_per_suite/1]). --export([end_per_suite/1]). --export([init_per_group/2]). --export([end_per_group/2]). - -%% misc. --export([misc_bad_transport/1]). - -%% ssl. --export([ssl_accept_error/1]). --export([ssl_accept_socket/1]). --export([ssl_active_echo/1]). --export([ssl_echo/1]). - -%% tcp. --export([tcp_accept_socket/1]). --export([tcp_active_echo/1]). --export([tcp_echo/1]). --export([tcp_inherit_options/1]). --export([tcp_max_connections/1]). --export([tcp_max_connections_and_beyond/1]). --export([tcp_set_max_connections/1]). --export([tcp_infinity_max_connections/1]). --export([tcp_clean_set_max_connections/1]). --export([tcp_upgrade/1]). - -%% supervisor. --export([supervisor_clean_restart/1]). --export([supervisor_clean_child_restart/1]). --export([supervisor_conns_alive/1]). --export([supervisor_server_recover_state/1]). --export([supervisor_clean_conns_sup_restart/1]). +-import(ct_helper, [doc/1]). %% ct. @@ -84,76 +49,40 @@ groups() -> supervisor_clean_conns_sup_restart ]}]. -init_per_suite(Config) -> - ok = application:start(ranch), - Config. - -end_per_suite(_) -> - application:stop(ranch), - ok. - -init_per_group(ssl, Config) -> - application:start(crypto), - application:start(asn1), - application:start(public_key), - application:start(ssl), - Config; -init_per_group(_, Config) -> - Config. - -end_per_group(ssl, _) -> - application:stop(ssl), - application:stop(public_key), - application:stop(asn1), - application:stop(crypto), - ok; -end_per_group(_, _) -> - ok. - %% misc. misc_bad_transport(_) -> + doc("Reject invalid transport modules."), {error, badarg} = ranch:start_listener(misc_bad_transport, 1, - bad_transport, [{port, 0}], - echo_protocol, []), + bad_transport, [], echo_protocol, []), ok. %% ssl. ssl_accept_error(_) -> + doc("Acceptor must not crash if client disconnects in the middle of SSL handshake."), Name = ssl_accept_error, - {_, Cert, Key} = ct_helper:make_certs(), - {ok, ListenerSup} = ranch:start_listener(Name, 1, - ranch_ssl, [{port, 0}, {cert, Cert}, {key, Key}], - echo_protocol, []), + Opts = ct_helper:get_certs_from_ets(), + {ok, ListenerSup} = ranch:start_listener(Name, 1, ranch_ssl, Opts, echo_protocol, []), Port = ranch:get_port(Name), ListenerSupChildren = supervisor:which_children(ListenerSup), - {_, AcceptorsSup, _, _} - = lists:keyfind(ranch_acceptors_sup, 1, ListenerSupChildren), - [{{acceptor, _, _}, AcceptorPid, _, _}] - = supervisor:which_children(AcceptorsSup), + {_, AcceptorsSup, _, _} = lists:keyfind(ranch_acceptors_sup, 1, ListenerSupChildren), + [{{acceptor, _, _}, AcceptorPid, _, _}] = supervisor:which_children(AcceptorsSup), true = is_process_alive(AcceptorPid), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, false}, {packet, raw}]), ok = gen_tcp:close(Socket), receive after 500 -> ok end, true = is_process_alive(AcceptorPid), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). ssl_accept_socket(_) -> - %%% XXX we can't do the spawn to test the controlling process change - %%% because of the bug in ssl + doc("Ensure that listener can use an externally opened SSL listen socket."), Name = ssl_accept_socket, - {_, Cert, Key} = ct_helper:make_certs(), - {ok, S} = ssl:listen(0, - [{cert, Cert}, {key, Key}, binary, - {active, false}, {packet, raw}, {reuseaddr, true}]), - {ok, _} = ranch:start_listener(Name, 1, - ranch_ssl, [{socket, S}], echo_protocol, []), + Opts = ct_helper:get_certs_from_ets(), + {ok, S} = ssl:listen(0, [binary, {active, false}, {packet, raw}, {reuseaddr, true}|Opts]), + {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, [{socket, S}], echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = ssl:connect("localhost", Port, - [binary, {active, false}, {packet, raw}, - {cert, Cert}, {key, Key}]), + {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]), ok = ssl:send(Socket, <<"TCP Ranch is working!">>), {ok, <<"TCP Ranch is working!">>} = ssl:recv(Socket, 21, 1000), ok = ranch:stop_listener(Name), @@ -163,15 +92,12 @@ ssl_accept_socket(_) -> ok. ssl_active_echo(_) -> + doc("Ensure that active mode works with SSL transport."), Name = ssl_active_echo, - {_, Cert, Key} = ct_helper:make_certs(), - {ok, _} = ranch:start_listener(Name, 1, - ranch_ssl, [{port, 0}, {cert, Cert}, {key, Key}], - active_echo_protocol, []), + Opts = ct_helper:get_certs_from_ets(), + {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, Opts, active_echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = ssl:connect("localhost", Port, - [binary, {active, false}, {packet, raw}, - {cert, Cert}, {key, Key}]), + {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]), ok = ssl:send(Socket, <<"SSL Ranch is working!">>), {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000), ok = ranch:stop_listener(Name), @@ -181,15 +107,12 @@ ssl_active_echo(_) -> ok. ssl_echo(_) -> + doc("Ensure that passive mode works with SSL transport."), Name = ssl_echo, - {_, Cert, Key} = ct_helper:make_certs(), - {ok, _} = ranch:start_listener(Name, 1, - ranch_ssl, [{port, 0}, {cert, Cert}, {key, Key}], - echo_protocol, []), + Opts = ct_helper:get_certs_from_ets(), + {ok, _} = ranch:start_listener(Name, 1, ranch_ssl, Opts, echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = ssl:connect("localhost", Port, - [binary, {active, false}, {packet, raw}, - {cert, Cert}, {key, Key}]), + {ok, Socket} = ssl:connect("localhost", Port, [binary, {active, false}, {packet, raw}]), ok = ssl:send(Socket, <<"SSL Ranch is working!">>), {ok, <<"SSL Ranch is working!">>} = ssl:recv(Socket, 21, 1000), ok = ranch:stop_listener(Name), @@ -201,22 +124,12 @@ ssl_echo(_) -> %% tcp. tcp_accept_socket(_) -> + doc("Ensure that listener can use an externally opened TCP listen socket."), Name = tcp_accept_socket, - Ref = make_ref(), - Parent = self(), - spawn(fun() -> - {ok, S} = gen_tcp:listen(0, [binary, {active, false}, {packet, raw}, - {reuseaddr, true}]), - {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{socket, S}], echo_protocol, []), - Parent ! Ref - end), - receive - Ref -> ok - end, + {ok, S} = gen_tcp:listen(0, [binary, {active, false}, {packet, raw}, {reuseaddr, true}]), + {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [{socket, S}], echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + {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(Name), @@ -226,12 +139,11 @@ tcp_accept_socket(_) -> ok. tcp_active_echo(_) -> + doc("Ensure that active mode works with TCP transport."), Name = tcp_active_echo, - {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}], active_echo_protocol, []), + {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], active_echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + {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(Name), @@ -241,12 +153,11 @@ tcp_active_echo(_) -> ok. tcp_echo(_) -> + doc("Ensure that passive mode works with TCP transport."), Name = tcp_echo, - {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}], echo_protocol, []), + {ok, _} = ranch:start_listener(Name, 1, ranch_tcp, [], echo_protocol, []), Port = ranch:get_port(Name), - {ok, Socket} = gen_tcp:connect("localhost", Port, - [binary, {active, false}, {packet, raw}]), + {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(Name), @@ -258,19 +169,19 @@ tcp_echo(_) -> tcp_max_connections(_) -> Name = tcp_max_connections, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}, {max_connections, 10}], + ranch_tcp, [{max_connections, 10}], notify_and_wait_protocol, [{msg, connected}, {pid, self()}]), Port = ranch:get_port(Name), ok = connect_loop(Port, 11, 150), 10 = ranch_server:count_connections(Name), 10 = receive_loop(connected, 400), 1 = receive_loop(connected, 1000), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_max_connections_and_beyond(_) -> Name = tcp_max_connections_and_beyond, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}, {max_connections, 10}], + ranch_tcp, [{max_connections, 10}], remove_conn_and_wait_protocol, [{remove, true}]), Port = ranch:get_port(Name), ok = connect_loop(Port, 10, 0), @@ -295,12 +206,12 @@ tcp_max_connections_and_beyond(_) -> ranch_server:get_connections_sup(Name)), {_, 20} = lists:keyfind(active, 1, Counts2), {_, 20} = lists:keyfind(workers, 1, Counts2), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_set_max_connections(_) -> Name = tcp_set_max_connections, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}, {max_connections, 10}], + ranch_tcp, [{max_connections, 10}], notify_and_wait_protocol, [{msg, connected}, {pid, self()}]), Port = ranch:get_port(Name), ok = connect_loop(Port, 20, 0), @@ -310,12 +221,12 @@ tcp_set_max_connections(_) -> ranch:set_max_connections(Name, 20), 10 = receive_loop(connected, 1000), 20 = ranch:get_max_connections(Name), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_infinity_max_connections(_) -> Name = tcp_infinity_max_connections, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}, {max_connections, 10}], + ranch_tcp, [{max_connections, 10}], notify_and_wait_protocol, [{msg, connected}, {pid, self()}]), Port = ranch:get_port(Name), ok = connect_loop(Port, 20, 0), @@ -330,14 +241,14 @@ tcp_infinity_max_connections(_) -> ranch:set_max_connections(Name, 10), 20 = ranch_server:count_connections(Name), 10 = receive_loop(connected, 1000), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_clean_set_max_connections(_) -> %% This is a regression test to check that setting max connections does not %% cause any processes to crash. Name = tcp_clean_set_max_connections, {ok, ListSupPid} = ranch:start_listener(Name, 4, ranch_tcp, - [{port, 0}, {max_connections, 4}], + [{max_connections, 4}], notify_and_wait_protocol, [{msg, connected}, {pid, self()}]), Children = supervisor:which_children(ListSupPid), {_, AccSupPid, _, _} = lists:keyfind(ranch_acceptors_sup, 1, Children), @@ -357,12 +268,12 @@ tcp_clean_set_max_connections(_) -> end, _ = erlang:trace(all, false, [all]), ok = clean_traces(), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_upgrade(_) -> Name = tcp_upgrade, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}], + ranch_tcp, [], notify_and_wait_protocol, [{msg, connected}, {pid, self()}]), Port = ranch:get_port(Name), ok = connect_loop(Port, 1, 0), @@ -370,20 +281,20 @@ tcp_upgrade(_) -> ranch:set_protocol_options(Name, [{msg, upgraded}, {pid, self()}]), ok = connect_loop(Port, 1, 0), receive upgraded -> ok after 1000 -> error(timeout) end, - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). tcp_inherit_options(_) -> Name = tcp_inherit_options, TcpOptions = [{nodelay, false}, {send_timeout_close, false}], {ok, _} = ranch:start_listener(Name, 4, ranch_tcp, - [{port, 0} | TcpOptions], + TcpOptions, check_tcp_options, [{pid, self()} | TcpOptions]), Port = ranch:get_port(Name), {ok, Socket} = gen_tcp:connect("localhost", Port, [binary, {active, true}, {packet, raw}]), receive checked -> ok after 1000 -> error(timeout) end, ok = gen_tcp:close(Socket), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). %% Supervisor tests @@ -394,7 +305,7 @@ supervisor_clean_restart(_) -> Name = supervisor_clean_restart, NbAcc = 4, {ok, Pid} = ranch:start_listener(Name, - NbAcc, ranch_tcp, [{port, 0}], echo_protocol, []), + NbAcc, ranch_tcp, [], echo_protocol, []), %% Trace supervisor spawns. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]), ConnsSup0 = ranch_server:get_connections_sup(Name), @@ -419,7 +330,7 @@ supervisor_clean_restart(_) -> ConnsSup = ranch_server:get_connections_sup(Name), _ = erlang:trace(all, false, [all]), ok = clean_traces(), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). supervisor_clean_child_restart(_) -> %% Then we verify that only parts of the supervision tree @@ -430,7 +341,7 @@ supervisor_clean_child_restart(_) -> 1 = erlang:trace_pattern({ranch_tcp, listen, 1}, [{'_', [], [{return_trace}]}], [global]), {ok, Pid} = ranch:start_listener(Name, - 1, ranch_tcp, [{port, 0}], echo_protocol, []), + 1, ranch_tcp, [], echo_protocol, []), %% Trace supervisor spawns. 1 = erlang:trace(Pid, true, [procs, set_on_spawn]), ConnsSup = ranch_server:get_connections_sup(Name), @@ -458,7 +369,7 @@ supervisor_clean_child_restart(_) -> _ = erlang:trace_pattern({ranch_tcp, listen, 1}, false, []), _ = erlang:trace(all, false, [all]), ok = clean_traces(), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). supervisor_conns_alive(_) -> %% And finally we make sure that in the case of partial failure @@ -468,7 +379,7 @@ supervisor_conns_alive(_) -> 1 = erlang:trace_pattern({ranch_tcp, listen, 1}, [{'_', [], [{return_trace}]}], [global]), {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}], + ranch_tcp, [], remove_conn_and_wait_protocol, [{remove, false}]), %% Get the listener socket LSocket = receive @@ -488,14 +399,14 @@ supervisor_conns_alive(_) -> receive {tcp_closed, _} -> ok end, _ = erlang:trace(all, false, [all]), ok = clean_traces(), - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). supervisor_server_recover_state(_) -> %% Verify that if ranch_server crashes it regains its state and monitors %% ranch_conns_sup that were previously registered. Name = supervisor_server_recover_state, {ok, _} = ranch:start_listener(Name, 1, - ranch_tcp, [{port, 0}], echo_protocol, []), + ranch_tcp, [], echo_protocol, []), _ = erlang:trace(new, true, [call]), 1 = erlang:trace_pattern({ranch_server, init, 1}, [{'_', [], [{return_trace}]}], [global]), @@ -513,7 +424,7 @@ supervisor_server_recover_state(_) -> error(timeout) end, ConnsSup = ranch_server:get_connections_sup(Name), - ranch:stop_listener(Name), + ok = ranch:stop_listener(Name), %% Check ranch_server has removed the ranch_conns_sup. {'EXIT', {badarg, _}} = (catch ranch_server:get_connections_sup(Name)), _ = erlang:trace(all, false, [all]), @@ -525,7 +436,7 @@ supervisor_clean_conns_sup_restart(_) -> %% the ranch_server. Name = supervisor_clean_conns_sup_restart, {ok, _} = ranch:start_listener(Name, - 1, ranch_tcp, [{port, 0}], echo_protocol, []), + 1, ranch_tcp, [], echo_protocol, []), Server = erlang:whereis(ranch_server), ServerMonRef = erlang:monitor(process, Server), %% Exit because Name already registered and is alive. @@ -537,7 +448,7 @@ supervisor_clean_conns_sup_restart(_) -> 1000 -> ok end, - ranch:stop_listener(Name). + ok = ranch:stop_listener(Name). %% Utility functions. diff --git a/test/ranch_ct_hook.erl b/test/ranch_ct_hook.erl new file mode 100644 index 0000000..051e5b2 --- /dev/null +++ b/test/ranch_ct_hook.erl @@ -0,0 +1,23 @@ +%% Copyright (c) 2015, Loïc Hoguin +%% +%% 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(ranch_ct_hook). + +-export([init/2]). + +init(_, _) -> + ct_helper:start([ranch, ssl]), + ct_helper:make_certs_in_ets(), + error_logger:add_report_handler(ct_helper_error_h), + {ok, undefined}. diff --git a/test/sendfile_SUITE.erl b/test/sendfile_SUITE.erl index c74659e..b067ded 100644 --- a/test/sendfile_SUITE.erl +++ b/test/sendfile_SUITE.erl @@ -13,29 +13,10 @@ %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. -module(sendfile_SUITE). +-compile(export_all). --include_lib("common_test/include/ct.hrl"). - -%% ct. --export([all/0]). --export([suite/0]). --export([groups/0]). --export([init_per_suite/1]). --export([end_per_suite/1]). --export([init_per_group/2]). --export([end_per_group/2]). - -%% Tests. --export([filename/1]). --export([rawfile/1]). --export([rawfile_bytes_large/1]). --export([rawfile_bytes_zero/1]). --export([rawfile_chunk_size_large/1]). --export([rawfile_offset_large/1]). --export([rawfile_range_large/1]). --export([rawfile_range_medium/1]). --export([rawfile_range_small/1]). --export([ssl_chunk_size/1]). +-import(ct_helper, [config/2]). +-import(ct_helper, [doc/1]). all() -> [{group, tcp}, {group, ssl}]. @@ -58,42 +39,30 @@ groups() -> [{tcp, [parallel], Tests}, {ssl, [parallel], Tests ++ [ssl_chunk_size]}]. init_per_suite(Config) -> - ok = application:start(ranch), - ok = application:start(crypto), - Filename = filename:join(?config(priv_dir, Config), "sendfile"), + Filename = filename:join(config(priv_dir, Config), "sendfile"), Binary = crypto:rand_bytes(20 * 1024 * 1024), ok = file:write_file(Filename, Binary), [{filename, Filename} | Config]. end_per_suite(Config) -> - application:stop(ranch), - application:stop(crypto), - Filename = ?config(filename, Config), + Filename = config(filename, Config), ok = file:delete(Filename), ok. init_per_group(ssl, Config) -> - application:start(asn1), - application:start(public_key), - application:start(ssl), - {_, Cert, Key} = ct_helper:make_certs(), - SslOpts = [{cert, Cert}, {key, Key}], + SslOpts = ct_helper:get_certs_from_ets(), [{transport, ranch_ssl}, {transport_opts, SslOpts} | Config]; init_per_group(tcp, Config) -> [{transport, ranch_tcp}, {transport_opts, []} | Config]. -end_per_group(ssl, _) -> - application:stop(ssl), - application:stop(public_key), - application:stop(asn1), - ok; end_per_group(_, _) -> ok. %% Check can send a whole file given with filename. + filename(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -105,9 +74,10 @@ filename(Config) -> ok = Transport:close(Server). %% Check can send a whole file with rawfile. + rawfile(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -122,9 +92,10 @@ rawfile(Config) -> ok = Transport:close(Server). %% Check can send a file where Bytes is larger than file size. + rawfile_bytes_large(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -140,9 +111,10 @@ rawfile_bytes_large(Config) -> ok = Transport:close(Server). %% Check can send whole file when Bytes =:= 0. + rawfile_bytes_zero(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -157,9 +129,10 @@ rawfile_bytes_zero(Config) -> ok = Transport:close(Server). %% Check can send file where chunk_size is greater than file size. + rawfile_chunk_size_large(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -176,9 +149,10 @@ rawfile_chunk_size_large(Config) -> %% Check send file where offset is larger than file size sends no bytes and %% returns {ok, 0}. + rawfile_offset_large(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -191,9 +165,10 @@ rawfile_offset_large(Config) -> %% Check can send file with positive Offset and Offset + Bytes larger than file %% size. + rawfile_range_large(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -215,9 +190,10 @@ rawfile_range_large(Config) -> %% Check can send file with positive Offset and Offset + Bytes less than file %% size. + rawfile_range_medium(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), {ok, {Server, Client}} = sockets(Config), @@ -238,9 +214,10 @@ rawfile_range_medium(Config) -> %% Check can send file with positive Offset, Offset + Bytes less than file %% size and Bytes less than chunk_size. + rawfile_range_small(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), {ok, {Server, Client}} = sockets(Config), {ok, RawFile} = file:open(Filename, [read, raw, binary]), @@ -260,9 +237,10 @@ rawfile_range_small(Config) -> ok = Transport:close(Server). %% Check ssl obeys chunk_size. + ssl_chunk_size(Config) -> - Transport = ?config(transport, Config), - Filename = ?config(filename, Config), + Transport = config(transport, Config), + Filename = config(filename, Config), {ok, Binary} = file:read_file(Filename), Size = byte_size(Binary), Self = self(), @@ -292,9 +270,11 @@ ssl_chunk_size(Config) -> _ = erlang:trace(all, false, [all]), ok = clean_traces(). +%% Internal. + sockets(Config) -> - Transport = ?config(transport, Config), - TransportOpts = ?config(transport_opts, Config), + Transport = config(transport, Config), + TransportOpts = config(transport_opts, Config), {ok, LSocket} = Transport:listen(TransportOpts), {ok, {_, Port}} = Transport:sockname(LSocket), Self = self(), diff --git a/test/shutdown_SUITE.erl b/test/shutdown_SUITE.erl index 109c381..97963d5 100644 --- a/test/shutdown_SUITE.erl +++ b/test/shutdown_SUITE.erl @@ -13,34 +13,14 @@ %% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. -module(shutdown_SUITE). +-compile(export_all). --include_lib("common_test/include/ct.hrl"). - -%% ct. --export([all/0]). --export([init_per_suite/1]). --export([end_per_suite/1]). - -%% Tests. - --export([brutal_kill/1]). --export([infinity/1]). --export([infinity_trap_exit/1]). --export([timeout/1]). --export([timeout_trap_exit/1]). +-import(ct_helper, [doc/1]). %% ct. all() -> - [brutal_kill, infinity, infinity_trap_exit, timeout, timeout_trap_exit]. - -init_per_suite(Config) -> - ok = application:start(ranch), - Config. - -end_per_suite(_) -> - application:stop(ranch), - ok. + ct_helper:all(?MODULE). %% Tests. @@ -57,7 +37,7 @@ brutal_kill(_) -> = lists:keyfind(ranch_conns_sup, 1, ListenerSupChildren), [{_, Pid, _, _}] = supervisor:which_children(ConnsSup), true = is_process_alive(Pid), - ranch:stop_listener(Name), + ok = ranch:stop_listener(Name), receive after 100 -> ok end, false = is_process_alive(Pid), false = is_process_alive(ListenerSup), @@ -77,7 +57,7 @@ infinity(_) -> = lists:keyfind(ranch_conns_sup, 1, ListenerSupChildren), [{_, Pid, _, _}] = supervisor:which_children(ConnsSup), true = is_process_alive(Pid), - ranch:stop_listener(Name), + ok = ranch:stop_listener(Name), receive after 100 -> ok end, false = is_process_alive(Pid), false = is_process_alive(ListenerSup), @@ -98,7 +78,7 @@ infinity_trap_exit(_) -> [{_, Pid, _, _}] = supervisor:which_children(ConnsSup), true = is_process_alive(Pid), %% This call will block infinitely. - SpawnPid = spawn(fun() -> ranch:stop_listener(Name) end), + SpawnPid = spawn(fun() -> ok = ranch:stop_listener(Name) end), receive after 100 -> ok end, %% The protocol traps exit signals, and ignore them, so it won't die. true = is_process_alive(Pid), @@ -127,7 +107,7 @@ timeout(_) -> = lists:keyfind(ranch_conns_sup, 1, ListenerSupChildren), [{_, Pid, _, _}] = supervisor:which_children(ConnsSup), true = is_process_alive(Pid), - ranch:stop_listener(Name), + ok = ranch:stop_listener(Name), receive after 100 -> ok end, false = is_process_alive(Pid), false = is_process_alive(ListenerSup), @@ -148,7 +128,7 @@ timeout_trap_exit(_) -> [{_, Pid, _, _}] = supervisor:which_children(ConnsSup), true = is_process_alive(Pid), %% This call will block for the duration of the shutdown. - SpawnPid = spawn(fun() -> ranch:stop_listener(Name) end), + SpawnPid = spawn(fun() -> ok = ranch:stop_listener(Name) end), receive after 100 -> ok end, %% The protocol traps exit signals, and ignore them, so it won't die. true = is_process_alive(Pid), -- cgit v1.2.3