зеркало из https://github.com/github/ruby.git
155 строки
3.4 KiB
Ruby
155 строки
3.4 KiB
Ruby
# frozen_string_literal: true
|
|
require 'rubygems/test_case'
|
|
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_raises(Timeout::Error) do
|
|
Timeout.timeout(1) do
|
|
pool.checkout
|
|
end
|
|
end
|
|
end.join
|
|
end
|
|
|
|
end
|