unicorn-shopify 4.8.2.5.23

data/t/t0023-before-murder.sh

@@ -0,0 +1,40 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 5 "before murder is called"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	cat >> $unicorn_config <<EOF
+timeout 5
+after_fork { |s,w| File.open('$fifo','w') { |f| f.write '.' } }
+before_murder do |s, w, p| 
+  \$stderr.puts "before murder called:"
+end
+EOF
+	unicorn -c $unicorn_config before_murder.ru &
+	test '.' = $(cat $fifo)
+	unicorn_wait_start
+}
+
+t_begin "start worker=0" && {
+	pid=$(curl http://$listen/) || true
+}
+
+t_begin "ensure before murder log is present" && {
+	dbgcat r_err
+	grep 'before murder called:' $r_err 
+	dbgcat r_err
+	> $r_err
+}
+
+t_begin "killing succeeds" && {
+  kill $unicorn_pid
+  wait
+  kill -0 $unicorn_pid && false
+}
+
+t_begin "check stderr" && {
+	check_stderr
+}
+
+t_done

data/t/t0024-before-murder_once.sh

@@ -0,0 +1,52 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 5 "before murder is called only once"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	cat >> $unicorn_config <<EOF
+timeout 5
+after_fork { |s,w| File.open('$fifo','w') { |f| f.write '.' } }
+class Unicorn::HttpServer
+  alias_method :kill_worker_always, :kill_worker
+  def kill_worker(signal, wpid)
+    @iteration ||= 0
+    \$stderr.puts "kill_worker #{signal} #{wpid} #{@iteration}"
+    kill_worker_always(signal, wpid) if @iteration > 0
+    @iteration += 1
+  end
+end
+before_murder do |s, w, p| 
+  \$stderr.puts "before murder called:"
+end
+EOF
+	unicorn -c $unicorn_config before_murder.ru &
+	test '.' = $(cat $fifo)
+	unicorn_wait_start
+}
+
+t_begin "start worker=0" && {
+	pid=$(curl http://$listen/) || true
+}
+
+t_begin "ensure before murder log is present once and killing is present twice" && {
+	dbgcat r_err
+	grep 'before murder called:' $r_err 
+  test 1 -eq $(grep "before murder called:" $r_err | count_lines)
+  test 2 -eq $(grep timeout $r_err | grep killing | count_lines)
+  test 2 -eq $(grep "kill_worker KILL" $r_err | count_lines)
+	dbgcat r_err
+	> $r_err
+}
+
+t_begin "killing succeeds" && {
+  kill $unicorn_pid
+  wait
+  kill -0 $unicorn_pid && false
+}
+
+t_begin "check stderr" && {
+	check_stderr
+}
+
+t_done

data/t/t0100-rack-input-tests.sh

@@ -0,0 +1,124 @@
+#!/bin/sh
+. ./test-lib.sh
+test -r random_blob || die "random_blob required, run with 'make $0'"
+
+t_plan 10 "rack.input read tests"
+
+t_begin "setup and startup" && {
+	rtmpfiles curl_out curl_err
+	unicorn_setup
+	unicorn -E none -D rack-input-tests.ru -c $unicorn_config
+	blob_sha1=$(rsha1 < random_blob)
+	blob_size=$(count_bytes < random_blob)
+	t_info "blob_sha1=$blob_sha1"
+	unicorn_wait_start
+}
+
+t_begin "corked identity request" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		printf 'PUT / HTTP/1.0\r\n'
+		printf 'Content-Length: %d\r\n\r\n' $blob_size
+		cat random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "corked chunked request" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		content-md5-put < random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "corked identity request (input#size first)" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		printf 'PUT /size_first HTTP/1.0\r\n'
+		printf 'Content-Length: %d\r\n\r\n' $blob_size
+		cat random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "corked identity request (input#rewind first)" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		printf 'PUT /rewind_first HTTP/1.0\r\n'
+		printf 'Content-Length: %d\r\n\r\n' $blob_size
+		cat random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "corked chunked request (input#size first)" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		printf 'PUT /size_first HTTP/1.1\r\n'
+		printf 'Host: example.com\r\n'
+		printf 'Transfer-Encoding: chunked\r\n'
+		printf 'Trailer: Content-MD5\r\n'
+		printf '\r\n'
+		content-md5-put --no-headers < random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "corked chunked request (input#rewind first)" && {
+	rm -f $tmp
+	(
+		cat $fifo > $tmp &
+		printf 'PUT /rewind_first HTTP/1.1\r\n'
+		printf 'Host: example.com\r\n'
+		printf 'Transfer-Encoding: chunked\r\n'
+		printf 'Trailer: Content-MD5\r\n'
+		printf '\r\n'
+		content-md5-put --no-headers < random_blob
+		wait
+		echo ok > $ok
+	) | ( sleep 1 && socat - TCP4:$listen > $fifo )
+	test 1 -eq $(grep $blob_sha1 $tmp |count_lines)
+	test x"$(cat $ok)" = xok
+}
+
+t_begin "regular request" && {
+	curl -sSf -T random_blob http://$listen/ > $curl_out 2> $curl_err
+        test x$blob_sha1 = x$(cat $curl_out)
+        test ! -s $curl_err
+}
+
+t_begin "chunked request" && {
+	curl -sSf -T- < random_blob http://$listen/ > $curl_out 2> $curl_err
+        test x$blob_sha1 = x$(cat $curl_out)
+        test ! -s $curl_err
+}
+
+dbgcat r_err
+
+t_begin "shutdown" && {
+	kill $unicorn_pid
+}
+
+t_done

data/t/t0116-client_body_buffer_size.sh

@@ -0,0 +1,80 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 12 "client_body_buffer_size settings"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	rtmpfiles unicorn_config_tmp one_meg
+	dd if=/dev/zero bs=1M count=1 of=$one_meg
+	cat >> $unicorn_config <<EOF
+after_fork do |server, worker|
+  File.open("$fifo", "wb") { |fp| fp.syswrite "START" }
+end
+EOF
+	cat $unicorn_config > $unicorn_config_tmp
+	echo client_body_buffer_size 0 >> $unicorn_config
+	unicorn -D -c $unicorn_config t0116.ru
+	unicorn_wait_start
+	fs_class=Unicorn::TmpIO
+	mem_class=StringIO
+
+	test x"$(cat $fifo)" = xSTART
+}
+
+t_begin "class for a zero-byte file should be StringIO" && {
+	> $tmp
+	test xStringIO = x"$(curl -T $tmp -sSf http://$listen/input_class)"
+}
+
+t_begin "class for a 1 byte file should be filesystem-backed" && {
+	echo > $tmp
+	test x$fs_class = x"$(curl -T $tmp -sSf http://$listen/tmp_class)"
+}
+
+t_begin "reload with default client_body_buffer_size" && {
+	mv $unicorn_config_tmp $unicorn_config
+	kill -HUP $unicorn_pid
+	test x"$(cat $fifo)" = xSTART
+}
+
+t_begin "class for a 1 byte file should be memory-backed" && {
+	echo > $tmp
+	test x$mem_class = x"$(curl -T $tmp -sSf http://$listen/tmp_class)"
+}
+
+t_begin "class for a random blob file should be filesystem-backed" && {
+	resp="$(curl -T random_blob -sSf http://$listen/tmp_class)"
+	test x$fs_class = x"$resp"
+}
+
+t_begin "one megabyte file should be filesystem-backed" && {
+	resp="$(curl -T $one_meg -sSf http://$listen/tmp_class)"
+	test x$fs_class = x"$resp"
+}
+
+t_begin "reload with a big client_body_buffer_size" && {
+	echo "client_body_buffer_size(1024 * 1024)" >> $unicorn_config
+	kill -HUP $unicorn_pid
+	test x"$(cat $fifo)" = xSTART
+}
+
+t_begin "one megabyte file should be memory-backed" && {
+	resp="$(curl -T $one_meg -sSf http://$listen/tmp_class)"
+	test x$mem_class = x"$resp"
+}
+
+t_begin "one megabyte + 1 byte file should be filesystem-backed" && {
+	echo >> $one_meg
+	resp="$(curl -T $one_meg -sSf http://$listen/tmp_class)"
+	test x$fs_class = x"$resp"
+}
+
+t_begin "killing succeeds" && {
+	kill $unicorn_pid
+}
+
+t_begin "check stderr" && {
+	check_stderr
+}
+
+t_done

