Fix bug with proxying chunked encoding

hashrocket · Feb 5, 2018 · 61ed5f4 · 61ed5f4
1 parent d407ef3
commit 61ed5f4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/lib/capybara/webmock/proxy.rb b/lib/capybara/webmock/proxy.rb
@@ -9,6 +9,11 @@ def initialize(pid)
     ensure_log_exists
   end
 
+  def call(env)
+    @streaming = true
+    super
+  end
+
   def perform_request(env)
     request = Rack::Request.new(env)
     allowed_urls = ['127.0.0.1', 'localhost', %r{(.*\.|\A)lvh.me}]
@@ -24,6 +29,12 @@ def self.remove_pid
     File.delete(PID_FILE) if File.exist?(PID_FILE)
   end
 
+  def rewrite_response(triplet)
+    status, headers, body = triplet
+    headers.delete "transfer-encoding"
+    triplet
+  end
+
   private
 
   def allowed_url?(urls, host)

diff --git a/lib/capybara/webmock/version.rb b/lib/capybara/webmock/version.rb
@@ -1,5 +1,5 @@
 module Capybara
   module Webmock
-    VERSION = "0.4.0"
+    VERSION = "0.4.1"
   end
 end