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
|
# frozen_string_literal: true
module Excon
module Middleware
class Mock < Excon::Middleware::Base
def self.valid_parameter_keys
[
:allow_unstubbed_requests,
:captures,
:mock
]
end
def request_call(datum)
if datum[:mock]
# convert File/Tempfile body to string before matching:
if datum[:body].respond_to?(:read)
if datum[:body].respond_to?(:binmode)
datum[:body].binmode
end
if datum[:body].respond_to?(:rewind)
datum[:body].rewind
end
datum[:body] = datum[:body].read
elsif !datum[:body].nil? && !datum[:body].is_a?(String)
raise Excon::Errors::InvalidStub.new("Request body should be a string or an IO object. #{datum[:body].class} provided")
end
if (stub = Excon.stub_for(datum))
datum[:remote_ip] ||= '127.0.0.1'
datum[:response] = {
:body => '',
:headers => {},
:status => 200,
:remote_ip => datum[:remote_ip]
}
stub_datum = case stub.last
when Proc
stub.last.call(datum)
else
stub.last
end
datum[:response].merge!(stub_datum.reject {|key,_| key == :headers})
if stub_datum.has_key?(:headers)
datum[:response][:headers].merge!(stub_datum[:headers])
end
elsif datum[:allow_unstubbed_requests] != true
# if we reach here no stubs matched
message = StringIO.new
message.puts('no stubs matched')
Excon::PrettyPrinter.pp(message, datum)
raise(Excon::Errors::StubNotFound.new(message.string))
end
end
@stack.request_call(datum)
end
end
end
end
|
