puma 0.8.2-java

data/test/test_persistent.rb

@@ -0,0 +1,205 @@
+require 'puma'
+require 'test/unit'
+require 'timeout'
+
+class TestPersistent < Test::Unit::TestCase
+  def setup
+    @valid_request = "GET / HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\n\r\n"
+    @close_request = "GET / HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\nConnection: close\r\n\r\n"
+    @http10_request = "GET / HTTP/1.0\r\nHost: test.com\r\nContent-Type: text/plain\r\n\r\n"
+    @keep_request = "GET / HTTP/1.0\r\nHost: test.com\r\nContent-Type: text/plain\r\nConnection: Keep-Alive\r\n\r\n"
+
+    @valid_post = "POST / HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\nContent-Length: 5\r\n\r\nhello"
+
+    @headers = { "X-Header" => "Works" }
+    @body = ["Hello"]
+    @inputs = []
+
+    @simple = lambda do |env|
+      @inputs << env['rack.input']
+      [200, @headers, @body]
+    end
+
+    @server = Puma::Server.new @simple
+    @server.add_tcp_listener "127.0.0.1", 9988
+    @server.run
+
+    @client = TCPSocket.new "127.0.0.1", 9988
+  end
+
+  def teardown
+    @client.close
+    @server.stop(true)
+  end
+
+  def lines(count, s=@client)
+    str = ""
+    timeout(5) do
+      count.times { str << s.gets }
+    end
+    str
+  end
+
+  def test_one_with_content_length
+    @client << @valid_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_two_back_to_back
+    @client << @valid_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    @client << @valid_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_post_then_get
+    @client << @valid_post
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    @client << @valid_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_chunked
+    @body << "Chunked"
+
+    @client << @valid_request
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nTransfer-Encoding: chunked\r\n\r\n5\r\nHello\r\n7\r\nChunked\r\n0\r\n\r\n", lines(10)
+  end
+
+  def test_no_chunked_in_http10
+    @body << "Chunked"
+
+    @client << @http10_request
+
+    assert_equal "HTTP/1.0 200 OK\r\nX-Header: Works\r\nConnection: close\r\n\r\n", lines(4)
+    assert_equal "HelloChunked", @client.read
+  end
+
+  def test_hex
+    str = "This is longer and will be in hex"
+    @body << str
+
+    @client << @valid_request
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nTransfer-Encoding: chunked\r\n\r\n5\r\nHello\r\n#{str.size.to_s(16)}\r\n#{str}\r\n0\r\n\r\n", lines(10)
+
+  end
+
+  def test_client11_close
+    @client << @close_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nConnection: close\r\nContent-Length: #{sz}\r\n\r\n", lines(5)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_client10_close
+    @client << @http10_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.0 200 OK\r\nX-Header: Works\r\nConnection: close\r\nContent-Length: #{sz}\r\n\r\n", lines(5)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_one_with_keep_alive_header
+    @client << @keep_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.0 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_persistent_timeout
+    @server.persistent_timeout = 2
+    @client << @valid_request
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    sleep 3
+
+    assert_raises EOFError do
+      @client.read_nonblock(1)
+    end
+  end
+
+  def test_app_sets_content_length
+    @body = ["hello", " world"]
+    @headers['Content-Length'] = "11"
+
+    @client << @valid_request
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: 11\r\n\r\n",
+                 lines(4)
+    assert_equal "hello world", @client.read(11)
+  end
+
+  def test_allow_app_to_chunk_itself
+    @headers = {'Transfer-Encoding' => "chunked" }
+
+    @body = ["5\r\nhello\r\n0\r\n\r\n"]
+
+    @client << @valid_request
+
+    assert_equal "HTTP/1.1 200 OK\r\nTransfer-Encoding: chunked\r\n\r\n5\r\nhello\r\n0\r\n\r\n", lines(7)
+  end
+
+
+  def test_two_requests_in_one_chunk
+    @server.persistent_timeout = 3
+
+    req = @valid_request.to_s
+    req << "GET /second HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\n\r\n"
+
+    @client << req
+
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+  end
+
+  def test_second_request_not_in_first_req_body
+    @server.persistent_timeout = 3
+
+    req = @valid_request.to_s
+    req << "GET /second HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\n\r\n"
+
+    @client << req
+
+    sz = @body[0].size.to_s
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    assert_equal "HTTP/1.1 200 OK\r\nX-Header: Works\r\nContent-Length: #{sz}\r\n\r\n", lines(4)
+    assert_equal "Hello", @client.read(5)
+
+    assert_equal "", @inputs[0].string
+    assert_equal "", @inputs[1].string
+
+  end
+
+end

data/test/test_rack_handler.rb

@@ -0,0 +1,10 @@
+require 'test/unit'
+
+class TestPumaUnixSocket < Test::Unit::TestCase
+  def test_handler
+    handler = Rack::Handler.get(:puma)
+    assert_equal Rack::Handler::Puma, handler
+    handler = Rack::Handler.get('Puma')
+    assert_equal Rack::Handler::Puma, handler
+  end
+end

data/test/test_rack_server.rb

