mirror of
https://github.com/jnunemaker/httparty
synced 2023-03-27 23:23:07 -04:00
e5a6edde06
`Net::HTTPReponse#new` will pass through whatever you give it as the `code`, although a real `Net::HTTPResponse` status code is always a string. http://docs.ruby-lang.org/en/2.0.0/Net/HTTPResponse.html#attribute-i-code.
49 lines
1.7 KiB
Ruby
49 lines
1.7 KiB
Ruby
module HTTParty
|
|
module StubResponse
|
|
def stub_http_response_with(filename)
|
|
format = filename.split('.').last.intern
|
|
data = file_fixture(filename)
|
|
|
|
response = Net::HTTPOK.new("1.1", 200, "Content for you")
|
|
allow(response).to receive(:body).and_return(data)
|
|
|
|
http_request = HTTParty::Request.new(Net::HTTP::Get, 'http://localhost', format: format)
|
|
allow(http_request).to receive_message_chain(:http, :request).and_return(response)
|
|
|
|
expect(HTTParty::Request).to receive(:new).and_return(http_request)
|
|
end
|
|
|
|
def stub_chunked_http_response_with(chunks, options = {format: "html"})
|
|
response = Net::HTTPResponse.new("1.1", 200, nil)
|
|
allow(response).to receive(:chunked_data).and_return(chunks)
|
|
def response.read_body(&block)
|
|
@body || chunked_data.each(&block)
|
|
end
|
|
|
|
http_request = HTTParty::Request.new(Net::HTTP::Get, 'http://localhost', options)
|
|
allow(http_request).to receive_message_chain(:http, :request).and_yield(response).and_return(response)
|
|
|
|
expect(HTTParty::Request).to receive(:new).and_return(http_request)
|
|
end
|
|
|
|
def stub_response(body, code = '200')
|
|
code = code.to_s
|
|
@request.options[:base_uri] ||= 'http://localhost'
|
|
unless defined?(@http) && @http
|
|
@http = Net::HTTP.new('localhost', 80)
|
|
allow(@request).to receive(:http).and_return(@http)
|
|
end
|
|
|
|
# CODE_TO_OBJ currently missing 308
|
|
if code == '308'
|
|
response = Net::HTTPRedirection.new("1.1", code, body)
|
|
else
|
|
response = Net::HTTPResponse::CODE_TO_OBJ[code].new("1.1", code, body)
|
|
end
|
|
allow(response).to receive(:body).and_return(body)
|
|
|
|
allow(@http).to receive(:request).and_return(response)
|
|
response
|
|
end
|
|
end
|
|
end
|