github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-05-22 16:34:31 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

DEV: avoid mocking FinalDestination (#20570)

Browse Source
This commit is contained in:
Sam
2023-03-08 12:09:18 +11:00
committed by GitHub
parent f2476d4b80
commit 3f5fa4eb09
2 changed files with 21 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/final_destination/ssrf_detector.rb
View File

@ -78,6 +78,14 @@ class FinalDestination
ips
end
def self.allow_ip_lookups_in_test!
@allow_ip_lookups_in_test = true
end
def self.disallow_ip_lookups_in_test!
@allow_ip_lookups_in_test = false
end
private
def self.ip_in_ranges?(ip, ranges)
@ -85,7 +93,7 @@ class FinalDestination
end
def self.lookup_ips(name, timeout: nil)
if Rails.env.test?
if Rails.env.test? && !@allow_ip_lookups_in_test
["1.2.3.4"]
else
FinalDestination::Resolver.lookup(name, timeout: timeout)