@@ -0,0 +1,122 @@
+require 'test/unit'
+require 'puma'
+require 'rack/lint'
+require 'test/testhelp'
+require 'rack/commonlogger'
+require 'puma/rack_patch'
+
+class TestRackServer < Test::Unit::TestCase
+
+  class ErrorChecker
+    def initialize(app)
+      @app = app
+      @exception = nil
+      @env = nil
+    end
+
+    attr_reader :exception, :env
+
+    def call(env)
+      begin
+        @env = env
+        return @app.call(env)
+      rescue Exception => e
+        @exception = e
+
+        [
+          500,
+          { "X-Exception" => e.message, "X-Exception-Class" => e.class.to_s },
+          ["Error detected"]
+        ]
+      end
+    end
+  end
+
+  class ServerLint < Rack::Lint
+    def call(env)
+      assert("No env given") { env }
+      check_env env
+
+      @app.call(env)
+    end
+  end
+
+  def setup
+    @valid_request = "GET / HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\n\r\n"
+    
+    @simple = lambda { |env| [200, { "X-Header" => "Works" }, ["Hello"]] }
+    @server = Puma::Server.new @simple
+    @server.add_tcp_listener "127.0.0.1", 9998
+
+    @stopped = false
+  end
+
+  def stop
+    @server.stop(true)
+    @stopped = true
+  end
+
+  def teardown
+    @server.stop(true) unless @stopped
+  end
+
+  def test_lint
+    @checker = ErrorChecker.new ServerLint.new(@simple)
+    @server.app = @checker
+
+    @server.run
+
+    hit(['http://localhost:9998/test'])
+
+    stop
+
+    if exc = @checker.exception
+      raise exc
+    end
+  end
+
+  def test_path_info
+    input = nil
+    @server.app = lambda { |env| input = env; @simple.call(env) }
+    @server.run
+
+    hit(['http://localhost:9998/test/a/b/c'])
+
+    stop
+
+    assert_equal "/test/a/b/c", input['PATH_INFO']
+  end
+
+  def test_after_reply
+    closed = false
+
+    @server.app = lambda do |env|
+      env['rack.after_reply'] << lambda { closed = true }
+      @simple.call(env)
+    end
+
+    @server.run
+
+    hit(['http://localhost:9998/test'])
+
+    stop
+
+    assert_equal true, closed
+  end
+
+  def test_common_logger
+    log = StringIO.new
+
+    logger = Rack::CommonLogger.new(@simple, log)
+
+    @server.app = logger
+
+    @server.run
+
+    hit(['http://localhost:9998/test'])
+
+    stop
+
+    assert_match %r!GET /test HTTP/1\.1!, log.string
+  end
+end

data/test/test_thread_pool.rb

@@ -0,0 +1,102 @@
+require 'test/unit'
+
+require 'puma/thread_pool'
+
+class TestThreadPool < Test::Unit::TestCase
+
+  def teardown
+    @pool.shutdown if @pool
+  end
+
+  def new_pool(min, max, &blk)
+    blk = proc { } unless blk
+    @pool = Puma::ThreadPool.new(min, max, &blk)
+  end
+
+  def pause
+    sleep 0.2
+  end
+
+  def test_append_spawns
+    saw = []
+
+    pool = new_pool(0, 1) do |work|
+      saw << work
+    end
+
+    pool << 1
+
+    pause
+
+    assert_equal [1], saw
+    assert_equal 1, pool.spawned
+  end
+
+  def test_append_queues_on_max
+    finish = false
+    pool = new_pool(0, 1) { Thread.pass until finish }
+
+    pool << 1
+    pool << 2
+    pool << 3
+
+    pause
+
+    assert_equal 2, pool.backlog
+
+    finish = true
+  end
+
+  def test_trim
+    pool = new_pool(0, 1)
+
+    pool << 1
+
+    pause
+
+    assert_equal 1, pool.spawned
+    pool.trim
+
+    pause
+    assert_equal 0, pool.spawned
+  end
+
+  def test_trim_leaves_min
+    finish = false
+    pool = new_pool(1, 2) { Thread.pass until finish }
+
+    pool << 1
+    pool << 2
+
+    finish = true
+
+    assert_equal 2, pool.spawned
+    pool.trim
+    pause
+
+    assert_equal 1, pool.spawned
+    pool.trim
+    pause
+
+    assert_equal 1, pool.spawned
+
+  end
+
+  def test_trim_doesnt_overtrim
+    finish = false
+    pool = new_pool(1, 2) { Thread.pass until finish }
+
+    pool << 1
+    pool << 2
+
+    assert_equal 2, pool.spawned
+    pool.trim
+    pool.trim
+
+    finish = true
+
+    pause
+
+    assert_equal 1, pool.spawned
+  end
+end

data/test/test_unix_socket.rb

@@ -0,0 +1,37 @@
+require 'test/unit'
+require 'puma/server'
+
+require 'socket'
+
+# UNIX sockets are not recommended on JRuby
+unless defined?(JRUBY_VERSION)
+  class TestPumaUnixSocket < Test::Unit::TestCase
+  
+    App = lambda { |env| [200, {}, ["Works"]] }
+  
+    Path = "test/puma.sock"
+  
+    def setup
+      @server = Puma::Server.new App
+      @server.add_unix_listener Path
+      @server.run
+    end
+  
+    def teardown
+      @server.stop(true)
+      File.unlink Path if File.exists? Path
+    end
+  
+    def test_server
+      sock = UNIXSocket.new Path
+  
+      sock << "GET / HTTP/1.0\r\nHost: blah.com\r\n\r\n"
+  
+      expected = "HTTP/1.0 200 OK\r\nConnection: close\r\nContent-Length: 5\r\n\r\nWorks"
+  
+      assert_equal expected, sock.read(expected.size)
+  
+      sock.close
+    end
+  end
+end