Contents

# frozen_string_literal: true

module DnsMock
  module Response
    class Answer
      TTL = 1
      REVERSE_TYPE_MAPPER = DnsMock::AVAILABLE_DNS_RECORD_TYPES.each_with_object({}) do |record_type, hash|
        hash[::Resolv::DNS::Resource::IN.const_get(record_type.upcase)] = record_type
      end.freeze

      def initialize(records)
        @records = records
      end

      def build(hostname, record_class)
        @hostname = hostname
        record_by_type(record_class).map { |record| [hostname, DnsMock::Response::Answer::TTL, record] }
      end

      private

      attr_reader :records, :hostname

      def record_by_type(record_class)
        record_type = DnsMock::Response::Answer::REVERSE_TYPE_MAPPER[record_class]
        found_records = records.dig(hostname.to_s, record_type)
        raise DnsMock::Error::RecordNotFound.new(record_type, hostname) unless found_records
        found_records
      end
    end
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
dns_mock-0.2.1	lib/dns_mock/response/answer.rb
dns_mock-0.2.0	lib/dns_mock/response/answer.rb
dns_mock-0.1.0	lib/dns_mock/response/answer.rb