mirror of
https://github.com/discourse/discourse.git
synced 2024-11-23 18:36:35 +08:00
38 lines
1.2 KiB
Ruby
38 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
WebMock::HttpLibAdapterRegistry.instance.register(
|
|
:final_destination,
|
|
Class.new do
|
|
OriginalHTTP = FinalDestination::HTTP unless const_defined?(:OriginalHTTP)
|
|
|
|
def self.enable!
|
|
FinalDestination.send(:remove_const, :HTTP)
|
|
|
|
# At this point, `Net::HTTP` has already been patched by WebMock so we need to re-declare `FinalDestination::HTTP`
|
|
# but inherit from the patched `Net::HTTP` class. This is to allow requests made using `FinalDestination::HTTP` to be
|
|
# intercepted by WebMock.
|
|
FinalDestination.send(
|
|
:const_set,
|
|
:HTTP,
|
|
Class.new(Net::HTTP) { include FinalDestination::SSRFSafeNetHTTP },
|
|
)
|
|
end
|
|
|
|
def self.disable!
|
|
FinalDestination.send(:remove_const, :HTTP)
|
|
FinalDestination.send(:const_set, :HTTP, OriginalHTTP)
|
|
end
|
|
end,
|
|
)
|
|
|
|
module FinalDestination::TestHelper
|
|
def self.stub_to_fail(&blk)
|
|
WebMock::HttpLibAdapterRegistry.instance.http_lib_adapters[:final_destination].disable!
|
|
FinalDestination::SSRFDetector.stubs(:lookup_ips).returns(["0.0.0.0"])
|
|
yield
|
|
ensure
|
|
WebMock::HttpLibAdapterRegistry.instance.http_lib_adapters[:final_destination].enable!
|
|
FinalDestination::SSRFDetector.unstub(:lookup_ips)
|
|
end
|
|
end
|