aboutsummaryrefslogtreecommitdiffstats
path: root/test/stream_handler_SUITE.erl
diff options
context:
space:
mode:
Diffstat (limited to 'test/stream_handler_SUITE.erl')
-rw-r--r--test/stream_handler_SUITE.erl12
1 files changed, 10 insertions, 2 deletions
diff --git a/test/stream_handler_SUITE.erl b/test/stream_handler_SUITE.erl
index 9003e23..738a4a2 100644
--- a/test/stream_handler_SUITE.erl
+++ b/test/stream_handler_SUITE.erl
@@ -358,12 +358,20 @@ do_switch_protocol_after_response(TestCase, Config) ->
]),
%% Confirm init/3 is called and receive the response.
Pid = receive {Self, P, init, _, _, _} -> P after 1000 -> error(timeout) end,
- {response, nofin, 200, _} = gun:await(ConnPid, Ref),
+ {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
+ Gzipped =
+ lists:keyfind(<<"content-encoding">>, 1, Headers)
+ =:= {<<"content-encoding">>, <<"gzip">>},
case TestCase of
<<"switch_protocol_after_headers">> ->
ok;
_ ->
- {ok, <<"{}">>} = gun:await_body(ConnPid, Ref),
+ <<"{}">> = case gun:await_body(ConnPid, Ref) of
+ {ok, Body} when Gzipped ->
+ zlib:gunzip(Body);
+ {ok, Body} ->
+ Body
+ end,
ok
end,
{error, _} = gun:await(ConnPid, Ref),