puma-simon 3.7.1

data/test/shell/t3.rb

@@ -0,0 +1,25 @@
+system "ruby -rubygems -I../../lib ../../bin/puma -p 10102 -C t3_conf.rb ../hello.ru &"
+sleep 5
+
+worker_pid_was_present = File.file? "t3-worker-2-pid"
+
+system "kill `cat t3-worker-2-pid`" # kill off a worker
+
+sleep 2
+
+worker_index_within_number_of_workers = !File.file?("t3-worker-3-pid")
+
+system "kill `cat t3-pid`" 
+
+File.unlink "t3-pid" if File.file? "t3-pid"
+File.unlink "t3-worker-0-pid" if File.file? "t3-worker-0-pid"
+File.unlink "t3-worker-1-pid" if File.file? "t3-worker-1-pid"
+File.unlink "t3-worker-2-pid" if File.file? "t3-worker-2-pid"
+File.unlink "t3-worker-3-pid" if File.file? "t3-worker-3-pid"
+
+if worker_pid_was_present and worker_index_within_number_of_workers
+  exit 0
+else
+  exit 1
+end
+

data/test/shell/t3_conf.rb

@@ -0,0 +1,5 @@
+pidfile "t3-pid"
+workers 3
+on_worker_boot do |index|
+  File.open("t3-worker-#{index}-pid", "w") { |f| f.puts Process.pid }
+end

data/test/slow.ru

@@ -0,0 +1,4 @@
+run lambda { |env|
+  30000000.times { }
+  [200, {}, ["Hello World"]]
+}

data/test/ssl_config.rb

@@ -0,0 +1,4 @@
+key =  File.expand_path "../../examples/puma/puma_keypair.pem", __FILE__
+cert = File.expand_path "../../examples/puma/cert_puma.pem", __FILE__
+
+ssl_bind "0.0.0.0", 9292, :cert => cert, :key => key

data/test/test_app_status.rb

@@ -0,0 +1,93 @@
+require "test_helper"
+
+require "puma/app/status"
+require "rack"
+
+class TestAppStatus < Minitest::Test
+  class FakeServer
+    def initialize
+      @status = :running
+      @backlog = 0
+      @running = 0
+    end
+
+    attr_reader :status
+    attr_accessor :backlog, :running
+
+    def stop
+      @status = :stop
+    end
+
+    def halt
+      @status = :halt
+    end
+
+    def stats
+      "{}"
+    end
+  end
+
+  def setup
+    @server = FakeServer.new
+    @app = Puma::App::Status.new(@server)
+    @app.auth_token = nil
+  end
+
+  def lint(uri)
+    app = Rack::Lint.new @app
+    mock_env = Rack::MockRequest.env_for uri
+    app.call mock_env
+  end
+
+  def test_bad_token
+    @app.auth_token = "abcdef"
+
+    status, _, _ = lint('/whatever')
+
+    assert_equal 403, status
+  end
+
+  def test_good_token
+    @app.auth_token = "abcdef"
+
+    status, _, _ = lint('/whatever?token=abcdef')
+
+    assert_equal 404, status
+  end
+
+  def test_unsupported
+    status, _, _ = lint('/not-real')
+
+    assert_equal 404, status
+  end
+
+  def test_stop
+    status, _ , app = lint('/stop')
+
+    assert_equal :stop, @server.status
+    assert_equal 200, status
+    assert_equal ['{ "status": "ok" }'], app.enum_for.to_a
+  end
+
+  def test_halt
+    status, _ , app = lint('/halt')
+
+    assert_equal :halt, @server.status
+    assert_equal 200, status
+    assert_equal ['{ "status": "ok" }'], app.enum_for.to_a
+  end
+
+  def test_stats
+    @server.backlog = 1
+    @server.running = 9
+    status, _ , app = lint('/stats')
+
+    assert_equal 200, status
+    assert_equal ['{}'], app.enum_for.to_a
+  end
+
+  def test_alternate_location
+    status, _ , _ = lint('__alternatE_location_/stats')
+    assert_equal 200, status
+  end
+end

data/test/test_binder.rb

@@ -0,0 +1,31 @@
+require "test_helper"
+
+require "puma/binder"
+require "puma/events"
+require "puma/puma_http11"
+
+class TestBinder < Minitest::Test
+  def setup
+    @events = Puma::Events.null
+    @binder = Puma::Binder.new(@events)
+  end
+
+  def test_localhost_addresses_dont_alter_listeners_for_tcp_addresses
+    skip_on_jruby
+
+    @binder.parse(["tcp://localhost:10001"], @events)
+
+    assert_equal [], @binder.listeners
+  end
+
+  def test_localhost_addresses_dont_alter_listeners_for_ssl_addresses
+    skip_on_jruby
+
+    key =  File.expand_path "../../examples/puma/puma_keypair.pem", __FILE__
+    cert = File.expand_path "../../examples/puma/cert_puma.pem", __FILE__
+
+    @binder.parse(["ssl://localhost:10002?key=#{key}&cert=#{cert}"], @events)
+
+    assert_equal [], @binder.listeners
+  end
+end