data/t/t0116.ru

@@ -0,0 +1,16 @@
+#\ -E none
+use Rack::ContentLength
+use Rack::ContentType, 'text/plain'
+app = lambda do |env|
+  input = env['rack.input']
+  case env["PATH_INFO"]
+  when "/tmp_class"
+    body = input.instance_variable_get(:@tmp).class.name
+  when "/input_class"
+    body = input.class.name
+  else
+    return [ 500, {}, [] ]
+  end
+  [ 200, {}, [ body ] ]
+end
+run app

data/t/t0200-rack-hijack.sh

@@ -0,0 +1,27 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 5 "rack.hijack tests (Rack 1.5+ (Rack::VERSION >= [ 1,2]))"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	unicorn -D -c $unicorn_config hijack.ru
+	unicorn_wait_start
+}
+
+t_begin "check request hijack" && {
+	test "xrequest.hijacked" = x"$(curl -sSfv http://$listen/hijack_req)"
+}
+
+t_begin "check response hijack" && {
+	test "xresponse.hijacked" = x"$(curl -sSfv http://$listen/hijack_res)"
+}
+
+t_begin "killing succeeds" && {
+	kill $unicorn_pid
+}
+
+t_begin "check stderr" && {
+	check_stderr
+}
+
+t_done

data/t/t0300-no-default-middleware.sh

