Module: AssertZone

Defined in:
lib/assert_zone.rb,
lib/assert_zone/version.rb

Constant Summary collapse

VERSION = 
"0.3.0"

Class Method Summary collapse

Class Method Details

.run(filename, nameserver: nil, rspec_flags: [], use_local_resolver: false, skip_soa: false, skip_ns: false, ignore_ttl: false) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/assert_zone.rb', line 7

def self.run(filename, nameserver: nil, rspec_flags: [], use_local_resolver: false, skip_soa: false, skip_ns: false, ignore_ttl: false)
  ns_config = {}
  ns_config[:nameserver] = nameserver if nameserver

  zone = DNS::Zone.load(File.read(filename))

  unless use_local_resolver
    # default to nameserver from SOA in zone
    ns_config[:nameserver] ||= zone.soa.nameserver
    # output the custom nameserver
    puts "Using nameserver: #{ns_config[:nameserver]}"
  end

  zone.records.each do |record|
    next if record.is_a?(DNS::Zone::RR::SOA) if skip_soa
    next if record.is_a?(DNS::Zone::RR::NS) if skip_ns

    fqdn = record.label == '@' ? zone.origin : "#{record.label}.#{zone.origin}".chomp('.')

    describe fqdn do
      it do
        have_correct_dns = have_dns.with_type(record.type).config(ns_config)
        have_correct_dns = have_correct_dns.and_ttl(record.ttl) unless ignore_ttl

        if record.is_a?(DNS::Zone::RR::A)
          have_correct_dns = have_correct_dns.and_address(record.address)
        elsif record.is_a?(DNS::Zone::RR::CNAME)
          have_correct_dns = have_correct_dns.and_domainname(record.domainname.chomp('.'))
        elsif record.is_a?(DNS::Zone::RR::TXT)
          have_correct_dns = have_correct_dns.and_data(record.text)
        else
          warn "WARN: Unsupported record type: #{record.type}. Value of #{record.type} record for #{fqdn} will not be tested."
        end

        is_expected.to have_correct_dns
      end
    end
  end

  RSpec::Core::Runner.run(rspec_flags)

end