Sha256: 11890336205b8cbee03405aab0f718a65a50561abc106fd54714871655a2a2f9
Contents?: true
Size: 1.64 KB
Versions: 16
Compression:
Stored size: 1.64 KB
Contents
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] = yield(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 remove_request_stub(stub)
if not request_stubs.delete(stub)
raise "Request stub \n\n #{stub.to_s} \n\n is not registered."
end
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.dup.evaluate(request_signature)
end
end
end
Version data entries
16 entries across 16 versions & 2 rubygems