@@ -0,0 +1,20 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 3 "test the -N / --no-default-middleware option"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	unicorn -N -D -c $unicorn_config fails-rack-lint.ru
+	unicorn_wait_start
+}
+
+t_begin "check exit status with Rack::Lint not present" && {
+	test 42 -eq "$(curl -sf -o/dev/null -w'%{http_code}' http://$listen/)"
+}
+
+t_begin "killing succeeds" && {
+	kill $unicorn_pid
+	check_stderr
+}
+
+t_done

data/t/t9000-preread-input.sh

@@ -0,0 +1,48 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 9 "PrereadInput middleware tests"
+
+t_begin "setup and start" && {
+	random_blob_sha1=$(rsha1 < random_blob)
+	unicorn_setup
+	unicorn  -D -c $unicorn_config preread_input.ru
+	unicorn_wait_start
+}
+
+t_begin "single identity request" && {
+	curl -sSf -T random_blob http://$listen/ > $tmp
+}
+
+t_begin "sha1 matches" && {
+	test x"$(cat $tmp)" = x"$random_blob_sha1"
+}
+
+t_begin "single chunked request" && {
+	curl -sSf -T- < random_blob http://$listen/ > $tmp
+}
+
+t_begin "sha1 matches" && {
+	test x"$(cat $tmp)" = x"$random_blob_sha1"
+}
+
+t_begin "app only dispatched twice" && {
+	test 2 -eq "$(grep 'app dispatch:' < $r_err | count_lines )"
+}
+
+t_begin "aborted chunked request" && {
+	rm -f $tmp
+	curl -sSf -T- < $fifo http://$listen/ > $tmp &
+	curl_pid=$!
+	kill -9 $curl_pid
+	wait
+}
+
+t_begin "app only dispatched twice" && {
+	test 2 -eq "$(grep 'app dispatch:' < $r_err | count_lines )"
+}
+
+t_begin "killing succeeds" && {
+	kill -QUIT $unicorn_pid
+}
+
+t_done

data/t/t9001-oob_gc.sh

@@ -0,0 +1,47 @@
+#!/bin/sh
+. ./test-lib.sh
+t_plan 9 "OobGC test"
+
+t_begin "setup and start" && {
+	unicorn_setup
+	unicorn -D -c $unicorn_config oob_gc.ru
+	unicorn_wait_start
+}
+
+t_begin "test default interval (4 requests)" && {
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+}
+
+t_begin "GC starting-request returns immediately" && {
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+}
+
+t_begin "GC is started after 5 requests" && {
+	test xtrue = x$(curl -vsSf http://$listen/ 2>> $tmp)
+}
+
+t_begin "reset GC" && {
+	test xfalse = x$(curl -vsSf -X POST http://$listen/gc_reset 2>> $tmp)
+}
+
+t_begin "test default interval again (3 requests)" && {
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+	test xfalse = x$(curl -vsSf http://$listen/ 2>> $tmp)
+}
+
+t_begin "GC is started after 5 requests" && {
+	test xtrue = x$(curl -vsSf http://$listen/ 2>> $tmp)
+}
+
+t_begin "killing succeeds" && {
+	kill -QUIT $unicorn_pid
+}
+
+t_begin "check_stderr" && check_stderr
+dbgcat r_err
+
+t_done