Merge pull request #50632 from skipkayhil/hm-move-chunk-test-to-actionpack

Move Transfer-Encoding chunked test to Action Pack
This commit is contained in:
Jean Boussier 2024-01-09 14:35:50 +01:00 committed by GitHub
commit 2d6b02bad6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 32 additions and 54 deletions

View File

@ -0,0 +1,32 @@
# frozen_string_literal: true
require "stringio"
require "abstract_unit"
class ChunkedTest < ActionDispatch::IntegrationTest
class ChunkedController < ApplicationController
def chunk
render json: {
raw_post: request.raw_post,
content_length: request.content_length
}
end
end
# The TestInput class prevents Rack::MockRequest from adding a Content-Length when the method `size` is defined
class TestInput < StringIO
undef_method :size
end
test "parses request raw_post correctly when request has Transfer-Encoding header without a Content-Length value" do
@app = self.class.build_app
@app.routes.draw do
post "chunked", to: ChunkedController.action(:chunk)
end
post "/chunked", params: TestInput.new("foo=bar"), headers: { "Transfer-Encoding" => "gzip, chunked;foo=bar" }
assert_equal 7, response.parsed_body["content_length"]
assert_equal "foo=bar", response.parsed_body["raw_post"]
end
end

View File

@ -1,54 +0,0 @@
# frozen_string_literal: true
require "isolation/abstract_unit"
require "stringio"
require "rack/test"
require "active_support/core_ext/module/delegation"
module RailtiesTest
class HttpRequestTest < ActiveSupport::TestCase
include ActiveSupport::Testing::Isolation
include Rack::Test::Methods
def setup
build_app
app_file "config/routes.rb", <<-RUBY
Rails.application.routes.draw do
post "posts", to: "posts#create"
end
RUBY
controller "posts", <<-RUBY
class PostsController < ApplicationController
def create
render json: {
raw_post: request.raw_post,
content_length: request.content_length
}
end
end
RUBY
end
def teardown
teardown_app
end
# The TestInput class prevents Rack::MockRequest from adding a Content-Length when the method `size` is defined
class TestInput < StringIO
undef_method :size
end
test "parses request raw_post correctly when request has Transfer-Encoding header without a Content-Length value" do
require "#{app_path}/config/environment"
header "Transfer-Encoding", "gzip, chunked;foo=bar"
post "/posts", TestInput.new("foo=bar")
json_response = JSON.parse(last_response.body)
assert_equal 7, json_response["content_length"]
assert_equal "foo=bar", json_response["raw_post"]
end
end
end