/
response.rb
64 lines (53 loc) · 1.8 KB
/
response.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
module HTTP
class Response
def to_webmock
webmock_response = ::WebMock::Response.new
webmock_response.status = [status.to_i, reason]
webmock_response.body = body.to_s
webmock_response.headers = headers.to_h
webmock_response
end
class << self
def from_webmock(request, webmock_response, request_signature = nil)
status = Status.new(webmock_response.status.first)
headers = webmock_response.headers || {}
uri = normalize_uri(request_signature && request_signature.uri)
# HTTP.rb 3.0+ uses a keyword argument to pass the encoding, but 1.x
# and 2.x use a positional argument, and 0.x don't support supplying
# the encoding.
body = if HTTP::VERSION < "1.0.0"
Body.new(Streamer.new(webmock_response.body))
elsif HTTP::VERSION < "3.0.0"
Body.new(Streamer.new(webmock_response.body), webmock_response.body.encoding)
else
Body.new(Streamer.new(webmock_response.body), encoding: webmock_response.body.encoding)
end
return new(status, "1.1", headers, body, uri) if HTTP::VERSION < "1.0.0"
# 5.0.0 had a breaking change to require request instead of uri.
if HTTP::VERSION < '5.0.0'
return new({
status: status,
version: "1.1",
headers: headers,
body: body,
uri: uri
})
end
new({
status: status,
version: "1.1",
headers: headers,
body: body,
request: request,
})
end
private
def normalize_uri(uri)
return unless uri
uri = Addressable::URI.parse uri
uri.port = nil if uri.default_port && uri.port == uri.default_port
uri
end
end
end
end