aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorLoïc Hoguin <[email protected]>2014-02-06 19:36:25 +0100
committerLoïc Hoguin <[email protected]>2014-02-06 19:36:25 +0100
commit917cf99e10c41676183d501b86af6e47c95afb89 (patch)
tree4f8877626baaea0a749b8007aab1838aa372aecb /test
parent1f5342f3b8ce40b7bee0b9e8ff76981be6aea2df (diff)
downloadcowboy-917cf99e10c41676183d501b86af6e47c95afb89.tar.gz
cowboy-917cf99e10c41676183d501b86af6e47c95afb89.tar.bz2
cowboy-917cf99e10c41676183d501b86af6e47c95afb89.zip
Add and document the new multipart code
The old undocumented API is removed entirely. While a documentation exists for the new API, it will not be considered set in stone until further testing has been performed, and a file upload example has been added. The new API should be a little more efficient than the old API, especially with smaller messages.
Diffstat (limited to 'test')
-rw-r--r--test/http_SUITE.erl24
-rw-r--r--test/http_SUITE_data/http_multipart.erl22
-rw-r--r--test/http_SUITE_data/http_multipart_stream.erl34
3 files changed, 64 insertions, 16 deletions
diff --git a/test/http_SUITE.erl b/test/http_SUITE.erl
index 13e7b0b..20c65c8 100644
--- a/test/http_SUITE.erl
+++ b/test/http_SUITE.erl
@@ -1,4 +1,4 @@
-%% Copyright (c) 2011-2013, Loïc Hoguin <[email protected]>
+%% Copyright (c) 2011-2014, Loïc Hoguin <[email protected]>
%% Copyright (c) 2011, Anthony Ramine <[email protected]>
%%
%% Permission to use, copy, modify, and/or distribute this software for any
@@ -49,6 +49,7 @@
-export([keepalive_nl/1]).
-export([keepalive_stream_loop/1]).
-export([multipart/1]).
+-export([multipart_large/1]).
-export([nc_rand/1]).
-export([nc_zero/1]).
-export([onrequest/1]).
@@ -135,6 +136,7 @@ groups() ->
keepalive_nl,
keepalive_stream_loop,
multipart,
+ multipart_large,
nc_rand,
nc_zero,
pipeline,
@@ -391,6 +393,7 @@ init_dispatch(Config) ->
{"/static_specify_file/[...]", cowboy_static,
{file, ?config(static_dir, Config) ++ "/style.css"}},
{"/multipart", http_multipart, []},
+ {"/multipart/large", http_multipart_stream, []},
{"/echo/body", http_echo_body, []},
{"/echo/body_qs", http_body_qs, []},
{"/param_all", rest_param_all, []},
@@ -755,8 +758,8 @@ multipart(Config) ->
"This is a preamble."
"\r\n--OHai\r\nX-Name:answer\r\n\r\n42"
"\r\n--OHai\r\nServer:Cowboy\r\n\r\nIt rocks!\r\n"
- "\r\n--OHai--"
- "This is an epiloque."
+ "\r\n--OHai--\r\n"
+ "This is an epilogue."
>>,
{ok, Client2} = cowboy_client:request(<<"POST">>,
build_url("/multipart", Config),
@@ -770,6 +773,21 @@ multipart(Config) ->
{[{<<"server">>, <<"Cowboy">>}], <<"It rocks!\r\n">>}
].
+multipart_large(Config) ->
+ Client = ?config(client, Config),
+ Boundary = "----------",
+ Big = << 0:9000000/unit:8 >>,
+ Bigger = << 0:9999999/unit:8 >>,
+ Body = ["--", Boundary, "\r\ncontent-length: 9000000\r\n\r\n", Big, "\r\n",
+ "--", Boundary, "\r\ncontent-length: 9999999\r\n\r\n", Bigger, "\r\n",
+ "--", Boundary, "--\r\n"],
+ {ok, Client2} = cowboy_client:request(<<"POST">>,
+ build_url("/multipart/large", Config),
+ [{<<"content-type">>, ["multipart/x-large; boundary=", Boundary]}],
+ Body, Client),
+ {ok, 200, _, _} = cowboy_client:response(Client2),
+ ok.
+
nc_reqs(Config, Input) ->
Cat = os:find_executable("cat"),
Nc = os:find_executable("nc"),
diff --git a/test/http_SUITE_data/http_multipart.erl b/test/http_SUITE_data/http_multipart.erl
index c94739f..79bfeb8 100644
--- a/test/http_SUITE_data/http_multipart.erl
+++ b/test/http_SUITE_data/http_multipart.erl
@@ -8,22 +8,18 @@ init({_Transport, http}, Req, []) ->
{ok, Req, {}}.
handle(Req, State) ->
- {Result, Req2} = acc_multipart(Req),
+ {Result, Req2} = acc_multipart(Req, []),
{ok, Req3} = cowboy_req:reply(200, [], term_to_binary(Result), Req2),
{ok, Req3, State}.
terminate(_, _, _) ->
ok.
-acc_multipart(Req) ->
- acc_multipart(cowboy_req:multipart_data(Req), []).
-
-acc_multipart({headers, Headers, Req}, Acc) ->
- acc_multipart(cowboy_req:multipart_data(Req), [{Headers, []}|Acc]);
-acc_multipart({body, Data, Req}, [{Headers, BodyAcc}|Acc]) ->
- acc_multipart(cowboy_req:multipart_data(Req), [{Headers, [Data|BodyAcc]}|Acc]);
-acc_multipart({end_of_part, Req}, [{Headers, BodyAcc}|Acc]) ->
- acc_multipart(cowboy_req:multipart_data(Req),
- [{Headers, list_to_binary(lists:reverse(BodyAcc))}|Acc]);
-acc_multipart({eof, Req}, Acc) ->
- {lists:reverse(Acc), Req}.
+acc_multipart(Req, Acc) ->
+ case cowboy_req:part(Req) of
+ {ok, Headers, Req2} ->
+ {ok, Body, Req3} = cowboy_req:part_body(Req2),
+ acc_multipart(Req3, [{Headers, Body}|Acc]);
+ {done, Req2} ->
+ {lists:reverse(Acc), Req2}
+ end.
diff --git a/test/http_SUITE_data/http_multipart_stream.erl b/test/http_SUITE_data/http_multipart_stream.erl
new file mode 100644
index 0000000..926d150
--- /dev/null
+++ b/test/http_SUITE_data/http_multipart_stream.erl
@@ -0,0 +1,34 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(http_multipart_stream).
+-behaviour(cowboy_http_handler).
+-export([init/3, handle/2, terminate/3]).
+
+init(_, Req, []) ->
+ {ok, Req, undefined}.
+
+handle(Req, State) ->
+ Req2 = multipart(Req),
+ {ok, Req3} = cowboy_req:reply(200, Req2),
+ {ok, Req3, State}.
+
+terminate(_, _, _) ->
+ ok.
+
+multipart(Req) ->
+ case cowboy_req:part(Req) of
+ {ok, [{<<"content-length">>, BinLength}], Req2} ->
+ Length = list_to_integer(binary_to_list(BinLength)),
+ {Length, Req3} = stream_body(Req2, 0),
+ multipart(Req3);
+ {done, Req2} ->
+ Req2
+ end.
+
+stream_body(Req, N) ->
+ case cowboy_req:part_body(Req) of
+ {ok, Data, Req2} ->
+ {N + byte_size(Data), Req2};
+ {more, Data, Req2} ->
+ stream_body(Req2, N + byte_size(Data))
+ end.