mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-07 00:05:02 -04:00
54 lines
2.2 KiB
Ruby
54 lines
2.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "test_helper"
|
|
|
|
class ResolverTest < Minitest::Test
|
|
include HTTPX
|
|
|
|
def test_cached_lookup
|
|
assert_ips nil, Resolver.cached_lookup("test.com")
|
|
dns_entry = { "data" => "::2", "TTL" => 2, "name" => "test.com" }
|
|
Resolver.cached_lookup_set("test.com", Socket::AF_INET6, [dns_entry])
|
|
assert_ips ["::2"], Resolver.cached_lookup("test.com")
|
|
sleep 2
|
|
assert_ips nil, Resolver.cached_lookup("test.com")
|
|
alias_entry = { "alias" => "test.com", "TTL" => 2, "name" => "foo.com" }
|
|
Resolver.cached_lookup_set("test.com", Socket::AF_INET6, [dns_entry])
|
|
Resolver.cached_lookup_set("foo.com", Socket::AF_INET6, [alias_entry])
|
|
assert_ips ["::2"], Resolver.cached_lookup("foo.com")
|
|
|
|
Resolver.cached_lookup_set("test.com", Socket::AF_INET6, [{ "data" => "::3", "TTL" => 2, "name" => "test.com" }])
|
|
assert_ips %w[::2 ::3], Resolver.cached_lookup("test.com")
|
|
|
|
Resolver.cached_lookup_set("test.com", Socket::AF_INET, [{ "data" => "127.0.0.2", "TTL" => 2, "name" => "test.com" }])
|
|
assert_ips %w[127.0.0.2 ::2 ::3], Resolver.cached_lookup("test.com")
|
|
|
|
Resolver.cached_lookup_set("test2.com", Socket::AF_INET6, [{ "data" => "::4", "TTL" => 2, "name" => "test3.com" }])
|
|
assert_ips %w[::4], Resolver.cached_lookup("test2.com")
|
|
assert_ips %w[::4], Resolver.cached_lookup("test3.com")
|
|
|
|
Resolver.cached_lookup_set("test2.com", Socket::AF_INET, [{ "data" => "127.0.0.3", "TTL" => 2, "name" => "test3.com" }])
|
|
assert_ips %w[127.0.0.3 ::4], Resolver.cached_lookup("test2.com")
|
|
assert_ips %w[127.0.0.3 ::4], Resolver.cached_lookup("test3.com")
|
|
end
|
|
|
|
def test_resolver_for
|
|
assert Resolver.resolver_for(:native) == Resolver::Native
|
|
assert Resolver.resolver_for(:system) == Resolver::System
|
|
assert Resolver.resolver_for(:https) == Resolver::HTTPS
|
|
assert Resolver.resolver_for(Resolver::HTTPS) == Resolver::HTTPS
|
|
ex = assert_raises(Error) { Resolver.resolver_for(Object) }
|
|
assert(ex.message.include?("unsupported resolver type"))
|
|
end
|
|
|
|
private
|
|
|
|
def assert_ips(expected, actual)
|
|
if expected.nil?
|
|
assert_nil(actual)
|
|
else
|
|
assert_equal(expected.map(&IPAddr.method(:new)), actual)
|
|
end
|
|
end
|
|
end
|