aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorTom Burdick <[email protected]>2012-01-23 12:42:04 -0600
committerTom Burdick <[email protected]>2012-01-23 12:42:04 -0600
commit62de899c9597ae36bfb2124838bc434fe1e61cb6 (patch)
tree9454107fdcceb373e51c8ad867933127f5eba25f
parent9ee120db78dc3619292ae5117c8f25e8bd4aa621 (diff)
downloadcowboy-62de899c9597ae36bfb2124838bc434fe1e61cb6.tar.gz
cowboy-62de899c9597ae36bfb2124838bc434fe1e61cb6.tar.bz2
cowboy-62de899c9597ae36bfb2124838bc434fe1e61cb6.zip
added test for posting to a rest controller where forbidden returns true on a keep alive socket
-rw-r--r--test/http_SUITE.erl39
-rw-r--r--test/rest_forbidden_resource.erl43
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}.
+
+
+