From bc43fca9e61b770ed9d6aa003fd56272ce4d2d59 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= Date: Sun, 23 Sep 2018 14:10:04 +0200 Subject: Fix stream_handler_SUITE test failures due to compression We always compress when streaming the body back to the client regardless of the size actually streamed. --- test/stream_handler_SUITE.erl | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'test/stream_handler_SUITE.erl') 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), -- cgit v1.2.3