aboutsummaryrefslogtreecommitdiffstats
path: root/test
diff options
context:
space:
mode:
authorLoïc Hoguin <[email protected]>2013-12-02 16:14:37 +0100
committerLoïc Hoguin <[email protected]>2013-12-02 16:14:37 +0100
commit9f33a20d9425daf7491836e73f503c206cc170be (patch)
tree17c4f08a02b5823696e3353d9f45776248dbe8e8 /test
parent02a04b1a562b1aa454bae98764da5935f12da3c8 (diff)
parent1c474af8ee4c61c9cbbf6ef4e121d1d82af75151 (diff)
downloadcowboy-9f33a20d9425daf7491836e73f503c206cc170be.tar.gz
cowboy-9f33a20d9425daf7491836e73f503c206cc170be.tar.bz2
cowboy-9f33a20d9425daf7491836e73f503c206cc170be.zip
Merge branch 'fix/keepalive_loop_stream_recv' of git://github.com/fishcakez/cowboy
Diffstat (limited to 'test')
-rw-r--r--test/http_SUITE.erl24
-rw-r--r--test/http_SUITE_data/http_loop_stream_recv.erl41
2 files changed, 65 insertions, 0 deletions
diff --git a/test/http_SUITE.erl b/test/http_SUITE.erl
index f0196ec..977cc1a 100644
--- a/test/http_SUITE.erl
+++ b/test/http_SUITE.erl
@@ -47,6 +47,7 @@
-export([http10_hostless/1]).
-export([keepalive_max/1]).
-export([keepalive_nl/1]).
+-export([keepalive_stream_loop/1]).
-export([multipart/1]).
-export([nc_rand/1]).
-export([nc_zero/1]).
@@ -132,6 +133,7 @@ groups() ->
http10_hostless,
keepalive_max,
keepalive_nl,
+ keepalive_stream_loop,
multipart,
nc_rand,
nc_zero,
@@ -407,6 +409,7 @@ init_dispatch(Config) ->
{"/rest_expires", rest_expires, []},
{"/rest_empty_resource", rest_empty_resource, []},
{"/loop_recv", http_loop_recv, []},
+ {"/loop_stream_recv", http_loop_stream_recv, []},
{"/loop_timeout", http_loop_timeout, []},
{"/", http_handler, []}
]}
@@ -725,6 +728,27 @@ keepalive_nl_loop(Client, URL, N) ->
ok = Transport:send(Socket, <<"\r\n">>), %% empty line
keepalive_nl_loop(Client3, URL, N - 1).
+keepalive_stream_loop(Config) ->
+ Client = ?config(client, Config),
+ Transport = ?config(transport, Config),
+ {ok, Client2} = cowboy_client:connect(
+ Transport, "localhost", ?config(port, Config), Client),
+ keepalive_stream_loop(Client2, 10).
+
+keepalive_stream_loop(Client, 0) ->
+ {error, closed} = cowboy_client:response(Client),
+ ok;
+keepalive_stream_loop(Client, N) ->
+ {ok, _} = cowboy_client:raw_request("POST /loop_stream_recv HTTP/1.1\r\n"
+ "Host: localhost\r\n"
+ "Connection: keepalive\r\n"
+ "Transfer-Encoding: chunked\r\n\r\n", Client),
+ _ = [{ok, _} = cowboy_client:raw_request(<<"4\r\n",Id:32,"\r\n">>, Client) ||
+ Id <- lists:seq(1, 250)],
+ {ok, _} = cowboy_client:raw_request(<<"0\r\n\r\n">>, Client),
+ {ok, 200, _, _} = cowboy_client:response(Client),
+ keepalive_stream_loop(Client, N-1).
+
multipart(Config) ->
Client = ?config(client, Config),
Body = <<
diff --git a/test/http_SUITE_data/http_loop_stream_recv.erl b/test/http_SUITE_data/http_loop_stream_recv.erl
new file mode 100644
index 0000000..87113c6
--- /dev/null
+++ b/test/http_SUITE_data/http_loop_stream_recv.erl
@@ -0,0 +1,41 @@
+%% Feel free to use, reuse and abuse the code in this file.
+
+-module(http_loop_stream_recv).
+-export([init/3]).
+-export([info/3]).
+-export([terminate/3]).
+
+init({_, http}, Req, _) ->
+ receive after 100 -> ok end,
+ self() ! stream,
+ {loop, Req, 1, 100}.
+
+info(stream, Req, Id) ->
+ case stream_next(Req) of
+ {ok, Id, Req2} ->
+ info(stream, Req2, Id+1);
+ {done, Req2} ->
+ {ok, Req3} = cowboy_req:reply(200, Req2),
+ {ok, Req3, Id}
+ end.
+
+terminate({normal, shutdown}, _, _) ->
+ ok.
+
+stream_next(Req) ->
+ stream_next(<<>>, Req).
+
+stream_next(Buffer, Req) ->
+ case cowboy_req:stream_body(Req) of
+ {ok, Packet, Req2} ->
+ case <<Buffer/binary, Packet/binary>> of
+ <<Id:32>> ->
+ {ok, Id, Req2};
+ Buffer2 when byte_size(Buffer2) < 4 ->
+ stream_next(Buffer2, Req2);
+ _InvalidBuffer ->
+ {error, invalid_chunk}
+ end;
+ Other ->
+ Other
+ end.