cowboy/test/handlers/if_range_h.erl

%% This module defines the ranges_provided callback
%% and a generate_etag callback that returns something
%% different depending on query string. It also defines
%% a last_modified callback that must be ignored when a
%% date is provided in if_range.

-module(if_range_h).

-export([init/2]).
-export([content_types_provided/2]).
-export([ranges_provided/2]).
-export([generate_etag/2]).
-export([last_modified/2]).
-export([get_text_plain/2]).
-export([get_text_plain_bytes/2]).

init(Req, State) ->
	{cowboy_rest, Req, State}.

content_types_provided(Req, State) ->
	{[{{<<"text">>, <<"plain">>, []}, get_text_plain}], Req, State}.

%% Simulate the callback being missing.
ranges_provided(#{qs := <<"missing-ranges_provided">>}, _) ->
	no_call;
ranges_provided(Req=#{qs := <<"empty-ranges_provided">>}, State) ->
	{[], Req, State};
ranges_provided(Req, State) ->
	{[{<<"bytes">>, get_text_plain_bytes}], Req, State}.

generate_etag(Req=#{qs := <<"weak-etag">>}, State) ->
	{{weak, <<"weak-no-match">>}, Req, State};
generate_etag(Req, State) ->
	{{strong, <<"strong-and-match">>}, Req, State}.

last_modified(Req, State) ->
	{{{2222, 2, 22}, {11, 11, 11}}, Req, State}.

get_text_plain(Req, State) ->
	{<<"This is REST!">>, Req, State}.

get_text_plain_bytes(Req, State) ->
	%% We send everything in one part, since we are not testing
	%% this callback specifically.
	Body = <<"This is ranged REST!">>,
	{[{{0, byte_size(Body) - 1, byte_size(Body)}, Body}], Req, State}.