lib/capybara/server.rb
# frozen_string_literal: true require 'uri' require 'net/http' require 'rack' require 'capybara/server/middleware' require 'capybara/server/animation_disabler' require 'capybara/server/checker' module Capybara # @api private class Server class << self def ports @ports ||= {} end end attr_reader :app, :port, :host def initialize(app, *deprecated_options, port: Capybara.server_port, host: Capybara.server_host, reportable_errors: Capybara.server_errors, extra_middleware: []) warn 'Positional arguments, other than the application, to Server#new are deprecated, please use keyword arguments' unless deprecated_options.empty? @app = app @extra_middleware = extra_middleware @server_thread = nil # suppress warnings @host = deprecated_options[1] || host @reportable_errors = deprecated_options[2] || reportable_errors @port = deprecated_options[0] || port @port ||= Capybara::Server.ports[port_key] @port ||= find_available_port(host) @checker = Checker.new(@host, @port) end def reset_error! middleware.clear_error end def error middleware.error end def using_ssl? @checker.ssl? end def responsive? return false if @server_thread&.join(0) res = @checker.request { |http| http.get('/__identify__') } if res.is_a?(Net::HTTPSuccess) || res.is_a?(Net::HTTPRedirection) return res.body == app.object_id.to_s end rescue SystemCallError, Net::ReadTimeout, OpenSSL::SSL::SSLError false end def wait_for_pending_requests timer = Capybara::Helpers.timer(expire_in: 60) while pending_requests? raise 'Requests did not finish in 60 seconds' if timer.expired? sleep 0.01 end end def boot unless responsive? Capybara::Server.ports[port_key] = port @server_thread = Thread.new do Capybara.server.call(middleware, port, host) end timer = Capybara::Helpers.timer(expire_in: 60) until responsive? raise 'Rack application timed out during boot' if timer.expired? @server_thread.join(0.1) end end self end private def middleware @middleware ||= Middleware.new(app, @reportable_errors, @extra_middleware) end def port_key Capybara.reuse_server ? app.object_id : middleware.object_id end def pending_requests? middleware.pending_requests? end def find_available_port(host) server = TCPServer.new(host, 0) server.addr[1] ensure server&.close end end end