Merge pull request #135 from Earlopain/port-tests

Port some tests

test/integration/apps/rack_input_class.ru

@@ -0,0 +1,6 @@
+# frozen_string_literal: true
+
+app = lambda do |env|
+  [ 200, {}, [env["rack.input"].class.name] ]
+end
+run app

test/integration/test_configuration.rb

@@ -140,7 +140,7 @@ def test_listen_queues
     addr, port = unused_port
 
     pid = spawn_server(app: File.join(ROOT, "test/integration/env.ru"), config: <<~CONFIG)
-      listen "#{addr}:#{port}", queues: 2, workers_per_queue: 1
+      listen "#{addr}:#{port}", queues: 2, queues_per_worker: 1
       worker_processes 2
     CONFIG
 
@@ -153,4 +153,24 @@ def test_listen_queues
 
     assert_clean_shutdown(pid)
   end
+
+  def test_modify_internals
+    addr, port = unused_port
+
+    pid = spawn_server(app: File.join(ROOT, "test/integration/env.ru"), config: <<~CONFIG)
+      listen "#{addr}:#{port}", queues: 1
+      worker_processes 1
+
+      HttpParser::DEFAULTS["rack.url_scheme"] = "https"
+      Configurator::DEFAULTS[:logger].progname = "[FOO]"
+    CONFIG
+
+    assert_healthy("http://#{addr}:#{port}")
+
+    env = Net::HTTP.get(URI("http://#{addr}:#{port}/"))
+    assert_match('"rack.url_scheme"=>"https"', env)
+    assert_stderr(/\[FOO\]/)
+
+    assert_clean_shutdown(pid)
+  end
 end

test/integration/test_http_basic.rb

@@ -120,4 +120,20 @@ def test_streaming_body
 
     assert_clean_shutdown(pid)
   end
+
+  def test_write_on_close
+    addr, port = unused_port
+
+    pid = spawn_server(app: File.join(ROOT, "test/integration/apps/write-on-close.ru"), config: <<~CONFIG)
+      listen "#{addr}:#{port}"
+      worker_processes 1
+    CONFIG
+
+    assert_healthy("http://#{addr}:#{port}")
+
+    response = Net::HTTP.get_response(URI("http://#{addr}:#{port}"))
+    assert_equal "Goodbye", response.body
+
+    assert_clean_shutdown(pid)
+  end
 end

test/integration/test_rewindable_input_test.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+require "integration_test_helper"
+
+class RewindableInputTest < Pitchfork::IntegrationTest
+  def test_rewindable_input_false
+    addr, port = unused_port
+
+    pid = spawn_server(app: File.join(ROOT, "test/integration/apps/rack_input_class.ru"), lint: false, config: <<~CONFIG)
+      listen "#{addr}:#{port}"
+      worker_processes 1
+
+      rewindable_input false
+    CONFIG
+
+    assert_healthy("http://#{addr}:#{port}")
+
+    http = Net::HTTP.new(addr, port)
+    assert_equal "Pitchfork::StreamInput", http.send_request("PUT", "/", "foo").body
+
+    assert_clean_shutdown(pid)
+  end
+
+  def test_rewindable_input_true
+    addr, port = unused_port
+
+    pid = spawn_server(app: File.join(ROOT, "test/integration/apps/rack_input_class.ru"), lint: false, config: <<~CONFIG)
+      listen "#{addr}:#{port}"
+      worker_processes 1
+
+      rewindable_input true
+    CONFIG
+
+    assert_healthy("http://#{addr}:#{port}")
+
+    http = Net::HTTP.new(addr, port)
+    assert_equal "Pitchfork::TeeInput", http.send_request("PUT", "/", "foo").body
+
+    assert_clean_shutdown(pid)
+  end
+end