about summary refs log tree commit homepage
path: root/t
diff options
context:
space:
mode:
authorEric Wong <normalperson@yhbt.net>2011-01-05 22:39:03 -0800
committerEric Wong <normalperson@yhbt.net>2011-01-05 22:58:18 -0800
commitb72a86f66c722d56a6d77ed1d2779ace6ad103ed (patch)
treec074232094580831bd110f1dcce528fd69cdcd63 /t
parent1b69686fd28347eb5c071a9b76e2939bca424f04 (diff)
downloadunicorn-b72a86f66c722d56a6d77ed1d2779ace6ad103ed.tar.gz
Response bodies may capture the block passed to each
and save it for body.close, so don't close the socket
before we have a chance to call body.close
Diffstat (limited to 't')
-rwxr-xr-xt/t0018-write-on-close.sh23
-rw-r--r--t/write-on-close.ru11
2 files changed, 34 insertions, 0 deletions
diff --git a/t/t0018-write-on-close.sh b/t/t0018-write-on-close.sh
new file mode 100755
index 0000000..3afefea
--- /dev/null
+++ b/t/t0018-write-on-close.sh
@@ -0,0 +1,23 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 4 "write-on-close tests for funky response-bodies"
+
+t_begin "setup and start" && {
+        unicorn_setup
+        unicorn -D -c $unicorn_config write-on-close.ru
+        unicorn_wait_start
+}
+
+t_begin "write-on-close response body succeeds" && {
+        test xGoodbye = x"$(curl -sSf http://$listen/)"
+}
+
+t_begin "killing succeeds" && {
+        kill $unicorn_pid
+}
+
+t_begin "check stderr" && {
+        check_stderr
+}
+
+t_done
diff --git a/t/write-on-close.ru b/t/write-on-close.ru
new file mode 100644
index 0000000..54a2f2e
--- /dev/null
+++ b/t/write-on-close.ru
@@ -0,0 +1,11 @@
+class WriteOnClose
+  def each(&block)
+    @callback = block
+  end
+
+  def close
+    @callback.call "7\r\nGoodbye\r\n0\r\n\r\n"
+  end
+end
+use Rack::ContentType, "text/plain"
+run(lambda { |_| [ 200, [%w(Transfer-Encoding chunked)], WriteOnClose.new ] })