puma--puma/test/test_launcher.rb

require_relative "helper"

require "puma/configuration"
require 'puma/events'

class TestLauncher < Minitest::Test
  def test_dependencies_and_files_to_require_after_prune_is_correctly_built_for_no_extra_deps
    skip_on :no_bundler

    deps, dirs = launcher.send(:dependencies_and_files_to_require_after_prune)

    assert_equal(1, deps.length)
    assert_match(%r{^nio4r:[\d.]+$}, deps.first)
    assert_equal(2, dirs.length)
    assert_match(%r{puma/lib$}, dirs[0]) # lib dir
    assert_match(%r{puma-#{Puma::Const::PUMA_VERSION}$}, dirs[1]) # native extension dir
    refute_match(%r{gems/rdoc-[\d.]+/lib$}, dirs[2])
  end

  def test_dependencies_and_files_to_require_after_prune_is_correctly_built_with_extra_deps
    skip_on :no_bundler
    conf = Puma::Configuration.new do |c|
      c.extra_runtime_dependencies ['rdoc']
    end

    deps, dirs = launcher(conf).send(:dependencies_and_files_to_require_after_prune)

    assert_equal(1, deps.length)
    assert_match(%r{^nio4r:[\d.]+$}, deps.first)
    assert_equal(3, dirs.length)
    assert_match(%r{puma/lib$}, dirs[0]) # lib dir
    assert_match(%r{puma-#{Puma::Const::PUMA_VERSION}$}, dirs[1]) # native extension dir
    assert_match(%r{gems/rdoc-[\d.]+/lib$}, dirs[2]) # rdoc dir
  end

  def test_extra_runtime_deps_directories_is_empty_for_no_config
    assert_equal([], launcher.send(:extra_runtime_deps_directories))
  end

  def test_extra_runtime_deps_directories_is_correctly_built
    skip_on :no_bundler
    conf = Puma::Configuration.new do |c|
      c.extra_runtime_dependencies ['rdoc']
    end
    dep_dirs = launcher(conf).send(:extra_runtime_deps_directories)

    assert_equal(1, dep_dirs.length)
    assert_match(%r{gems/rdoc-[\d.]+/lib$}, dep_dirs.first)
  end

  def test_puma_wild_location_is_an_absolute_path
    skip_on :no_bundler
    puma_wild_location = launcher.send(:puma_wild_location)

    assert_match(%r{bin/puma-wild$}, puma_wild_location)
    # assert no "/../" in path
    refute_match(%r{/\.\./}, puma_wild_location)
  end

  def test_prints_thread_traces
    launcher.thread_status do |name, _backtrace|
      assert_match "Thread: TID", name
    end
  end

  def test_pid_file
    tmp_file = Tempfile.new("puma-test")
    tmp_path = tmp_file.path
    tmp_file.close!

    conf = Puma::Configuration.new do |c|
      c.pidfile tmp_path
    end

    launcher(conf).write_state

    assert_equal File.read(tmp_path).strip.to_i, Process.pid

    File.unlink tmp_path
  end

  def test_state_permission_0640
    tmp_file = Tempfile.new("puma-test")
    tmp_path = tmp_file.path
    tmp_file.close!
    tmp_permission = 0640

    conf = Puma::Configuration.new do |c|
      c.state_path tmp_path
      c.state_permission tmp_permission
    end

    launcher(conf).write_state

    assert File.stat(tmp_path).mode.to_s(8)[-4..-1], tmp_permission
  ensure
    File.unlink tmp_path
  end

  def test_state_permission_nil
    tmp_file = Tempfile.new("puma-test")
    tmp_path = tmp_file.path
    tmp_file.close!

    conf = Puma::Configuration.new do |c|
      c.state_path tmp_path
      c.state_permission nil
    end

    launcher(conf).write_state

    assert File.exist?(tmp_path)
  ensure
    File.unlink tmp_path
  end

  def test_no_state_permission
    tmp_file = Tempfile.new("puma-test")
    tmp_path = tmp_file.path
    tmp_file.close!

    conf = Puma::Configuration.new do |c|
      c.state_path tmp_path
    end

    launcher(conf).write_state

    assert File.exist?(tmp_path)
  ensure
    File.unlink tmp_path
  end

  def test_puma_stats
    conf = Puma::Configuration.new do |c|
      c.app -> {[200, {}, ['']]}
      c.clear_binds!
    end
    launcher = launcher(conf)
    launcher.events.on_booted {launcher.stop}
    launcher.run
    Puma::Server::STAT_METHODS.each do |stat|
      assert_includes Puma.stats_hash, stat
    end
  end

  def test_puma_stats_clustered
    skip NO_FORK_MSG unless HAS_FORK

    conf = Puma::Configuration.new do |c|
      c.app -> {[200, {}, ['']]}
      c.workers 1
      c.clear_binds!
    end
    launcher = launcher(conf)
    Thread.new do
      sleep Puma::Const::WORKER_CHECK_INTERVAL + 1
      status = Puma.stats_hash[:worker_status].first[:last_status]
      Puma::Server::STAT_METHODS.each do |stat|
        assert_includes status, stat
      end
      launcher.stop
    end
    launcher.run
  end

  private

  def events
    @events ||= Puma::Events.strings
  end

  def launcher(config = Puma::Configuration.new, evts = events)
    @launcher ||= Puma::Launcher.new(config, events: evts)
  end
end