diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/stream_handler_SUITE.erl | 12 |
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), |