diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_proxy_pass_no_buffering.rb | 148 | ||||
| -rw-r--r-- | test/test_wbuf.rb | 8 |
2 files changed, 152 insertions, 4 deletions
diff --git a/test/test_proxy_pass_no_buffering.rb b/test/test_proxy_pass_no_buffering.rb new file mode 100644 index 0000000..48b8241 --- /dev/null +++ b/test/test_proxy_pass_no_buffering.rb @@ -0,0 +1,148 @@ +# Copyright (C) 2015-2016 all contributors <yahns-public@yhbt.net> +# License: GPL-3.0+ (https://www.gnu.org/licenses/gpl-3.0.txt) +# frozen_string_literal: true +require_relative 'server_helper' +begin + require 'kcar' +rescue LoadError +end +require 'digest/md5' +class TestProxyPassNoBuffering < Testcase + ENV["N"].to_i > 1 and parallelize_me! + include ServerHelper + STR4 = 'abcd' * (256 * 1024) + NCHUNK = 50 + class ProxiedApp + def call(env) + case env['REQUEST_METHOD'] + when 'GET' + case env['PATH_INFO'] + when '/giant-body' + h = [ %W(content-type text/pain), + %W(content-length #{NCHUNK * STR4.size}) ] + body = Object.new + def body.each + NCHUNK.times { yield STR4 } + end + [ 200, h, body ] + end + end + end + end + + def setup + @srv2 = TCPServer.new(ENV["TEST_HOST"] || "127.0.0.1", 0) + server_helper_setup + skip "kcar missing yahns/proxy_pass" unless defined?(Kcar) + require 'yahns/proxy_pass' + end + + def teardown + @srv2.close if defined?(@srv2) && !@srv2.closed? + server_helper_teardown + end + + def check_headers(io) + l = io.gets + assert_match %r{\AHTTP/1\.[01] 200\b}, l + begin + l = io.gets + end until l == "\r\n" + end + + def test_proxy_pass_no_buffering + err, cfg, host, port = @err, Yahns::Config.new, @srv.addr[3], @srv.addr[1] + host2, port2 = @srv2.addr[3], @srv2.addr[1] + pxp = Yahns::ProxyPass.new("http://#{host2}:#{port2}", + proxy_buffering: false) + pid = mkserver(cfg) do + ObjectSpace.each_object(Yahns::TmpIO) { |io| io.close unless io.closed? } + @srv2.close + cfg.instance_eval do + app(:rack, pxp) { listen "#{host}:#{port}" } + stderr_path err.path + end + end + + pid2 = mkserver(cfg, @srv2) do + ObjectSpace.each_object(Yahns::TmpIO) { |io| io.close unless io.closed? } + @srv.close + cfg.instance_eval do + app(:rack, ProxiedApp.new) do + output_buffering false + listen "#{host2}:#{port2}" + end + stderr_path err.path + end + end + s = TCPSocket.new(host, port) + req = "GET /giant-body HTTP/1.1\r\nHost: example.com\r\n" \ + "Connection: close\r\n\r\n" + s.write(req) + sleep 1 + 10.times do + sleep 0.1 + # ensure no files get created + if RUBY_PLATFORM =~ /\blinux\b/ && `which lsof 2>/dev/null`.size >= 4 + deleted1 = `lsof -p #{pid}`.split("\n") + deleted1 = deleted1.grep(/\bREG\b.* \(deleted\)/) + deleted2 = `lsof -p #{pid2}`.split("\n") + deleted2 = deleted2.grep(/\bREG\b.* \(deleted\)/) + [ deleted1, deleted2 ].each do |ary| + ary.delete_if { |x| x =~ /\.(?:err|out) \(deleted\)/ } + end + assert_equal 0, deleted1.size, "pid1=#{deleted1.inspect}" + assert_equal 0, deleted2.size, "pid2=#{deleted2.inspect}" + end + end + buf = ''.dup + slow = Digest::MD5.new + ft = Thread.new do + fast = Digest::MD5.new + f = TCPSocket.new(host2, port2) + f.write(req) + b2 = ''.dup + check_headers(f) + nf = 0 + begin + f.readpartial(1024 * 1024, b2) + nf += b2.bytesize + fast.update(b2) + rescue EOFError + f = f.close + end while f + b2.clear + [ nf, fast.hexdigest ] + end + Thread.abort_on_exception = true + check_headers(s) + n = 0 + begin + s.readpartial(1024 * 1024, buf) + slow.update(buf) + n += buf.bytesize + sleep 0.01 + rescue EOFError + s = s.close + end while s + ft.join(5) + assert_equal [n, slow.hexdigest ], ft.value + + fast = Digest::MD5.new + f = TCPSocket.new(host, port) + f.write(req) + check_headers(f) + begin + f.readpartial(1024 * 1024, buf) + fast.update(buf) + rescue EOFError + f = f.close + end while f + buf.clear + assert_equal slow.hexdigest, fast.hexdigest + ensure + s.close if s + quit_wait(pid) + quit_wait(pid2) + end +end diff --git a/test/test_wbuf.rb b/test/test_wbuf.rb index 1c8c0d0..990ad9d 100644 --- a/test/test_wbuf.rb +++ b/test/test_wbuf.rb @@ -20,8 +20,8 @@ class TestWbuf < Testcase buf = "*" * (16384 * 2) nr = 1000 [ true, false ].each do |persist| - wbuf = Yahns::Wbuf.new([], persist, Dir.tmpdir, :wait_writable) - assert_equal :wait_writable, wbuf.busy + wbuf = Yahns::Wbuf.new([], persist, Dir.tmpdir) + assert_equal false, wbuf.busy a, b = socketpair assert_nil wbuf.wbuf_write(a, "HIHI") assert_equal "HIHI", b.read(4) @@ -71,7 +71,7 @@ class TestWbuf < Testcase break end while true end - wbuf = Yahns::Wbuf.new([], true, Dir.tmpdir, :wait_writable) + wbuf = Yahns::Wbuf.new([], true, Dir.tmpdir) rv1 = wbuf.wbuf_write(a, buf) rv2 = wbuf.wbuf_flush(a) @@ -104,7 +104,7 @@ class TestWbuf < Testcase def test_wbuf_flush_close pipe = cloexec_pipe persist = true - wbuf = Yahns::Wbuf.new(pipe[0], persist, Dir.tmpdir, :wait_writable) + wbuf = Yahns::Wbuf.new(pipe[0], persist, Dir.tmpdir) refute wbuf.respond_to?(:close) # we don't want this for HttpResponse body sp = socketpair rv = nil |