From 980342f73c381413a09fbea94b43428e0f16d787 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Lo=C3=AFc=20Hoguin?= <essen@ninenines.eu>
Date: Sat, 26 Apr 2014 13:46:55 +0200
Subject: Make loop handlers work with SPDY

Adds a loop_handler test suite that runs all tests under HTTP, HTTPS,
SPDY each with and without the compress option enabled.

Fixes output filtering that used to filter more than it should have.
This forces us to parse the string sent by the emulator, which means
it's probably not perfect yet. But it should at least not hide errors
we want to see.

Fix a crash in the output filtering code that entirely disabled
output. Now when there is a crash the normal tty output is restored.

Handlers are now in test/handlers/ as they can be reused between
suites.

Only generate a single certificate for the whole ct run to speed
things up when we got many different test groups each needing
certificates.
---
 test/handlers/long_polling_h.erl         | 27 +++++++++++++++++++++++++++
 test/handlers/loop_handler_body_h.erl    | 24 ++++++++++++++++++++++++
 test/handlers/loop_handler_timeout_h.erl | 23 +++++++++++++++++++++++
 3 files changed, 74 insertions(+)
 create mode 100644 test/handlers/long_polling_h.erl
 create mode 100644 test/handlers/loop_handler_body_h.erl
 create mode 100644 test/handlers/loop_handler_timeout_h.erl

(limited to 'test/handlers')

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