Fix Content-Length for rack >= 2.1.0

lib/web_console/injector.rb

@@ -13,9 +13,11 @@ def initialize(body, headers)
     end
 
     def inject(content)
-      # Remove any previously set Content-Length header because we modify
-      # the body. Otherwise the response will be truncated.
-      @headers.delete("Content-Length")
+      # Set Content-Length header to the size of the current body
+      # + the extra content. Otherwise the response will be truncated.
+      if @headers["Content-Length"]
+        @headers["Content-Length"] = @body.bytesize + content.bytesize
+      end
 
       [
         if position = @body.rindex("</body>")

test/web_console/injector_test.rb

@@ -28,11 +28,11 @@ class InjectorTest < ActiveSupport::TestCase
       assert_equal [ [ "foobar" ], {} ], Injector.new(body, {}).inject("bar")
     end
 
-    test "deletes the Content-Length header" do
+    test "updates the Content-Length header" do
       body = [ "foo" ]
       headers = { "Content-Length" => 3 }
 
-      assert_equal [ [ "foobar" ], {} ], Injector.new(body, headers).inject("bar")
+      assert_equal [ [ "foobar" ], { "Content-Length" => 6 } ], Injector.new(body, headers).inject("bar")
     end
   end
 end

test/web_console/middleware_test.rb

@@ -89,6 +89,7 @@ def headers
       @app = Middleware.new(Application.new(response_content_length: 7))
 
       get "/", params: nil
+
       assert_equal(response.body.size, response.headers["Content-Length"].to_i)
     end