1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 |
- -module(rest_handler_SUITE).
- -compile(export_all).
- -compile(nowarn_export_all).
- -import(ct_helper, [config/2]).
- -import(ct_helper, [doc/1]).
- -import(cowboy_test, [gun_open/1]).
- all() ->
- cowboy_test:common_all().
- groups() ->
- cowboy_test:common_groups(ct_helper:all(?MODULE)).
- init_per_group(Name, Config) ->
- cowboy_test:init_common_groups(Name, Config, ?MODULE).
- end_per_group(Name, _) ->
- cowboy:stop_listener(Name).
- init_dispatch(_) ->
- cowboy_router:compile([{'_', [
- {"/switch_handler", switch_handler_h, run},
- {"/switch_handler_opts", switch_handler_h, hibernate}
- ]}]).
- do_decode(Headers, Body) ->
- case lists:keyfind(<<"content-encoding">>, 1, Headers) of
- {_, <<"gzip">>} -> zlib:gunzip(Body);
- _ -> Body
- end.
- switch_handler(Config) ->
- doc("Switch REST to loop handler for streaming the response body."),
- ConnPid = gun_open(Config),
- Ref = gun:get(ConnPid, "/switch_handler", [{<<"accept-encoding">>, <<"gzip">>}]),
- {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
- {ok, Body} = gun:await_body(ConnPid, Ref),
- <<"Hello\nstreamed\nworld!\n">> = do_decode(Headers, Body),
- ok.
- switch_handler_opts(Config) ->
- doc("Switch REST to loop handler for streaming the response body; with options."),
- ConnPid = gun_open(Config),
- Ref = gun:get(ConnPid, "/switch_handler_opts", [{<<"accept-encoding">>, <<"gzip">>}]),
- {response, nofin, 200, Headers} = gun:await(ConnPid, Ref),
- {ok, Body} = gun:await_body(ConnPid, Ref),
- <<"Hello\nstreamed\nworld!\n">> = do_decode(Headers, Body),
- ok.
|