Unfloyded/Legamunt
0
0
Fork 0
Code Activity

Add DomainHelpers spec support module for DNS/MX stub (#32690)

Browse source
This commit is contained in:
Matt Jankowski 2024-10-29 11:35:25 -04:00 • committed by GitHub
parent de1d8dc63a
commit df3b954720
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 47 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

44
spec/support/domain_helpers.rb Normal file
View file

@ -0,0 +1,44 @@
# frozen_string_literal: true
module DomainHelpers
def configure_mx(domain:, exchange:, ip_v4_addr: '2.3.4.5', ip_v6_addr: 'fd00::2')
resolver = instance_double(Resolv::DNS, :timeouts= => nil)
allow(resolver).to receive(:getresources)
.with(domain, Resolv::DNS::Resource::IN::MX)
.and_return([double_mx(exchange)])
allow(resolver)
.to receive(:getresources)
.with(domain, Resolv::DNS::Resource::IN::A)
.and_return([])
allow(resolver)
.to receive(:getresources)
.with(domain, Resolv::DNS::Resource::IN::AAAA)
.and_return([])
allow(resolver)
.to receive(:getresources)
.with(exchange, Resolv::DNS::Resource::IN::A)
.and_return([double_resource_v4(ip_v4_addr)])
allow(resolver)
.to receive(:getresources)
.with(exchange, Resolv::DNS::Resource::IN::AAAA)
.and_return([double_resource_v6(ip_v6_addr)])
allow(Resolv::DNS)
.to receive(:open)
.and_yield(resolver)
end
private
def double_mx(exchange)
instance_double(Resolv::DNS::Resource::MX, exchange: exchange)
end
def double_resource_v4(addr)
instance_double(Resolv::DNS::Resource::IN::A, address: addr)
end
def double_resource_v6(addr)
instance_double(Resolv::DNS::Resource::IN::AAAA, address: addr)
end
end