%% Copyright (c) 2018-2019, 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(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, ">] <- 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.