3 files changed, 74 insertions, 0 deletions
diff --git a/test/handlers/long_polling_h.erl b/test/handlers/long_polling_h.erl
new file mode 100644
index 0000000..21f1d4d
--- /dev/null
+++ b/test/handlers/long_polling_h.erl
@@ -0,0 +1,27 @@
+%% This module implements a loop handler for long-polling.
+%% It starts by sending itself a message after 200ms,
+%% then sends another after that for a total of 3 messages.
+%% When it receives the last message, it sends a 102 reply back.
+
+-module(long_polling_h).
+-behaviour(cowboy_loop_handler).
+
+-export([init/3]).
+-export([info/3]).
+-export([terminate/3]).
+
+init(_, Req, _) ->
+	erlang:send_after(200, self(), timeout),
+	{loop, Req, 2, 5000, hibernate}.
+
+info(timeout, Req, 0) ->
+	{ok, Req2} = cowboy_req:reply(102, Req),
+	{ok, Req2, 0};
+info(timeout, Req, Count) ->
+	erlang:send_after(200, self(), timeout),
+	{loop, Req, Count - 1, hibernate}.
+
+terminate({normal, shutdown}, _, 0) ->
+	ok;
+terminate({error, overflow}, _, _) ->
+	ok.
diff --git a/test/handlers/loop_handler_body_h.erl b/test/handlers/loop_handler_body_h.erl
new file mode 100644
index 0000000..db69b02
--- /dev/null
+++ b/test/handlers/loop_handler_body_h.erl
@@ -0,0 +1,24 @@
+%% This module implements a loop handler that reads
+%% the request body after sending itself a message,
+%% checks that its size is exactly 100000 bytes,
+%% then sends a 200 reply back.
+
+-module(loop_handler_body_h).
+-behaviour(cowboy_loop_handler).
+
+-export([init/3]).
+-export([info/3]).
+-export([terminate/3]).
+
+init(_, Req, _) ->
+	self() ! timeout,
+	{loop, Req, undefined, 5000, hibernate}.
+
+info(timeout, Req, State) ->
+	{ok, Body, Req2} = cowboy_req:body(Req),
+	100000 = byte_size(Body),
+	{ok, Req3} = cowboy_req:reply(200, Req2),
+	{ok, Req3, State}.
+
+terminate({normal, shutdown}, _, _) ->
+	ok.
diff --git a/test/handlers/loop_handler_timeout_h.erl b/test/handlers/loop_handler_timeout_h.erl
new file mode 100644
index 0000000..1125046
--- /dev/null
+++ b/test/handlers/loop_handler_timeout_h.erl
@@ -0,0 +1,23 @@
+%% This module implements a loop handler that sends
+%% itself a timeout that will intentionally arrive
+%% too late, as it configures itself to only wait
+%% 200ms before closing the connection in init/3.
+%% This results in a 204 reply being sent back by Cowboy.
+
+-module(loop_handler_timeout_h).
+-behaviour(cowboy_loop_handler).
+
+-export([init/3]).
+-export([info/3]).
+-export([terminate/3]).
+
+init(_, Req, _) ->
+	erlang:send_after(1000, self(), timeout),
+	{loop, Req, undefined, 200, hibernate}.
+
+info(timeout, Req, State) ->
+	{ok, Req2} = cowboy_req:reply(500, Req),
+	{ok, Req2, State}.
+
+terminate({normal, timeout}, _, _) ->
+	ok.