data/test/test_cli.rb

@@ -0,0 +1,209 @@
+require "test_helper"
+
+require "puma/cli"
+
+class TestCLI < Minitest::Test
+  def setup
+    @environment = 'production'
+    @tmp_file = Tempfile.new("puma-test")
+    @tmp_path = @tmp_file.path
+    @tmp_file.close!
+
+    @tmp_path2 = "#{@tmp_path}2"
+
+    File.unlink @tmp_path if File.exist? @tmp_path
+    File.unlink @tmp_path2 if File.exist? @tmp_path2
+
+    @wait, @ready = IO.pipe
+
+    @events = Puma::Events.strings
+    @events.on_booted { @ready << "!" }
+  end
+
+  def wait_booted
+    @wait.sysread 1
+  end
+
+  def teardown
+    File.unlink @tmp_path if File.exist? @tmp_path
+    File.unlink @tmp_path2 if File.exist? @tmp_path2
+
+    @wait.close
+    @ready.close
+  end
+
+  def test_pid_file
+    cli = Puma::CLI.new ["--pidfile", @tmp_path]
+    cli.launcher.write_pid
+
+    assert_equal File.read(@tmp_path).strip.to_i, Process.pid
+  end
+
+  def test_control_for_tcp
+    url = "tcp://127.0.0.1:9877/"
+    cli = Puma::CLI.new ["-b", "tcp://127.0.0.1:9876",
+                         "--control", url,
+                         "--control-token", "",
+                         "test/lobster.ru"], @events
+
+    t = Thread.new do
+      Thread.current.abort_on_exception = true
+      cli.run
+    end
+
+    wait_booted
+
+    s = TCPSocket.new "127.0.0.1", 9877
+    s << "GET /stats HTTP/1.0\r\n\r\n"
+    body = s.read
+    assert_equal '{ "backlog": 0, "running": 0 }', body.split("\r\n").last
+
+    cli.launcher.stop
+    t.join
+  end
+
+  unless Puma.jruby? || Puma.windows?
+  def test_control_clustered
+    url = "unix://#{@tmp_path}"
+
+    cli = Puma::CLI.new ["-b", "unix://#{@tmp_path2}",
+                         "-t", "2:2",
+                         "-w", "2",
+                         "--control", url,
+                         "--control-token", "",
+                         "test/lobster.ru"], @events
+
+    t = Thread.new { cli.run }
+    t.abort_on_exception = true
+
+    wait_booted
+
+    sleep 2
+
+    s = UNIXSocket.new @tmp_path
+    s << "GET /stats HTTP/1.0\r\n\r\n"
+    body = s.read
+
+    require 'json'
+    status = JSON.parse(body.split("\n").last)
+
+    assert_equal 2, status["workers"]
+
+    # wait until the first status ping has come through
+    sleep 6
+    s = UNIXSocket.new @tmp_path
+    s << "GET /stats HTTP/1.0\r\n\r\n"
+    body = s.read
+    assert_match(/\{ "workers": 2, "phase": 0, "booted_workers": 2, "old_workers": 0, "worker_status": \[\{ "pid": \d+, "index": 0, "phase": 0, "booted": true, "last_checkin": "[^"]+", "last_status": \{ "backlog":0, "running":2 \} \},\{ "pid": \d+, "index": 1, "phase": 0, "booted": true, "last_checkin": "[^"]+", "last_status": \{ "backlog":0, "running":2 \} \}\] \}/, body.split("\r\n").last)
+
+    cli.launcher.stop
+    t.join
+  end
+
+  def test_control
+    url = "unix://#{@tmp_path}"
+
+    cli = Puma::CLI.new ["-b", "unix://#{@tmp_path2}",
+                         "--control", url,
+                         "--control-token", "",
+                         "test/lobster.ru"], @events
+
+    t = Thread.new { cli.run }
+    t.abort_on_exception = true
+
+    wait_booted
+
+    s = UNIXSocket.new @tmp_path
+    s << "GET /stats HTTP/1.0\r\n\r\n"
+    body = s.read
+
+    assert_equal '{ "backlog": 0, "running": 0 }', body.split("\r\n").last
+
+    cli.launcher.stop
+    t.join
+  end
+
+  def test_control_stop
+    url = "unix://#{@tmp_path}"
+
+    cli = Puma::CLI.new ["-b", "unix://#{@tmp_path2}",
+                         "--control", url,
+                         "--control-token", "",
+                         "test/lobster.ru"], @events
+
+    t = Thread.new { cli.run }
+    t.abort_on_exception = true
+
+    wait_booted
+
+    s = UNIXSocket.new @tmp_path
+    s << "GET /stop HTTP/1.0\r\n\r\n"
+    body = s.read
+
+    assert_equal '{ "status": "ok" }', body.split("\r\n").last
+
+    t.join
+  end
+
+  def test_tmp_control
+    url = "tcp://127.0.0.1:8232"
+    cli = Puma::CLI.new ["--state", @tmp_path, "--control", "auto"]
+    cli.launcher.write_state
+
+    data = YAML.load File.read(@tmp_path)
+
+    assert_equal Process.pid, data["pid"]
+
+    url = data["control_url"]
+
+    m = %r!unix://(.*)!.match(url)
+
+    assert m, "'#{url}' is not a URL"
+  end
+
+  def test_state_file_callback_filtering
+    cli = Puma::CLI.new [ "--config", "test/config/state_file_testing_config.rb",
+                          "--state", @tmp_path ]
+    cli.launcher.write_state
+
+    data = YAML.load_file(@tmp_path)
+
+    keys_not_stripped = data.keys & Puma::CLI::KEYS_NOT_TO_PERSIST_IN_STATE
+    assert_empty keys_not_stripped
+  end
+
+  end # JRUBY or Windows
+
+  def test_state
+    url = "tcp://127.0.0.1:8232"
+    cli = Puma::CLI.new ["--state", @tmp_path, "--control", url]
+    cli.launcher.write_state
+
+    data = YAML.load File.read(@tmp_path)
+
+    assert_equal Process.pid, data["pid"]
+    assert_equal url, data["control_url"]
+  end
+
+  def test_load_path
+    cli = Puma::CLI.new ["--include", 'foo/bar']
+
+    assert_equal 'foo/bar', $LOAD_PATH[0]
+    $LOAD_PATH.shift
+
+    cli = Puma::CLI.new ["--include", 'foo/bar:baz/qux']
+
+    assert_equal 'foo/bar', $LOAD_PATH[0]
+    $LOAD_PATH.shift
+    assert_equal 'baz/qux', $LOAD_PATH[0]
+    $LOAD_PATH.shift
+  end
+
+  def test_environment
+    ENV.delete 'RACK_ENV'
+
+    Puma::CLI.new ["--environment", @environment]
+
+    assert_equal ENV['RACK_ENV'], @environment
+  end
+end

