diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/http_SUITE.erl | 39 | ||||
| -rw-r--r-- | test/rest_forbidden_resource.erl | 43 |
2 files changed, 79 insertions, 3 deletions
diff --git a/test/http_SUITE.erl b/test/http_SUITE.erl index 002ac4d..74d24b6 100644 --- a/test/http_SUITE.erl +++ b/test/http_SUITE.erl @@ -28,7 +28,7 @@ file_200/1, file_403/1, dir_403/1, file_404/1, file_400/1]). %% http and https. -export([http_10_hostless/1]). %% misc. --export([rest_simple/1, rest_keepalive/1]). %% rest. +-export([rest_simple/1, rest_keepalive/1, rest_keepalive_post/1]). %% rest. %% ct. @@ -46,7 +46,7 @@ groups() -> static_function_etag] ++ BaseTests}, {https, [], BaseTests}, {misc, [], [http_10_hostless]}, - {rest, [], [rest_simple, rest_keepalive]}]. + {rest, [], [rest_simple, rest_keepalive, rest_keepalive_post]}]. init_per_suite(Config) -> application:start(inets), @@ -94,7 +94,9 @@ init_per_group(rest, Config) -> cowboy:start_listener(reset, 100, cowboy_tcp_transport, [{port, Port}], cowboy_http_protocol, [{dispatch, [{'_', [ - {[<<"simple">>], rest_simple_resource, []} + {[<<"simple">>], rest_simple_resource, []}, + {[<<"forbidden_post">>], rest_forbidden_resource, [true]}, + {[<<"simple_post">>], rest_forbidden_resource, [false]} ]}]}]), [{port, Port}|Config]. @@ -563,3 +565,34 @@ rest_keepalive_loop(Socket, N) -> {0, 12} = binary:match(Data, <<"HTTP/1.1 200">>), nomatch = binary:match(Data, <<"Connection: close">>), rest_keepalive_loop(Socket, N - 1). + +rest_keepalive_post(Config) -> + {port, Port} = lists:keyfind(port, 1, Config), + {ok, Socket} = gen_tcp:connect("localhost", Port, + [binary, {active, false}, {packet, raw}]), + ok = rest_keepalive_post_loop(Socket, 10, forbidden_post), + ok = gen_tcp:close(Socket). + +rest_keepalive_post_loop(_Socket, 0, _) -> + ok; +rest_keepalive_post_loop(Socket, N, simple_post) -> + ct:print("simple_post~n"), + ok = gen_tcp:send(Socket, "POST /simple_post HTTP/1.1\r\n" + "Host: localhost\r\nConnection: keep-alive\r\n" + "Content-Length: 5\r\nContent-Type: text/plain\r\n\r\n12345"), + {ok, Data} = gen_tcp:recv(Socket, 0, 6000), + ct:print("data ~p~n", [Data]), + {0, 12} = binary:match(Data, <<"HTTP/1.1 200">>), + nomatch = binary:match(Data, <<"Connection: close">>), + rest_keepalive_post_loop(Socket, N - 1, forbidden_post); +rest_keepalive_post_loop(Socket, N, forbidden_post) -> + ct:print("forbidden~n"), + ok = gen_tcp:send(Socket, "POST /forbidden_post HTTP/1.1\r\n" + "Host: localhost\r\nConnection: keep-alive\r\n" + "Content-Length: 5\r\nContent-Type: text/plain\r\n\r\n12345"), + {ok, Data} = gen_tcp:recv(Socket, 0, 6000), + ct:print("data ~p~n", [Data]), + {0, 12} = binary:match(Data, <<"HTTP/1.1 403">>), + nomatch = binary:match(Data, <<"Connection: close">>), + rest_keepalive_post_loop(Socket, N - 1, simple_post). + diff --git a/test/rest_forbidden_resource.erl b/test/rest_forbidden_resource.erl new file mode 100644 index 0000000..5c6aca0 --- /dev/null +++ b/test/rest_forbidden_resource.erl @@ -0,0 +1,43 @@ +-module(rest_forbidden_resource). +-export([init/3, rest_init/2, allowed_methods/2, forbidden/2, + content_types_provided/2, content_types_accepted/2, + post_is_create/2, create_path/2, to_text/2, from_text/2]). + +init(_Transport, _Req, _Opts) -> + {upgrade, protocol, cowboy_http_rest}. + +rest_init(Req, [Forbidden]) -> + {Headers, Req2} = cowboy_http_req:headers(Req), + {Method, Req3} = cowboy_http_req:method(Req2), + ct:print("method ~p headers ~p", [Method, Headers]), + {ok, Req3, Forbidden}. + +allowed_methods(Req, State) -> + {['GET', 'HEAD', 'POST'], Req, State}. + +forbidden(Req, State=true) -> + {true, Req, State}; +forbidden(Req, State=false) -> + {false, Req, State}. + +content_types_provided(Req, State) -> + {[{{<<"text">>, <<"plain">>, []}, to_text}], Req, State}. + +content_types_accepted(Req, State) -> + {[{{<<"text">>, <<"plain">>, []}, from_text}], Req, State}. + +post_is_create(Req, State) -> + {true, Req, State}. + +create_path(Req, State) -> + {Path, Req2} = cowboy_http_req:raw_path(Req), + {Path, Req2, State}. + +to_text(Req, State) -> + {<<"This is REST!">>, Req, State}. + +from_text(Req, State) -> + {true, Req, State}. + + + |