mirror of
https://github.com/HoneyryderChuck/httpx.git
synced 2025-10-26 00:00:42 -04:00
64 lines
2.6 KiB
Ruby
64 lines
2.6 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "test_helper"
|
|
|
|
class ResolverTest < Minitest::Test
|
|
include HTTPX
|
|
|
|
def test_cached_lookup
|
|
now = Process.clock_gettime(Process::CLOCK_MONOTONIC)
|
|
assert_ips nil, Resolver.cached_lookup("test.com")
|
|
dns_entry = { "data" => "::2", "TTL" => now + 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")
|
|
|
|
now = Process.clock_gettime(Process::CLOCK_MONOTONIC)
|
|
dns_entry = { "data" => "::2", "TTL" => now + 2, "name" => "test.com" }
|
|
alias_entry = { "alias" => "test.com", "TTL" => now + 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" => now + 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" => now + 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" => now + 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" => now + 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
|
|
options = Options.new
|
|
assert Resolver.resolver_for(:native, options) < Resolver::Native
|
|
assert Resolver.resolver_for(:system, options) < Resolver::System
|
|
assert Resolver.resolver_for(:https, options) < Resolver::HTTPS
|
|
ex = assert_raises(Error) { Resolver.resolver_for(:smth, options) }
|
|
assert(ex.message.include?("unsupported resolver type"))
|
|
assert Resolver.resolver_for(Resolver::HTTPS, options) == Resolver::HTTPS
|
|
|
|
return if defined?(RBS)
|
|
|
|
ex = assert_raises(Error) { Resolver.resolver_for(Object, options) }
|
|
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
|