wendell-puma 2.9.2

data/test/test_puma_server_ssl.rb

@@ -0,0 +1,137 @@
+require "rbconfig"
+require 'test/unit'
+require 'socket'
+require 'openssl'
+
+require 'puma/minissl'
+require 'puma/server'
+
+require 'net/https'
+
+class TestPumaServerSSL < Test::Unit::TestCase
+
+  def setup
+    @port = 3212
+    @host = "127.0.0.1"
+
+    @app = lambda { |env| [200, {}, [env['rack.url_scheme']]] }
+
+    @ctx = Puma::MiniSSL::Context.new
+
+    if defined?(JRUBY_VERSION)
+      @ctx.keystore =  File.expand_path "../../examples/puma/keystore.jks", __FILE__
+      @ctx.keystore_pass = 'blahblah'
+    else
+      @ctx.key =  File.expand_path "../../examples/puma/puma_keypair.pem", __FILE__
+      @ctx.cert = File.expand_path "../../examples/puma/cert_puma.pem", __FILE__
+    end
+
+    @ctx.verify_mode = Puma::MiniSSL::VERIFY_NONE
+
+    @events = Puma::Events.new STDOUT, STDERR
+    @server = Puma::Server.new @app, @events
+    @server.add_ssl_listener @host, @port, @ctx
+    @server.run
+
+    @http = Net::HTTP.new @host, @port
+    @http.use_ssl = true
+    @http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+  end
+
+  def teardown
+    @server.stop(true)
+  end
+
+  def test_url_scheme_for_https
+    body = nil
+    @http.start do
+      req = Net::HTTP::Get.new "/", {}
+
+      @http.request(req) do |rep|
+        body = rep.body
+      end
+    end
+
+    assert_equal "https", body
+  end
+
+  def test_very_large_return
+    giant = "x" * 2056610
+
+    @server.app = proc do
+      [200, {}, [giant]]
+    end
+
+    body = nil
+    @http.start do
+      req = Net::HTTP::Get.new "/"
+      @http.request(req) do |rep|
+        body = rep.body
+      end
+    end
+
+    assert_equal giant.bytesize, body.bytesize
+  end
+
+  def test_form_submit
+    body = nil
+    @http.start do
+      req = Net::HTTP::Post.new '/'
+      req.set_form_data('a' => '1', 'b' => '2')
+
+      @http.request(req) do |rep|
+        body = rep.body
+      end
+
+    end
+
+    assert_equal "https", body
+  end
+
+  if defined?(JRUBY_VERSION)
+    def test_ssl_v3_support_disabled_by_default
+      @http.ssl_version='SSLv3'
+      assert_raises(OpenSSL::SSL::SSLError) do
+        @http.start do
+          Net::HTTP::Get.new '/'
+        end
+      end
+    end
+
+    def test_enabling_ssl_v3_support
+      @server.stop(true)
+      @ctx.enable_SSLv3 = true
+      @server = Puma::Server.new @app, @events
+      @server.add_ssl_listener @host, @port, @ctx
+      @server.run
+      @http.ssl_version='SSLv3'
+
+      body = nil
+      @http.start do
+        req = Net::HTTP::Get.new "/", {}
+
+        @http.request(req) do |rep|
+          body = rep.body
+        end
+      end
+
+      assert_equal "https", body
+    end
+
+    def test_enabling_ssl_v3_support_requires_true
+      @server.stop(true)
+      @ctx.enable_SSLv3 = "truthy but not true"
+      @server = Puma::Server.new @app, @events
+      @server.add_ssl_listener @host, @port, @ctx
+      @server.run
+      @http.ssl_version='SSLv3'
+
+      assert_raises(OpenSSL::SSL::SSLError) do
+        @http.start do
+          Net::HTTP::Get.new '/'
+        end
+      end
+    end
+  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,141 @@
+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://127.0.0.1:9998/test'])
+
+    stop
+
+    if exc = @checker.exception
+      raise exc
+    end
+  end
+
+  def test_large_post_body
+    @checker = ErrorChecker.new ServerLint.new(@simple)
+    @server.app = @checker
+
+    @server.run
+
+    big = "x" * (1024 * 16)
+
+    Net::HTTP.post_form URI.parse('http://127.0.0.1:9998/test'),
+                 { "big" => big }
+
+    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://127.0.0.1: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://127.0.0.1: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://127.0.0.1:9998/test'])
+
+    stop
+
+    assert_match %r!GET /test HTTP/1\.1!, log.string
+  end
+
+end

data/test/test_tcp_rack.rb

@@ -0,0 +1,42 @@
+require "rbconfig"
+require 'test/unit'
+require 'socket'
+require 'openssl'
+
+require 'puma/minissl'
+require 'puma/server'
+
+require 'net/https'
+
+class TestTCPRack < Test::Unit::TestCase
+
+  def setup
+    @port = 3212
+    @host = "127.0.0.1"
+
+    @events = Puma::Events.new STDOUT, STDERR
+    @server = Puma::Server.new nil, @events
+  end
+
+  def teardown
+    @server.stop(true)
+  end
+
+  def test_passes_the_socket
+    @server.tcp_mode!
+
+    body = "We sell hats for a discount!\n"
+
+    @server.app = proc do |env, socket|
+      socket << body
+      socket.close
+    end
+
+    @server.add_tcp_listener @host, @port
+    @server.run
+
+    sock = TCPSocket.new @host, @port
+
+    assert_equal body, sock.read
+  end
+end

data/test/test_thread_pool.rb

@@ -0,0 +1,156 @@
+require 'test/unit'
+
+require 'puma/thread_pool'
+
+class TestThreadPool < Test::Unit::TestCase
+
+  def teardown
+    @pool.shutdown if @pool
+  end
+
+  def new_pool(min, max, &block)
+    block = proc { } unless block
+    @pool = Puma::ThreadPool.new(min, max, &block)
+  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_converts_pool_sizes
+    pool = new_pool('0', '1')
+
+    assert_equal 0, pool.spawned
+
+    pool << 1
+
+    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
+
+    pause
+
+    assert_equal 2, pool.spawned
+    pool.trim
+    pause
+
+    assert_equal 1, pool.spawned
+    pool.trim
+    pause
+
+    assert_equal 1, pool.spawned
+
+  end
+
+  def test_force_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 true
+    pool.trim true
+
+    finish = true
+
+    pause
+
+    assert_equal 1, pool.spawned
+  end
+
+  def test_trim_is_ignored_if_no_waiting_threads
+    finish = false
+    pool = new_pool(1, 2) { Thread.pass until finish }
+
+    pool << 1
+    pool << 2
+
+    assert_equal 2, pool.spawned
+    pool.trim
+    pool.trim
+
+    assert_equal 0, pool.trim_requested
+
+    finish = true
+
+    pause
+  end
+
+  def test_autotrim
+    finish = false
+    pool = new_pool(1, 2) { Thread.pass until finish }
+
+    pool << 1
+    pool << 2
+
+    assert_equal 2, pool.spawned
+
+    finish = true
+
+    pause
+
+    assert_equal 2, pool.spawned
+
+    pool.auto_trim! 1
+
+    sleep 1
+
+    pause
+
+    assert_equal 1, pool.spawned
+  end
+end