-
Notifications
You must be signed in to change notification settings - Fork 556
/
stub_registry.rb
61 lines (46 loc) · 1.48 KB
/
stub_registry.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
module WebMock
class StubRegistry
include Singleton
attr_accessor :request_stubs
def initialize
reset!
end
def global_stubs
@global_stubs ||= []
end
def reset!
self.request_stubs = []
end
def register_global_stub(&block)
# This hash contains the responses returned by the block,
# keyed by the exact request (using the object_id).
# That way, there's no race condition in case #to_return
# doesn't run immediately after stub.with.
responses = {}
stub = ::WebMock::RequestStub.new(:any, /.*/).with { |request|
responses[request.object_id] = block.call(request)
}.to_return(lambda { |request| responses.delete(request.object_id) })
global_stubs.push stub
end
def register_request_stub(stub)
request_stubs.insert(0, stub)
stub
end
def registered_request?(request_signature)
request_stub_for(request_signature)
end
def response_for_request(request_signature)
stub = request_stub_for(request_signature)
stub ? evaluate_response_for_request(stub.response, request_signature) : nil
end
private
def request_stub_for(request_signature)
(global_stubs + request_stubs).detect { |registered_request_stub|
registered_request_stub.request_pattern.matches?(request_signature)
}
end
def evaluate_response_for_request(response, request_signature)
response.evaluate(request_signature)
end
end
end