module AssertZone

Constants

VERSION

Public Class Methods

run(filename, nameserver: nil, rspec_flags: [], use_local_resolver: false, skip_soa: false, skip_ns: false, ignore_ttl: false) click to toggle source
# 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