diff options
| -rw-r--r-- | test/handlers/switch_handler_h.erl | 148 | ||||
| -rw-r--r-- | test/rest_handler_SUITE.erl | 105 |
2 files changed, 244 insertions, 9 deletions
diff --git a/test/handlers/switch_handler_h.erl b/test/handlers/switch_handler_h.erl index f1580d1..264be68 100644 --- a/test/handlers/switch_handler_h.erl +++ b/test/handlers/switch_handler_h.erl @@ -1,21 +1,163 @@ -module(switch_handler_h). -export([init/2]). + +-export([allowed_methods/2]). +-export([allow_missing_post/2]). +-export([charsets_provided/2]). +-export([content_types_accepted/2]). -export([content_types_provided/2]). +-export([delete_completed/2]). +-export([delete_resource/2]). +-export([forbidden/2]). +-export([is_authorized/2]). +-export([is_conflict/2]). +-export([known_methods/2]). +-export([languages_provided/2]). +-export([malformed_request/2]). +-export([moved_permanently/2]). +-export([moved_temporarily/2]). +-export([multiple_choices/2]). +-export([options/2]). +-export([previously_existed/2]). +-export([rate_limited/2]). +-export([resource_exists/2]). +-export([service_available/2]). +-export([uri_too_long/2]). +-export([valid_content_headers/2]). +-export([valid_entity_length/2]). + +-export([accept/2]). -export([provide/2]). + -export([info/3]). init(Req, State) -> {cowboy_rest, Req, State}. +allowed_methods(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +allow_missing_post(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +charsets_provided(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +content_types_accepted(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + content_types_provided(Req, State) -> - {[{<<"text/plain">>, provide}], Req, State}. + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +delete_completed(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +delete_resource(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +forbidden(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +is_authorized(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +is_conflict(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +known_methods(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +languages_provided(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +malformed_request(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +moved_permanently(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +moved_temporarily(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +multiple_choices(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +options(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +previously_existed(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +rate_limited(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +resource_exists(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +service_available(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +uri_too_long(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +valid_content_headers(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +valid_entity_length(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +accept(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +provide(Req, State) -> + maybe_switch_handler(Req, State, ?FUNCTION_NAME). + +maybe_switch_handler(Req=#{qs := Qs}, State, StateName) -> + case atom_to_binary(StateName, latin1) of + Qs -> do_switch_handler(Req, State); + _ -> do_default(Req, State, StateName) + end. + +%% These are all the methods necessary to reach all callbacks. +do_default(Req, State, allowed_methods) -> + {[<<"GET">>, <<"POST">>, <<"PUT">>, <<"DELETE">>, <<"OPTIONS">>], Req, State}; +%% We need to accept/provide media types to reach these callbacks. +do_default(Req, State, content_types_accepted) -> + {[{<<"text/plain">>, accept}], Req, State}; +do_default(Req, State, content_types_provided) -> + {[{<<"text/plain">>, provide}], Req, State}; +%% We need resource_exists to return false to reach these callbacks. +do_default(Req=#{qs := <<"allow_missing_post">>}, State, resource_exists) -> + {false, Req, State}; +do_default(Req=#{qs := <<"moved_permanently">>}, State, resource_exists) -> + {false, Req, State}; +do_default(Req=#{qs := <<"moved_temporarily">>}, State, resource_exists) -> + {false, Req, State}; +do_default(Req=#{qs := <<"previously_existed">>}, State, resource_exists) -> + {false, Req, State}; +%% We need previously_existed to return true to reach these callbacks. +do_default(Req=#{qs := <<"moved_permanently">>}, State, previously_existed) -> + {true, Req, State}; +do_default(Req=#{qs := <<"moved_temporarily">>}, State, previously_existed) -> + {true, Req, State}; +%% We need the DELETE to suceed to reach this callback. +do_default(Req=#{qs := <<"delete_completed">>}, State, delete_resource) -> + {true, Req, State}; +%% We should never reach these two callbacks. +do_default(Req, State, accept) -> + {false, Req, State}; +do_default(Req, State, provide) -> + {<<"This is REST!">>, Req, State}; +%% Simulate the callback being missing in any other cases. +do_default(_, _, _) -> + no_call. -provide(Req0, run) -> +do_switch_handler(Req0, run) -> Req = cowboy_req:stream_reply(200, Req0), send_after(0), {{switch_handler, cowboy_loop}, Req, 0}; -provide(Req0, hibernate) -> +do_switch_handler(Req0, hibernate) -> Req = cowboy_req:stream_reply(200, Req0), send_after(0), {{switch_handler, cowboy_loop, hibernate}, Req, 0}. diff --git a/test/rest_handler_SUITE.erl b/test/rest_handler_SUITE.erl index 09ce8fb..32dbdff 100644 --- a/test/rest_handler_SUITE.erl +++ b/test/rest_handler_SUITE.erl @@ -313,19 +313,112 @@ rate_not_limited(Config) -> {response, nofin, 200, _} = gun:await(ConnPid, Ref), ok. -switch_handler(Config) -> - doc("Switch REST to loop handler for streaming the response body."), +switch_handler_allowed_methods(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_allow_missing_post(Config) -> + do_req_body_switch_handler(Config, post, ?FUNCTION_NAME). + +switch_handler_charsets_provided(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_content_types_accepted(Config) -> + do_req_body_switch_handler(Config, post, ?FUNCTION_NAME). + +switch_handler_content_types_provided(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_delete_completed(Config) -> + do_no_body_switch_handler(Config, delete, ?FUNCTION_NAME). + +switch_handler_delete_resource(Config) -> + do_no_body_switch_handler(Config, delete, ?FUNCTION_NAME). + +switch_handler_forbidden(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_is_authorized(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_is_conflict(Config) -> + do_req_body_switch_handler(Config, put, ?FUNCTION_NAME). + +switch_handler_known_methods(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_languages_provided(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_malformed_request(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_moved_permanently(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_moved_temporarily(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_multiple_choices(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_options(Config) -> + do_no_body_switch_handler(Config, options, ?FUNCTION_NAME). + +switch_handler_previously_existed(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_rate_limited(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_resource_exists(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_service_available(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_uri_too_long(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_valid_content_headers(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_valid_entity_length(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +switch_handler_accept(Config) -> + do_req_body_switch_handler(Config, post, ?FUNCTION_NAME). + +switch_handler_provide(Config) -> + do_no_body_switch_handler(Config, get, ?FUNCTION_NAME). + +do_no_body_switch_handler(Config, Method, StateName0) -> + doc("Switch REST to loop handler for streaming the response body, " + "with and without options."), + "switch_handler_" ++ StateName = atom_to_list(StateName0), + do_no_body_switch_handler1(Config, Method, "/switch_handler?" ++ StateName), + do_no_body_switch_handler1(Config, Method, "/switch_handler_opts?" ++ StateName). + +do_no_body_switch_handler1(Config, Method, Path) -> ConnPid = gun_open(Config), - Ref = gun:get(ConnPid, "/switch_handler", [{<<"accept-encoding">>, <<"gzip">>}]), + Ref = gun:Method(ConnPid, Path, [{<<"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."), +do_req_body_switch_handler(Config, Method, StateName0) -> + doc("Switch REST to loop handler for streaming the response body, " + "with and without options."), + "switch_handler_" ++ StateName = atom_to_list(StateName0), + do_req_body_switch_handler1(Config, Method, "/switch_handler?" ++ StateName), + do_req_body_switch_handler1(Config, Method, "/switch_handler_opts?" ++ StateName). + +do_req_body_switch_handler1(Config, Method, Path) -> ConnPid = gun_open(Config), - Ref = gun:get(ConnPid, "/switch_handler_opts", [{<<"accept-encoding">>, <<"gzip">>}]), + Ref = gun:Method(ConnPid, Path, [ + {<<"accept-encoding">>, <<"gzip">>}, + {<<"content-type">>, <<"text/plain">>} + ], <<"Hocus PocuSwitch!">>), {response, nofin, 200, Headers} = gun:await(ConnPid, Ref), {ok, Body} = gun:await_body(ConnPid, Ref), <<"Hello\nstreamed\nworld!\n">> = do_decode(Headers, Body), |