aboutsummaryrefslogtreecommitdiffstats
path: root/test/http_SUITE_data/http_set_resp.erl
blob: 6ac4c8e73111b943b6196f37e069a4d26e900c22 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
%% Feel free to use, reuse and abuse the code in this file.

-module(http_set_resp).

-export([init/2]).

init(Req, Opts) ->
	Headers = proplists:get_value(headers, Opts, #{}),
	Body = proplists:get_value(body, Opts, <<"http_handler_set_resp">>),
	Req2 = lists:foldl(fun({Name, Value}, R) ->
		cowboy_req:set_resp_header(Name, Value, R)
	end, Req, maps:to_list(Headers)),
	Req3 = cowboy_req:set_resp_body(Body, Req2),
	Req4 = cowboy_req:set_resp_header(<<"x-cowboy-test">>, <<"ok">>, Req3),
	Req5 = cowboy_req:set_resp_cookie(<<"cake">>, <<"lie">>, [], Req4),
	case cowboy_req:has_resp_header(<<"x-cowboy-test">>, Req5) of
		false -> {ok, Req5, Opts};
		true ->
			case cowboy_req:has_resp_body(Req5) of
				false ->
					{ok, Req5, Opts};
				true ->
					{ok, cowboy_req:reply(200, Req5), Opts}
			end
	end.