umgeher/cowboy
0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-14 20:30:23 +00:00
Code Issues Wiki Activity

Add tests for the streams shutdown mechanism

Browse source
This commit is contained in:
Loïc Hoguin 2017-08-14 17:17:44 +02:00
parent 58b70a594b
commit a2facaf2da
No known key found for this signature in database
GPG key ID: 71366FF21851DF03
2 changed files with 145 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

51
test/handlers/stream_handler_h.erl
View file

@ -9,21 +9,56 @@
-export([terminate/3]).
-export([early_error/5]).
init(StreamID, Req, Opts) ->
%% @todo Vary behavior depending on x-test-case.
Pid = list_to_pid(binary_to_list(cowboy_req:header(<<"x-test-pid">>, Req))),
Pid ! {Pid, self(), init, StreamID, Req, Opts},
{[{headers, 200, #{}}], Pid}.
-record(state, {
pid,
test
}).
data(StreamID, IsFin, Data, State=Pid) ->
init(StreamID, Req, Opts) ->
Pid = list_to_pid(binary_to_list(cowboy_req:header(<<"x-test-pid">>, Req))),
Test = binary_to_atom(cowboy_req:header(<<"x-test-case">>, Req), latin1),
State = #state{pid=Pid, test=Test},
Pid ! {Pid, self(), init, StreamID, Req, Opts},
{init_commands(StreamID, Req, State), State}.
init_commands(_, _, State=#state{test=shutdown_on_stream_stop}) ->
Spawn = init_process(false, State),
[{headers, 200, #{}}, {spawn, Spawn, 5000}, stop];
init_commands(_, _, State=#state{test=shutdown_on_socket_close}) ->
Spawn = init_process(false, State),
[{headers, 200, #{}}, {spawn, Spawn, 5000}];
init_commands(_, _, State=#state{test=shutdown_timeout_on_stream_stop}) ->
Spawn = init_process(true, State),
[{headers, 200, #{}}, {spawn, Spawn, 2000}, stop];
init_commands(_, _, State=#state{test=shutdown_timeout_on_socket_close}) ->
Spawn = init_process(true, State),
[{headers, 200, #{}}, {spawn, Spawn, 2000}];
init_commands(_, _, _) ->
[{headers, 200, #{}}].
init_process(TrapExit, #state{pid=Pid}) ->
Self = self(),
Spawn = spawn_link(fun() ->
process_flag(trap_exit, TrapExit),
Pid ! {Pid, Self, spawned, self()},
receive {Pid, ready} -> ok after 1000 -> error(timeout) end,
Self ! {self(), ready},
receive after 5000 ->
Pid ! {Pid, Self, still_alive, self()}
end
end),
receive {Spawn, ready} -> ok after 1000 -> error(timeout) end,
Spawn.
data(StreamID, IsFin, Data, State=#state{pid=Pid}) ->
Pid ! {Pid, self(), data, StreamID, IsFin, Data, State},
{[], State}.
info(StreamID, Info, State=Pid) ->
info(StreamID, Info, State=#state{pid=Pid}) ->
Pid ! {Pid, self(), info, StreamID, Info, State},
{[], State}.
terminate(StreamID, Reason, State=Pid) ->
terminate(StreamID, Reason, State=#state{pid=Pid}) ->
Pid ! {Pid, self(), terminate, StreamID, Reason, State},
ok.