test/resolv/test_mdns.rb

   1 # frozen_string_literal: false
   2 require 'test/unit'
   3 require 'resolv'
   4
   5 class TestResolvMDNS < Test::Unit::TestCase
   6   def test_mdns_each_address
   7     mdns = Resolv::MDNS.new
   8     def mdns.each_resource(name, typeclass)
   9       if typeclass == Resolv::DNS::Resource::IN::A
  10         yield typeclass.new("127.0.0.1")
  11       else
  12         yield typeclass.new("::1")
  13       end
  14     end
  15     addrs = mdns.__send__(:use_ipv6?) ? ["127.0.0.1", "::1"] : ["127.0.0.1"]
  16     [
  17       ["example.com", []],
  18       ["foo.local", addrs],
  19     ].each do |name, expect|
  20       results = []
  21       mdns.each_address(name) do |result|
  22         results << result.to_s
  23       end
  24       assert_equal expect, results.sort, "GH-1484"
  25     end
  26   end
  27 end