140 lines
3.9 KiB
Erlang
140 lines
3.9 KiB
Erlang
%% Copyright (c) 2018-2023, Loïc Hoguin <essen@ninenines.eu>
|
|
%%
|
|
%% 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(h2specd_SUITE).
|
|
-compile(export_all).
|
|
-compile(nowarn_export_all).
|
|
|
|
-import(ct_helper, [config/2]).
|
|
-import(ct_helper, [doc/1]).
|
|
|
|
%% ct.
|
|
|
|
all() ->
|
|
[h2specd].
|
|
|
|
init_per_suite(Config) ->
|
|
case os:getenv("H2SPECD") of
|
|
false -> {skip, "$H2SPECD isn't set."};
|
|
H2specd ->
|
|
case filelib:is_file(H2specd) of
|
|
false -> {skip, "$H2SPECD file not found."};
|
|
true ->
|
|
%% We ensure that SASL is started for this test suite
|
|
%% to have the crash reports in the CT logs.
|
|
{ok, Apps} = application:ensure_all_started(sasl),
|
|
[{sasl_started, Apps =/= []}|Config]
|
|
end
|
|
end.
|
|
|
|
end_per_suite(Config) ->
|
|
case config(sasl_started, Config) of
|
|
true -> application:stop(sasl);
|
|
false -> ok
|
|
end.
|
|
|
|
%% Tests.
|
|
|
|
h2specd(Config) ->
|
|
doc("h2specd test suite for the HTTP/2 protocol."),
|
|
Self = self(),
|
|
Pid = spawn_link(fun() -> start_port(Config, Self) end),
|
|
receive ready -> ok after 10000 -> error(timeout) end,
|
|
try
|
|
run_tests(),
|
|
timer:sleep(100),
|
|
maybe_fail()
|
|
after
|
|
unlink(Pid),
|
|
os:cmd("killall h2specd")
|
|
end.
|
|
|
|
start_port(Config, Pid) ->
|
|
H2specd = os:getenv("H2SPECD"),
|
|
Port = open_port(
|
|
{spawn, H2specd ++ " -S -p 45678"},
|
|
[{line, 10000}, {cd, config(priv_dir, Config)}, binary, exit_status]),
|
|
Pid ! ready,
|
|
receive_infinity(Port, []).
|
|
|
|
receive_infinity(Port, Acc) ->
|
|
receive
|
|
{Port, {data, {eol, Line}}} ->
|
|
ct:log("~ts", [Line]),
|
|
%% Somehow we may receive the same line multiple times.
|
|
%% We therefore only print if it's a line we didn't print before.
|
|
case lists:member(Line, Acc) of
|
|
false -> io:format(user, "~s~n", [Line]);
|
|
true -> ok
|
|
end,
|
|
receive_infinity(Port, [Line|Acc]);
|
|
{Port, Reason={exit_status, _}} ->
|
|
ct:log("~ts", [[[L, $\n] || L <- lists:reverse(Acc)]]),
|
|
exit({shutdown, Reason})
|
|
end.
|
|
|
|
run_tests() ->
|
|
timer:sleep(1000),
|
|
Tests = scrape_tests(),
|
|
ct:pal("Test ports: ~p~n", [Tests]),
|
|
run_tests(Tests),
|
|
timer:sleep(1000).
|
|
|
|
run_tests([]) ->
|
|
ok;
|
|
run_tests([Port|Tail]) ->
|
|
try
|
|
{ok, Conn} = gun:open("127.0.0.1", Port, #{
|
|
protocols => [http2],
|
|
retry => 0,
|
|
tcp_opts => [{nodelay, true}]
|
|
}),
|
|
MRef = monitor(process, Conn),
|
|
{ok, http2} = gun:await_up(Conn),
|
|
StreamRef = gun:get(Conn, "/"),
|
|
receive
|
|
{gun_response, Conn, StreamRef, _, _, _} ->
|
|
timer:sleep(100);
|
|
{'DOWN', MRef, process, Conn, _} ->
|
|
ok
|
|
after 100 ->
|
|
ok
|
|
end
|
|
after
|
|
run_tests(Tail)
|
|
end.
|
|
|
|
scrape_tests() ->
|
|
{ok, Conn} = gun:open("127.0.0.1", 45678),
|
|
{ok, http} = gun:await_up(Conn),
|
|
StreamRef = gun:get(Conn, "/"),
|
|
{response, nofin, 200, _} = gun:await(Conn, StreamRef),
|
|
{ok, Body} = gun:await_body(Conn, StreamRef),
|
|
ok = gun:close(Conn),
|
|
{match, Matches} = re:run(Body, "<a href=\"(.*?)\"", [global, {capture, all, binary}]),
|
|
[binary_to_integer(Port)
|
|
|| [_, <<"http://127.0.0.1:", Port:5/binary, "/">>] <- Matches].
|
|
|
|
maybe_fail() ->
|
|
{ok, Conn} = gun:open("127.0.0.1", 45678),
|
|
{ok, http} = gun:await_up(Conn),
|
|
StreamRef = gun:get(Conn, "/report", [{<<"accept">>, "text/plain"}]),
|
|
{response, nofin, 200, _} = gun:await(Conn, StreamRef),
|
|
{ok, Body} = gun:await_body(Conn, StreamRef),
|
|
ok = gun:close(Conn),
|
|
case binary:match(Body, <<"0 skipped, 0 failed">>) of
|
|
nomatch -> exit(failed);
|
|
_ -> ok
|
|
end.
|