From 62de899c9597ae36bfb2124838bc434fe1e61cb6 Mon Sep 17 00:00:00 2001
From: Tom Burdick <thomas.burdick@gmail.com>
Date: Mon, 23 Jan 2012 12:42:04 -0600
Subject: added test for posting to a rest controller where forbidden returns
 true on a keep alive socket

---
 test/http_SUITE.erl              | 39 +++++++++++++++++++++++++++++++++---
 test/rest_forbidden_resource.erl | 43 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 79 insertions(+), 3 deletions(-)
 create mode 100644 test/rest_forbidden_resource.erl

(limited to 'test')

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}.
+
+
+
-- 
cgit v1.2.3