umgeher/cowboy
0
Fork 0
mirror of https://github.com/ninenines/cowboy.git synced 2025-07-16 05:00:24 +00:00
Code Issues Wiki Activity

Add functions for getting and setting response headers

Browse source
This commit is contained in:
Johan Lövdahl 2016-11-13 15:39:40 +01:00 • committed by Loïc Hoguin
parent 42c95b1584
commit 3f4e79d42f
No known key found for this signature in database
GPG key ID: 71366FF21851DF03
7 changed files with 231 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

23
test/handlers/resp_h.erl
View file

@ -25,6 +25,29 @@ do(<<"set_resp_cookie4">>, Req0, Opts) ->
do(<<"set_resp_header">>, Req0, Opts) ->
Req = cowboy_req:set_resp_header(<<"content-type">>, <<"text/plain">>, Req0),
{ok, cowboy_req:reply(200, #{}, "OK", Req), Opts};
do(<<"set_resp_headers">>, Req0, Opts) ->
Req = cowboy_req:set_resp_headers(#{<<"x-header-test1">> => <<"test1">>, <<"x-header-test2">> => <<"test2">>}, Req0),
{ok, cowboy_req:reply(200, #{}, "OK", Req), Opts};
do(<<"resp_headers">>, Req0, Opts) ->
Req1 = cowboy_req:set_resp_header(<<"x-header-test1">>, <<"test1">>, Req0),
Req2 = cowboy_req:set_resp_headers(#{<<"x-header-test2">> => <<"test2">>, <<"x-header-test3">> => <<"test3">>}, Req1),
Headers = cowboy_req:resp_headers(Req2),
true = maps:is_key(<<"x-header-test1">>, Headers),
true = maps:is_key(<<"x-header-test2">>, Headers),
true = maps:is_key(<<"x-header-test3">>, Headers),
{ok, cowboy_req:reply(200, #{}, "OK", Req2), Opts};
do(<<"resp_header_defined">>, Req0, Opts) ->
Req1 = cowboy_req:set_resp_header(<<"x-header-test1">>, <<"test1">>, Req0),
<<"test1">> = cowboy_req:resp_header(<<"x-header-test1">>, Req1),
<<"test1">> = cowboy_req:resp_header(<<"x-header-test1">>, Req1, foo),
{ok, cowboy_req:reply(200, #{}, "OK", Req0), Opts};
do(<<"resp_header_default">>, Req0, Opts) ->
undefined = cowboy_req:resp_header(<<"x-header-test1">>, Req0),
<<"ok">> = cowboy_req:resp_header(<<"x-header-test1">>, Req0, <<"ok">>),
{ok, cowboy_req:reply(200, #{}, "OK", Req0), Opts};
do(<<"resp_headers_empty">>, Req0, Opts) ->
#{} = cowboy_req:resp_headers(Req0),
{ok, cowboy_req:reply(200, #{}, "OK", Req0), Opts};
do(<<"set_resp_body">>, Req0, Opts) ->
Arg = cowboy_req:binding(arg, Req0),
Req1 = case Arg of

19
test/req_SUITE.erl
View file

@ -487,6 +487,25 @@ set_resp_header(Config) ->
true = lists:keymember(<<"content-type">>, 1, Headers),
ok.
set_resp_headers(Config) ->
doc("Response using set_resp_headers."),
{200, Headers, <<"OK">>} = do_get("/resp/set_resp_headers", Config),
true = lists:keymember(<<"x-header-test1">>, 1, Headers),
true = lists:keymember(<<"x-header-test2">>, 1, Headers),
ok.
resp_header(Config) ->
doc("Response header with/without default."),
{200, _, <<"OK">>} = do_get("/resp/resp_header_defined", Config),
{200, _, <<"OK">>} = do_get("/resp/resp_header_default", Config),
ok.
resp_headers(Config) ->
doc("Get all response headers."),
{200, _, <<"OK">>} = do_get("/resp/resp_headers", Config),
{200, _, <<"OK">>} = do_get("/resp/resp_headers_empty", Config),
ok.
set_resp_body(Config) ->
doc("Response using set_resp_body."),
{200, _, <<"OK">>} = do_get("/resp/set_resp_body", Config),