require "rbconfig" require 'test/unit' require 'socket' require 'openssl' require 'puma/minissl' require 'puma/server' require 'net/https' class TestPumaServer < Test::Unit::TestCase def setup @port = 0 @host = "127.0.0.1" @app = lambda { |env| [200, {}, [env['rack.url_scheme']]] } @events = Puma::Events.new STDOUT, STDERR @server = Puma::Server.new @app, @events end def teardown @server.stop(true) end def test_proper_stringio_body data = nil @server.app = proc do |env| data = env['rack.input'].read [200, {}, ["ok"]] end @server.add_tcp_listener @host, @port @server.run fifteen = "1" * 15 sock = TCPSocket.new @host, @server.connected_port sock << "PUT / HTTP/1.0\r\nContent-Length: 30\r\n\r\n#{fifteen}" sleep 0.1 # important so that the previous data is sent as a packet sock << fifteen sock.read assert_equal "#{fifteen}#{fifteen}", data end def test_puma_socket body = "HTTP/1.1 750 Upgraded to Awesome\r\nDone: Yep!\r\n" @server.app = proc do |env| io = env['puma.socket'] io.write body io.close [-1, {}, []] end @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "PUT / HTTP/1.0\r\n\r\nHello" assert_equal body, sock.read end def test_very_large_return giant = "x" * 2056610 @server.app = proc do |env| [200, {}, [giant]] end @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\n\r\n" while true line = sock.gets break if line == "\r\n" end out = sock.read assert_equal giant.bytesize, out.bytesize end def test_respect_x_forwarded_proto @server.app = proc do |env| [200, {}, [env['SERVER_PORT']]] end @server.add_tcp_listener @host, @port @server.run req = Net::HTTP::Get.new("/") req['HOST'] = "example.com" req['X_FORWARDED_PROTO'] = "https" res = Net::HTTP.start @host, @server.connected_port do |http| http.request(req) end assert_equal "443", res.body end def test_default_server_port @server.app = proc do |env| [200, {}, [env['SERVER_PORT']]] end @server.add_tcp_listener @host, @port @server.run req = Net::HTTP::Get.new("/") req['HOST'] = "example.com" res = Net::HTTP.start @host, @server.connected_port do |http| http.request(req) end assert_equal "80", res.body end def test_HEAD_has_no_body @server.app = proc { |env| [200, {"Foo" => "Bar"}, ["hello"]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "HEAD / HTTP/1.0\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nFoo: Bar\r\nContent-Length: 5\r\n\r\n", data end def test_GET_with_empty_body_has_sane_chunking @server.app = proc { |env| [200, {}, [""]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "HEAD / HTTP/1.0\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nContent-Length: 0\r\n\r\n", data end def test_GET_with_no_body_has_sane_chunking @server.app = proc { |env| [200, {}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "HEAD / HTTP/1.0\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\n\r\n", data end def test_doesnt_print_backtrace_in_production @events = Puma::Events.strings @server = Puma::Server.new @app, @events @server.app = proc { |e| raise "don't leak me bro" } @server.leak_stack_on_error = false @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\n\r\n" data = sock.read assert_not_match(/don't leak me bro/, data) assert_match(/HTTP\/1.0 500 Internal Server Error/, data) end def test_prints_custom_error @events = Puma::Events.strings re = lambda { |err| [302, {'Content-Type' => 'text', 'Location' => 'foo.html'}, ['302 found']] } @server = Puma::Server.new @app, @events, {:lowlevel_error_handler => re} @server.app = proc { |e| raise "don't leak me bro" } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\n\r\n" data = sock.read assert_match(/HTTP\/1.0 302 Found/, data) end def test_custom_http_codes_10 @server.app = proc { |env| [449, {}, [""]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 449 CUSTOM\r\nContent-Length: 0\r\n\r\n", data end def test_custom_http_codes_11 @server.app = proc { |env| [449, {}, [""]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.1\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.1 449 CUSTOM\r\nConnection: close\r\nContent-Length: 0\r\n\r\n", data end def test_HEAD_returns_content_headers @server.app = proc { |env| [200, {"Content-Type" => "application/pdf", "Content-Length" => "4242"}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "HEAD / HTTP/1.0\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nContent-Type: application/pdf\r\nContent-Length: 4242\r\n\r\n", data end def test_status_hook_fires_when_server_changes_states states = [] @events.register(:state) { |s| states << s } @server.app = proc { |env| [200, {}, [""]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "HEAD / HTTP/1.0\r\n\r\n" sock.read assert_equal [:booting, :running], states @server.stop(true) assert_equal [:booting, :running, :stop, :done], states end def test_timeout_in_data_phase @server.first_data_timeout = 2 @server.add_tcp_listener @host, @port @server.run client = TCPSocket.new @host, @server.connected_port client << "POST / HTTP/1.1\r\nHost: test.com\r\nContent-Type: text/plain\r\nContent-Length: 5\r\n\r\n" data = client.gets assert_equal "HTTP/1.1 408 Request Timeout\r\n", data end def test_http_11_keep_alive_with_body @server.app = proc { |env| [200, {"Content-Type" => "plain/text"}, ["hello"]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.1\r\nConnection: Keep-Alive\r\n\r\n" data = sock.read assert_equal "HTTP/1.1 200 OK\r\nContent-Type: plain/text\r\nContent-Length: 5\r\n\r\nhello", data end def test_http_11_close_with_body @server.app = proc { |env| [200, {"Content-Type" => "plain/text"}, ["hello"]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.1\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.1 200 OK\r\nContent-Type: plain/text\r\nConnection: close\r\nContent-Length: 5\r\n\r\nhello", data end def test_http_11_keep_alive_without_body @server.app = proc { |env| [204, {}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.1\r\nConnection: Keep-Alive\r\n\r\n" data = sock.read assert_equal "HTTP/1.1 204 No Content\r\n\r\n", data end def test_http_11_close_without_body @server.app = proc { |env| [204, {}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.1\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.1 204 No Content\r\nConnection: close\r\n\r\n", data end def test_http_10_keep_alive_with_body @server.app = proc { |env| [200, {"Content-Type" => "plain/text"}, ["hello"]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\nConnection: Keep-Alive\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nContent-Type: plain/text\r\nConnection: Keep-Alive\r\nContent-Length: 5\r\n\r\nhello", data end def test_http_10_close_with_body @server.app = proc { |env| [200, {"Content-Type" => "plain/text"}, ["hello"]] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nContent-Type: plain/text\r\nContent-Length: 5\r\n\r\nhello", data end def test_http_10_partial_hijack_with_content_length body_parts = ['abc', 'de'] @server.app = proc do |env| hijack_lambda = proc do | io | io.write(body_parts[0]) io.write(body_parts[1]) io.close end [200, {"Content-Length" => "5", 'rack.hijack' => hijack_lambda}, nil] end @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 200 OK\r\nContent-Length: 5\r\n\r\nabcde", data end def test_http_10_keep_alive_without_body @server.app = proc { |env| [204, {}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\nConnection: Keep-Alive\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 204 No Content\r\nConnection: Keep-Alive\r\n\r\n", data end def test_http_10_close_without_body @server.app = proc { |env| [204, {}, []] } @server.add_tcp_listener @host, @port @server.run sock = TCPSocket.new @host, @server.connected_port sock << "GET / HTTP/1.0\r\nConnection: close\r\n\r\n" data = sock.read assert_equal "HTTP/1.0 204 No Content\r\n\r\n", data end end