diff --git a/lib/final_destination.rb b/lib/final_destination.rb index ff4ac8396cf..b1360d20734 100644 --- a/lib/final_destination.rb +++ b/lib/final_destination.rb @@ -304,7 +304,7 @@ class FinalDestination def self.lookup_ip(host) if Rails.env.test? - "0.0.0.0" + "1.1.1.1" else IPSocket::getaddress(host) end diff --git a/spec/components/final_destination_spec.rb b/spec/components/final_destination_spec.rb index 45fa30fbb85..6a30757a7c3 100644 --- a/spec/components/final_destination_spec.rb +++ b/spec/components/final_destination_spec.rb @@ -338,8 +338,17 @@ describe FinalDestination do end it "returns false for short ip" do - expect(FinalDestination.new('https://0/logo.png').is_dest_valid?).to eq(false) - expect(FinalDestination.new('https://1/logo.png').is_dest_valid?).to eq(false) + lookup = lambda do |host| + # How IPs are looked up for single digits + if host == "0" + "0.0.0.0" + elsif host == "1" + "0.0.0.1" + end + end + + expect(FinalDestination.new('https://0/logo.png', lookup_ip: lookup).is_dest_valid?).to eq(false) + expect(FinalDestination.new('https://1/logo.png', lookup_ip: lookup).is_dest_valid?).to eq(false) end it "returns false for private ipv4" do