ruby/test/rubygems/test_gem_request_connection...

151 строка
3.4 KiB
Ruby

# frozen_string_literal: true
require_relative "helper"
require "rubygems/request"
require "timeout"
class TestGemRequestConnectionPool < Gem::TestCase
class FakeHttp
def initialize(*args)
end
def start
end
end
def setup
super
@old_client = Gem::Request::ConnectionPools.client
Gem::Request::ConnectionPools.client = FakeHttp
@proxy = URI "http://proxy.example"
end
def teardown
Gem::Request::ConnectionPools.client = @old_client
super
end
def test_to_proxy_substring
pools = Gem::Request::ConnectionPools.new nil, []
env_no_proxy = %w[
ems.example
]
no_proxy = pools.send :no_proxy?, "rubygems.example", env_no_proxy
refute no_proxy, "mismatch"
end
def test_to_proxy_empty_string
pools = Gem::Request::ConnectionPools.new nil, []
env_no_proxy = [""]
no_proxy = pools.send :no_proxy?, "ems.example", env_no_proxy
refute no_proxy, "mismatch"
end
def test_checkout_same_connection
uri = URI.parse("http://example/some_endpoint")
pools = Gem::Request::ConnectionPools.new nil, []
pool = pools.pool_for uri
conn = pool.checkout
pool.checkin conn
assert_equal conn, pool.checkout
end
def test_to_proxy_eh
pools = Gem::Request::ConnectionPools.new nil, []
env_no_proxy = %w[
1.no-proxy.example
2.no-proxy.example
]
no_proxy = pools.send :no_proxy?, "2.no-proxy.example", env_no_proxy
assert no_proxy, "match"
no_proxy = pools.send :no_proxy?, "proxy.example", env_no_proxy
refute no_proxy, "mismatch"
end
def test_to_proxy_eh_wildcard
pools = Gem::Request::ConnectionPools.new nil, []
env_no_proxy = %w[
.no-proxy.example
]
no_proxy = pools.send :no_proxy?, "2.no-proxy.example", env_no_proxy
assert no_proxy, "wildcard matching subdomain"
no_proxy = pools.send :no_proxy?, "no-proxy.example", env_no_proxy
assert no_proxy, "wildcard matching dotless domain"
no_proxy = pools.send :no_proxy?, "proxy.example", env_no_proxy
refute no_proxy, "wildcard mismatch"
end
def test_net_http_args
pools = Gem::Request::ConnectionPools.new nil, []
net_http_args = pools.send :net_http_args, URI("http://example"), nil
assert_equal ["example", 80], net_http_args
end
def test_net_http_args_ipv6
pools = Gem::Request::ConnectionPools.new nil, []
net_http_args = pools.send :net_http_args, URI("http://[::1]"), nil
assert_equal ["::1", 80], net_http_args
end
def test_net_http_args_proxy
pools = Gem::Request::ConnectionPools.new nil, []
net_http_args = pools.send :net_http_args, URI("http://example"), @proxy
assert_equal ["example", 80, "proxy.example", 80, nil, nil], net_http_args
end
def test_net_http_args_no_proxy
orig_no_proxy, ENV["no_proxy"] = ENV["no_proxy"], "example"
pools = Gem::Request::ConnectionPools.new nil, []
net_http_args = pools.send :net_http_args, URI("http://example"), @proxy
assert_equal ["example", 80, nil, nil], net_http_args
ensure
ENV["no_proxy"] = orig_no_proxy
end
def test_thread_waits_for_connection
uri = URI.parse("http://example/some_endpoint")
pools = Gem::Request::ConnectionPools.new nil, []
pool = pools.pool_for uri
pool.checkout
Thread.new do
assert_raise(Timeout::Error) do
Timeout.timeout(1) do
pool.checkout
end
end
end.join
end
end