t/t0005-large-file-response.sh

   1 #!/bin/sh
   2 . ./test-lib.sh
   3 test -r random_blob || die "random_blob required, run with 'make $0'"
   4
   5 if ! grep -v ^VmRSS: /proc/self/status >/dev/null 2>&1
   6 then
   7         t_info "skipping, can't read RSS from /proc/self/status"
   8         exit 0
   9 fi
  10
  11 t_plan 10 "large file response slurp avoidance for $model"
  12
  13 t_begin "setup and startup" && {
  14         rtmpfiles curl_out
  15         zbatery_setup $model 1
  16         # can't load Rack::Lint here since it'll cause Rev to slurp
  17         zbatery -E none -D large-file-response.ru -c $unicorn_config
  18         zbatery_wait_start
  19 }
  20
  21 t_begin "read random blob sha1 and size" && {
  22         random_blob_sha1=$(rsha1 < random_blob)
  23         random_blob_size=$(wc -c < random_blob)
  24 }
  25
  26 t_begin "read current RSS" && {
  27         curl -v http://$listen/rss
  28         dbgcat r_err
  29         rss_before=$(curl -sSfv http://$listen/rss)
  30         t_info "rss_before=$rss_before"
  31 }
  32
  33 t_begin "send a series HTTP/1.1 requests sequentially" && {
  34         for i in a b c
  35         do
  36                 sha1=$( (curl -sSfv http://$listen/random_blob &&
  37                          echo ok >$ok) | rsha1)
  38                 test $sha1 = $random_blob_sha1
  39                 test xok = x$(cat $ok)
  40         done
  41 }
  42
  43 # this was a problem during development
  44 t_begin "HTTP/1.0 test" && {
  45         sha1=$( (curl -0 -sSfv http://$listen/random_blob &&
  46                  echo ok >$ok) | rsha1)
  47         test $sha1 = $random_blob_sha1
  48         test xok = x$(cat $ok)
  49 }
  50
  51 t_begin "HTTP/0.9 test" && {
  52         (
  53                 printf 'GET /random_blob\r\n'
  54                 rsha1 < $fifo > $tmp &
  55                 wait
  56                 echo ok > $ok
  57         ) | socat - TCP:$listen > $fifo
  58         test $(cat $tmp) = $random_blob_sha1
  59         test xok = x$(cat $ok)
  60 }
  61
  62 dbgcat r_err
  63
  64 t_begin "read RSS again" && {
  65         curl -v http://$listen/rss
  66         rss_after=$(curl -sSfv http://$listen/rss)
  67         t_info "rss_after=$rss_after"
  68 }
  69
  70 t_begin "shutdown server" && {
  71         kill -QUIT $zbatery_pid
  72 }
  73
  74 t_begin "compare RSS before and after" && {
  75         diff=$(( $rss_after - $rss_before ))
  76
  77         # default GC malloc limit in MRI:
  78         fudge=$(( 8 * 1024 * 1024 ))
  79
  80         t_info "test diff=$diff < orig=$random_blob_size"
  81         test $diff -le $(( $random_blob_size + $fudge ))
  82 }
  83
  84 dbgcat r_err
  85
  86 t_begin "check stderr" && check_stderr
  87
  88 t_done