From 370fb8c7811704ed65384f599b52ac1b6d0c36c9 Mon Sep 17 00:00:00 2001
From: Eric Wong <normalperson@yhbt.net>
Date: Wed, 5 Jan 2011 18:01:36 -0800
Subject: event_machine: refactor async.callback for keepalive

async.callback will be useful with Coolio (and more!) soon, so
ensure it works as well as the rest of Rainbows!
---
 t/async_chunk_app.ru          | 44 ++++++++++++++++++++++
 t/t0402-em-async-keepalive.sh | 86 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 130 insertions(+)
 create mode 100644 t/async_chunk_app.ru
 create mode 100644 t/t0402-em-async-keepalive.sh

(limited to 't')

diff --git a/t/async_chunk_app.ru b/t/async_chunk_app.ru
new file mode 100644
index 0000000..26b9915
--- /dev/null
+++ b/t/async_chunk_app.ru
@@ -0,0 +1,44 @@
+# based on async_examples/async_app.ru by James Tucker
+class DeferrableChunkBody
+  include EventMachine::Deferrable
+
+  def call(*body)
+    body.each do |chunk|
+      @body_callback.call("#{chunk.size.to_s(16)}\r\n")
+      @body_callback.call(chunk)
+      @body_callback.call("\r\n")
+    end
+  end
+
+  def each(&block)
+    @body_callback = block
+  end
+
+  def finish
+    @body_callback.call("0\r\n\r\n")
+  end
+end
+
+class AsyncChunkApp
+  def call(env)
+    body = DeferrableChunkBody.new
+    body.callback { body.finish }
+    headers = {
+      'Content-Type' => 'text/plain',
+      'Transfer-Encoding' => 'chunked',
+    }
+    EM.next_tick {
+      env['async.callback'].call([ 200, headers, body ])
+    }
+    EM.add_timer(1) {
+      body.call "Hello "
+
+      EM.add_timer(1) {
+        body.call "World #{env['PATH_INFO']}\n"
+        body.succeed
+      }
+    }
+    nil
+  end
+end
+run AsyncChunkApp.new
diff --git a/t/t0402-em-async-keepalive.sh b/t/t0402-em-async-keepalive.sh
new file mode 100644
index 0000000..24eb678
--- /dev/null
+++ b/t/t0402-em-async-keepalive.sh
@@ -0,0 +1,86 @@
+#!/bin/sh
+. ./test-lib.sh
+case $model in
+NeverBlock|EventMachine) ;;
+*)
+	t_info "skipping $T since it's not compatible with $model"
+	exit 0
+	;;
+esac
+
+t_plan 9 "async_chunk_app test for test for EM"
+
+CONFIG_RU=async_chunk_app.ru
+
+t_begin "setup and start" && {
+	rainbows_setup
+	rtmpfiles a b c curl_err expect
+
+	# this does not does not support Rack::Lint
+	rainbows -E none -D $CONFIG_RU -c $unicorn_config
+	rainbows_wait_start
+
+	echo 'Hello World /0' >> $expect
+	echo 'Hello World /1' >> $expect
+	echo 'Hello World /2' >> $expect
+}
+
+t_begin "async.callback supports pipelining" && {
+	rm -f $tmp
+	t0=$(date +%s)
+	(
+		cat $fifo > $tmp &
+		printf 'GET /0 HTTP/1.1\r\nHost: example.com\r\n\r\n'
+		printf 'GET /1 HTTP/1.1\r\nHost: example.com\r\n\r\n'
+		printf 'GET /2 HTTP/1.0\r\nHost: example.com\r\n\r\n'
+		wait
+	) | socat - TCP:$listen > $fifo
+	t1=$(date +%s)
+	elapsed=$(( $t1 - $t0 ))
+	t_info "elapsed=$elapsed $model.$0 ($t_current)"
+	test 3 -eq "$(fgrep 'HTTP/1.1 200 OK' $tmp | wc -l)"
+}
+
+t_begin "async.callback supports keepalive" && {
+	t0=$(date +%s)
+	curl -v --no-buffer -sSf http://$listen/[0-2] > $tmp 2>> $curl_err
+	t1=$(date +%s)
+	elapsed=$(( $t1 - $t0 ))
+	t_info "elapsed=$elapsed $model.$0 ($t_current)"
+	cmp $expect $tmp
+	test 2 -eq "$(fgrep 'Re-using existing connection!' $curl_err |wc -l)"
+	rm -f $curl_err
+}
+
+t_begin "send async requests off in parallel" && {
+	t0=$(date +%s)
+	curl --no-buffer -sSf http://$listen/[0-2] > $a 2>> $curl_err &
+	curl --no-buffer -sSf http://$listen/[0-2] > $b 2>> $curl_err &
+	curl --no-buffer -sSf http://$listen/[0-2] > $c 2>> $curl_err &
+}
+
+t_begin "wait for curl terminations" && {
+	wait
+	t1=$(date +%s)
+	elapsed=$(( $t1 - $t0 ))
+	t_info "elapsed=$elapsed"
+}
+
+t_begin "termination signal sent" && {
+	kill $rainbows_pid
+}
+
+t_begin "no errors from curl" && {
+	test ! -s $curl_err
+}
+
+t_begin "no errors in stderr" && check_stderr
+
+t_begin "responses match expected" && {
+	cmp $expect $a
+	cmp $expect $b
+	cmp $expect $c
+}
+
+t_done
+
-- 
cgit v1.2.3-24-ge0c7