test/multiverse/suites/rack/example_app.rb in newrelic_rpm-3.13.2.302 vs test/multiverse/suites/rack/example_app.rb in newrelic_rpm-3.14.0.305

@@ -96,5 +96,24 @@
     end
 
     [response_code, result[1], result[2]]
   end
 end
+
+class ResponseContentTypeMiddleware
+  def initialize(app)
+    @app = app
+  end
+
+  def call(env)
+    req = Rack::Request.new(env)
+
+    status, headers, body = @app.call(env)
+
+    if req.params['override-content-type']
+      content_type = req.params['override-content-type']
+      headers.update("Content-Type" => content_type)
+    end
+
+    [status, headers, body]
+  end
+end