mirror of
https://github.com/rails/rails.git
synced 2022-11-09 12:12:34 -05:00
333670ceb9
Previously we'd only assign a response parser when a request came through Action Dispatch integration tests. This made calls to `parsed_body` when a TestResponse was manually instantiated — though own doing or perhaps from a framework — unintentionally blow up because no parser was set at that time. The response can lookup a parser entirely through its own ivars. Extract request encoder to its own file and assume that a viable content type is present at TestResponse instantiation. Since the default response parser is a no-op, making `parsed_body` equal to `body`, no exceptions will be thrown.
54 lines
1.4 KiB
Ruby
54 lines
1.4 KiB
Ruby
module ActionDispatch
|
|
class RequestEncoder # :nodoc:
|
|
@encoders = {}
|
|
|
|
attr_reader :response_parser
|
|
|
|
def initialize(mime_name, param_encoder, response_parser, url_encoded_form = false)
|
|
@mime = Mime[mime_name]
|
|
|
|
unless @mime
|
|
raise ArgumentError, "Can't register a request encoder for " \
|
|
"unregistered MIME Type: #{mime_name}. See `Mime::Type.register`."
|
|
end
|
|
|
|
@url_encoded_form = url_encoded_form
|
|
@path_format = ".#{@mime.symbol}" unless @url_encoded_form
|
|
@response_parser = response_parser || -> body { body }
|
|
@param_encoder = param_encoder || :"to_#{@mime.symbol}".to_proc
|
|
end
|
|
|
|
def append_format_to(path)
|
|
if @url_encoded_form
|
|
path
|
|
else
|
|
path + @path_format
|
|
end
|
|
end
|
|
|
|
def content_type
|
|
@mime.to_s
|
|
end
|
|
|
|
def encode_params(params)
|
|
@param_encoder.call(params)
|
|
end
|
|
|
|
def self.parser(content_type)
|
|
mime = Mime::Type.lookup(content_type)
|
|
encoder(mime ? mime.ref : nil).response_parser
|
|
end
|
|
|
|
def self.encoder(name)
|
|
@encoders[name] || WWWFormEncoder
|
|
end
|
|
|
|
def self.register_encoder(mime_name, param_encoder: nil, response_parser: nil)
|
|
@encoders[mime_name] = new(mime_name, param_encoder, response_parser)
|
|
end
|
|
|
|
register_encoder :json, response_parser: -> body { JSON.parse(body) }
|
|
|
|
WWWFormEncoder = new(:url_encoded_form, -> params { params }, nil, true)
|
|
end
|
|
end
|