ruby--ruby/test/resolv/test_mdns.rb

# frozen_string_literal: false
require 'test/unit'
require 'resolv'

class TestResolvMDNS < Test::Unit::TestCase
  def test_mdns_each_address
    mdns = Resolv::MDNS.new
    def mdns.each_resource(name, typeclass)
      if typeclass == Resolv::DNS::Resource::IN::A
        yield typeclass.new("127.0.0.1")
      else
        yield typeclass.new("::1")
      end
    end
    addrs = mdns.__send__(:use_ipv6?) ? ["127.0.0.1", "::1"] : ["127.0.0.1"]
    [
      ["example.com", []],
      ["foo.local", addrs],
    ].each do |name, expect|
      results = []
      mdns.each_address(name) do |result|
        results << result.to_s
      end
      assert_equal expect, results.sort, "GH-1484"
    end
  end
end