1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
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_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_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
t1 = Thread.new {
timeout(1) do
pool.checkout
end
}
assert_raises(Timeout::Error) do
t1.join
end
end
end
|