data/test/test_config.rb

@@ -0,0 +1,95 @@
+require "test_helper"
+
+require "puma/configuration"
+
+class TestConfigFile < Minitest::Test
+  def test_app_from_rackup
+    conf = Puma::Configuration.new do |c|
+      c.rackup "test/hello-bind.ru"
+    end
+    conf.load
+
+    conf.app
+
+    assert_equal ["tcp://127.0.0.1:9292"], conf.options[:binds]
+  end
+
+  def test_app_from_app_DSL
+    conf = Puma::Configuration.new do |c|
+      c.load "test/config/app.rb"
+    end
+    conf.load
+
+    app = conf.app
+
+    assert_equal [200, {}, ["embedded app"]], app.call({})
+  end
+
+  def test_double_bind_port
+    port = (rand(10_000) + 30_000).to_s
+    with_env("PORT" => port) do
+      conf = Puma::Configuration.new do |c|
+        c.bind "tcp://#{Puma::Configuration::DefaultTCPHost}:#{port}"
+        c.load "test/config/app.rb"
+      end
+
+      conf.load
+
+      assert_equal ["tcp://0.0.0.0:#{port}"], conf.options[:binds]
+    end
+  end
+
+  def test_lowleve_error_handler_DSL
+    conf = Puma::Configuration.new do |c|
+      c.load "test/config/app.rb"
+    end
+    conf.load
+
+    app = conf.options[:lowlevel_error_handler]
+
+    assert_equal [200, {}, ["error page"]], app.call({})
+  end
+
+  def test_allow_users_to_override_default_options
+    conf = Puma::Configuration.new(restart_cmd: 'bin/rails server')
+
+    assert_equal 'bin/rails server', conf.options[:restart_cmd]
+  end
+
+  def test_overwrite_options
+    conf = Puma::Configuration.new do |c|
+      c.workers 3
+    end
+    conf.load
+
+    assert_equal conf.options[:workers], 3
+    conf.options[:workers] += 1
+    assert_equal conf.options[:workers], 4
+  end
+
+  def test_parameters_overwrite_files
+    conf = Puma::Configuration.new(config_files: ['test/config/settings.rb']) do |c|
+      c.port 3030
+    end
+    conf.load
+
+    assert_match(/:3030$/, conf.options[:binds].first)
+    assert_equal 3, conf.options[:min_threads]
+    assert_equal 5, conf.options[:max_threads]
+  end
+
+  private
+
+    def with_env(env = {})
+      original_env = {}
+      env.each do |k, v|
+        original_env[k] = ENV[k]
+        ENV[k] = v
+      end
+      yield
+    ensure
+      original_env.each do |k, v|
+        ENV[k] = v
+      end
+    end
+end