85
82
('pycurl', dict(_transport=PyCurlTransport,
86
83
_server=http_server.HttpServer_PyCurl,
87
84
_url_protocol='http+pycurl',)))
88
tests.multiply_tests(t_tests, transport_scenarios, result)
90
protocol_scenarios = [
91
('HTTP/1.0', dict(_protocol_version='HTTP/1.0')),
92
('HTTP/1.1', dict(_protocol_version='HTTP/1.1')),
95
# some tests are parametrized by the protocol version only
96
p_tests, remaining_tests = tests.split_suite_by_condition(
97
remaining_tests, tests.condition_isinstance((
100
tests.multiply_tests(p_tests, protocol_scenarios, result)
102
# each implementation tested with each HTTP version
103
tp_tests, remaining_tests = tests.split_suite_by_condition(
104
remaining_tests, tests.condition_isinstance((
105
SmartHTTPTunnellingTest,
106
TestDoCatchRedirections,
108
TestHTTPRedirections,
109
TestHTTPSilentRedirections,
110
TestLimitedRangeRequestServer,
114
TestSpecificRequestHandler,
116
tp_scenarios = tests.multiply_scenarios(transport_scenarios,
118
tests.multiply_tests(tp_tests, tp_scenarios, result)
120
# proxy auth: each auth scheme on all http versions on all implementations.
121
tppa_tests, remaining_tests = tests.split_suite_by_condition(
122
remaining_tests, tests.condition_isinstance((
125
proxy_auth_scheme_scenarios = [
126
('basic', dict(_auth_server=http_utils.ProxyBasicAuthServer)),
127
('digest', dict(_auth_server=http_utils.ProxyDigestAuthServer)),
129
dict(_auth_server=http_utils.ProxyBasicAndDigestAuthServer)),
85
return transport_scenarios
88
def vary_by_http_protocol_version():
89
"""Test on http/1.0 and 1.1"""
91
('HTTP/1.0', dict(_protocol_version='HTTP/1.0')),
92
('HTTP/1.1', dict(_protocol_version='HTTP/1.1')),
131
tppa_scenarios = tests.multiply_scenarios(tp_scenarios,
132
proxy_auth_scheme_scenarios)
133
tests.multiply_tests(tppa_tests, tppa_scenarios, result)
135
# auth: each auth scheme on all http versions on all implementations.
136
tpa_tests, remaining_tests = tests.split_suite_by_condition(
137
remaining_tests, tests.condition_isinstance((
140
auth_scheme_scenarios = [
96
def vary_by_http_auth_scheme():
141
98
('basic', dict(_auth_server=http_utils.HTTPBasicAuthServer)),
142
99
('digest', dict(_auth_server=http_utils.HTTPDigestAuthServer)),
144
dict(_auth_server=http_utils.HTTPBasicAndDigestAuthServer)),
146
tpa_scenarios = tests.multiply_scenarios(tp_scenarios,
147
auth_scheme_scenarios)
148
tests.multiply_tests(tpa_tests, tpa_scenarios, result)
150
# activity: on all http[s] versions on all implementations
151
tpact_tests, remaining_tests = tests.split_suite_by_condition(
152
remaining_tests, tests.condition_isinstance((
101
dict(_auth_server=http_utils.HTTPBasicAndDigestAuthServer)),
103
# Add some attributes common to all scenarios
104
for scenario_id, scenario_dict in scenarios:
105
scenario_dict.update(_auth_header='Authorization',
106
_username_prompt_prefix='',
107
_password_prompt_prefix='')
111
def vary_by_http_proxy_auth_scheme():
113
('proxy-basic', dict(_auth_server=http_utils.ProxyBasicAuthServer)),
114
('proxy-digest', dict(_auth_server=http_utils.ProxyDigestAuthServer)),
115
('proxy-basicdigest',
116
dict(_auth_server=http_utils.ProxyBasicAndDigestAuthServer)),
118
# Add some attributes common to all scenarios
119
for scenario_id, scenario_dict in scenarios:
120
scenario_dict.update(_auth_header='Proxy-Authorization',
121
_username_prompt_prefix='Proxy ',
122
_password_prompt_prefix='Proxy ')
126
def vary_by_http_activity():
155
127
activity_scenarios = [
156
128
('urllib,http', dict(_activity_server=ActivityHTTPServer,
157
_transport=_urllib.HttpTransport_urllib,)),
129
_transport=_urllib.HttpTransport_urllib,)),
159
if tests.HTTPSServerFeature.available():
131
if features.HTTPSServerFeature.available():
160
132
activity_scenarios.append(
161
133
('urllib,https', dict(_activity_server=ActivityHTTPSServer,
162
_transport=_urllib.HttpTransport_urllib,)),)
134
_transport=_urllib.HttpTransport_urllib,)),)
163
135
if features.pycurl.available():
164
136
activity_scenarios.append(
165
137
('pycurl,http', dict(_activity_server=ActivityHTTPServer,
166
_transport=PyCurlTransport,)),)
167
if tests.HTTPSServerFeature.available():
138
_transport=PyCurlTransport,)),)
139
if features.HTTPSServerFeature.available():
168
140
from bzrlib.tests import (
305
269
self.assertEqual('realm="Thou should not pass"', remainder)
272
class TestHTTPRangeParsing(tests.TestCase):
275
super(TestHTTPRangeParsing, self).setUp()
276
# We focus on range parsing here and ignore everything else
277
class RequestHandler(http_server.TestingHTTPRequestHandler):
278
def setup(self): pass
279
def handle(self): pass
280
def finish(self): pass
282
self.req_handler = RequestHandler(None, None, None)
284
def assertRanges(self, ranges, header, file_size):
285
self.assertEquals(ranges,
286
self.req_handler._parse_ranges(header, file_size))
288
def test_simple_range(self):
289
self.assertRanges([(0,2)], 'bytes=0-2', 12)
292
self.assertRanges([(8, 11)], 'bytes=-4', 12)
294
def test_tail_bigger_than_file(self):
295
self.assertRanges([(0, 11)], 'bytes=-99', 12)
297
def test_range_without_end(self):
298
self.assertRanges([(4, 11)], 'bytes=4-', 12)
300
def test_invalid_ranges(self):
301
self.assertRanges(None, 'bytes=12-22', 12)
302
self.assertRanges(None, 'bytes=1-3,12-22', 12)
303
self.assertRanges(None, 'bytes=-', 12)
308
306
class TestHTTPServer(tests.TestCase):
309
307
"""Test the HTTP servers implementations."""
401
399
class TestHttpTransportUrls(tests.TestCase):
402
400
"""Test the http urls."""
402
scenarios = vary_by_http_client_implementation()
404
404
def test_abs_url(self):
405
405
"""Construction of absolute http URLs"""
406
t = self._transport('http://bazaar-vcs.org/bzr/bzr.dev/')
406
t = self._transport('http://example.com/bzr/bzr.dev/')
407
407
eq = self.assertEqualDiff
408
eq(t.abspath('.'), 'http://bazaar-vcs.org/bzr/bzr.dev')
409
eq(t.abspath('foo/bar'), 'http://bazaar-vcs.org/bzr/bzr.dev/foo/bar')
410
eq(t.abspath('.bzr'), 'http://bazaar-vcs.org/bzr/bzr.dev/.bzr')
408
eq(t.abspath('.'), 'http://example.com/bzr/bzr.dev')
409
eq(t.abspath('foo/bar'), 'http://example.com/bzr/bzr.dev/foo/bar')
410
eq(t.abspath('.bzr'), 'http://example.com/bzr/bzr.dev/.bzr')
411
411
eq(t.abspath('.bzr/1//2/./3'),
412
'http://bazaar-vcs.org/bzr/bzr.dev/.bzr/1/2/3')
412
'http://example.com/bzr/bzr.dev/.bzr/1/2/3')
414
414
def test_invalid_http_urls(self):
415
415
"""Trap invalid construction of urls"""
416
t = self._transport('http://bazaar-vcs.org/bzr/bzr.dev/')
416
self._transport('http://example.com/bzr/bzr.dev/')
417
417
self.assertRaises(errors.InvalidURL,
419
'http://http://bazaar-vcs.org/bzr/bzr.dev/')
419
'http://http://example.com/bzr/bzr.dev/')
421
421
def test_http_root_urls(self):
422
422
"""Construction of URLs from server root"""
423
t = self._transport('http://bzr.ozlabs.org/')
423
t = self._transport('http://example.com/')
424
424
eq = self.assertEqualDiff
425
425
eq(t.abspath('.bzr/tree-version'),
426
'http://bzr.ozlabs.org/.bzr/tree-version')
426
'http://example.com/.bzr/tree-version')
428
428
def test_http_impl_urls(self):
429
429
"""There are servers which ask for particular clients to connect"""
525
530
class TestHttpTransportRegistration(tests.TestCase):
526
531
"""Test registrations of various http implementations"""
533
scenarios = vary_by_http_client_implementation()
528
535
def test_http_registered(self):
529
t = transport.get_transport('%s://foo.com/' % self._url_protocol)
536
t = transport.get_transport_from_url(
537
'%s://foo.com/' % self._url_protocol)
530
538
self.assertIsInstance(t, transport.Transport)
531
539
self.assertIsInstance(t, self._transport)
534
542
class TestPost(tests.TestCase):
544
scenarios = multiply_scenarios(
545
vary_by_http_client_implementation(),
546
vary_by_http_protocol_version(),
536
549
def test_post_body_is_received(self):
537
550
server = RecordingServer(expect_body_tail='end-of-body',
538
551
scheme=self._url_protocol)
539
552
self.start_server(server)
540
553
url = server.get_url()
541
554
# FIXME: needs a cleanup -- vila 20100611
542
http_transport = transport.get_transport(url)
555
http_transport = transport.get_transport_from_url(url)
543
556
code, response = http_transport._post('abc def end-of-body')
545
558
server.received_bytes.startswith('POST /.bzr/smart HTTP/1.'))
546
559
self.assertTrue('content-length: 19\r' in server.received_bytes.lower())
560
self.assertTrue('content-type: application/octet-stream\r'
561
in server.received_bytes.lower())
547
562
# The transport should not be assuming that the server can accept
548
563
# chunked encoding the first time it connects, because HTTP/1.1, so we
549
564
# check for the literal string.
1027
1049
self.assertEqual('single', t._range_hint)
1052
class TruncatedBeforeBoundaryRequestHandler(
1053
http_server.TestingHTTPRequestHandler):
1054
"""Truncation before a boundary, like in bug 198646"""
1056
_truncated_ranges = 1
1058
def get_multiple_ranges(self, file, file_size, ranges):
1059
self.send_response(206)
1060
self.send_header('Accept-Ranges', 'bytes')
1062
self.send_header('Content-Type',
1063
'multipart/byteranges; boundary=%s' % boundary)
1064
boundary_line = '--%s\r\n' % boundary
1065
# Calculate the Content-Length
1067
for (start, end) in ranges:
1068
content_length += len(boundary_line)
1069
content_length += self._header_line_length(
1070
'Content-type', 'application/octet-stream')
1071
content_length += self._header_line_length(
1072
'Content-Range', 'bytes %d-%d/%d' % (start, end, file_size))
1073
content_length += len('\r\n') # end headers
1074
content_length += end - start # + 1
1075
content_length += len(boundary_line)
1076
self.send_header('Content-length', content_length)
1079
# Send the multipart body
1081
for (start, end) in ranges:
1082
if cur + self._truncated_ranges >= len(ranges):
1083
# Abruptly ends the response and close the connection
1084
self.close_connection = 1
1086
self.wfile.write(boundary_line)
1087
self.send_header('Content-type', 'application/octet-stream')
1088
self.send_header('Content-Range', 'bytes %d-%d/%d'
1089
% (start, end, file_size))
1091
self.send_range_content(file, start, end - start + 1)
1094
self.wfile.write(boundary_line)
1097
class TestTruncatedBeforeBoundary(TestSpecificRequestHandler):
1098
"""Tests the case of bug 198646, disconnecting before a boundary."""
1100
_req_handler_class = TruncatedBeforeBoundaryRequestHandler
1103
super(TestTruncatedBeforeBoundary, self).setUp()
1104
self.build_tree_contents([('a', '0123456789')],)
1106
def test_readv_with_short_reads(self):
1107
server = self.get_readonly_server()
1108
t = self.get_readonly_transport()
1109
# Force separate ranges for each offset
1110
t._bytes_to_read_before_seek = 0
1111
ireadv = iter(t.readv('a', ((0, 1), (2, 1), (4, 2), (9, 1))))
1112
self.assertEqual((0, '0'), ireadv.next())
1113
self.assertEqual((2, '2'), ireadv.next())
1114
self.assertEqual((4, '45'), ireadv.next())
1115
self.assertEqual((9, '9'), ireadv.next())
1029
1118
class LimitedRangeRequestHandler(http_server.TestingHTTPRequestHandler):
1030
1119
"""Errors out when range specifiers exceed the limit"""
1095
1189
Only the urllib implementation is tested here.
1099
tests.TestCase.setUp(self)
1101
self.addCleanup(self._restore_env)
1103
def _install_env(self, env):
1104
for name, value in env.iteritems():
1105
self._old_env[name] = osutils.set_or_unset_env(name, value)
1107
def _restore_env(self):
1108
for name, value in self._old_env.iteritems():
1109
osutils.set_or_unset_env(name, value)
1111
1192
def _proxied_request(self):
1112
1193
handler = _urllib2_wrappers.ProxyHandler()
1113
request = _urllib2_wrappers.Request('GET','http://baz/buzzle')
1194
request = _urllib2_wrappers.Request('GET', 'http://baz/buzzle')
1114
1195
handler.set_proxy(request, 'http')
1198
def assertEvaluateProxyBypass(self, expected, host, no_proxy):
1199
handler = _urllib2_wrappers.ProxyHandler()
1200
self.assertEquals(expected,
1201
handler.evaluate_proxy_bypass(host, no_proxy))
1117
1203
def test_empty_user(self):
1118
self._install_env({'http_proxy': 'http://bar.com'})
1204
self.overrideEnv('http_proxy', 'http://bar.com')
1205
request = self._proxied_request()
1206
self.assertFalse(request.headers.has_key('Proxy-authorization'))
1208
def test_user_with_at(self):
1209
self.overrideEnv('http_proxy',
1210
'http://username@domain:password@proxy_host:1234')
1119
1211
request = self._proxied_request()
1120
1212
self.assertFalse(request.headers.has_key('Proxy-authorization'))
1122
1214
def test_invalid_proxy(self):
1123
1215
"""A proxy env variable without scheme"""
1124
self._install_env({'http_proxy': 'host:1234'})
1216
self.overrideEnv('http_proxy', 'host:1234')
1125
1217
self.assertRaises(errors.InvalidURL, self._proxied_request)
1219
def test_evaluate_proxy_bypass_true(self):
1220
"""The host is not proxied"""
1221
self.assertEvaluateProxyBypass(True, 'example.com', 'example.com')
1222
self.assertEvaluateProxyBypass(True, 'bzr.example.com', '*example.com')
1224
def test_evaluate_proxy_bypass_false(self):
1225
"""The host is proxied"""
1226
self.assertEvaluateProxyBypass(False, 'bzr.example.com', None)
1228
def test_evaluate_proxy_bypass_unknown(self):
1229
"""The host is not explicitly proxied"""
1230
self.assertEvaluateProxyBypass(None, 'example.com', 'not.example.com')
1231
self.assertEvaluateProxyBypass(None, 'bzr.example.com', 'example.com')
1233
def test_evaluate_proxy_bypass_empty_entries(self):
1234
"""Ignore empty entries"""
1235
self.assertEvaluateProxyBypass(None, 'example.com', '')
1236
self.assertEvaluateProxyBypass(None, 'example.com', ',')
1237
self.assertEvaluateProxyBypass(None, 'example.com', 'foo,,bar')
1128
1240
class TestProxyHttpServer(http_utils.TestCaseWithTwoWebservers):
1129
1241
"""Tests proxy server.
1153
1270
self.no_proxy_host = self.server_host_port
1154
1271
# The secondary server is the proxy
1155
1272
self.proxy_url = self.get_secondary_url()
1158
1274
def _testing_pycurl(self):
1159
1275
# TODO: This is duplicated for lots of the classes in this file
1160
1276
return (features.pycurl.available()
1161
1277
and self._transport == PyCurlTransport)
1163
def _install_env(self, env):
1164
for name, value in env.iteritems():
1165
self._old_env[name] = osutils.set_or_unset_env(name, value)
1167
def _restore_env(self):
1168
for name, value in self._old_env.iteritems():
1169
osutils.set_or_unset_env(name, value)
1171
def proxied_in_env(self, env):
1172
self._install_env(env)
1173
t = self.get_readonly_transport()
1175
self.assertEqual('proxied contents of foo\n', t.get('foo').read())
1179
def not_proxied_in_env(self, env):
1180
self._install_env(env)
1181
t = self.get_readonly_transport()
1183
self.assertEqual('contents of foo\n', t.get('foo').read())
1279
def assertProxied(self):
1280
t = self.get_readonly_transport()
1281
self.assertEqual('proxied contents of foo\n', t.get('foo').read())
1283
def assertNotProxied(self):
1284
t = self.get_readonly_transport()
1285
self.assertEqual('contents of foo\n', t.get('foo').read())
1187
1287
def test_http_proxy(self):
1188
self.proxied_in_env({'http_proxy': self.proxy_url})
1288
self.overrideEnv('http_proxy', self.proxy_url)
1289
self.assertProxied()
1190
1291
def test_HTTP_PROXY(self):
1191
1292
if self._testing_pycurl():
1194
1295
# about. Should we ?)
1195
1296
raise tests.TestNotApplicable(
1196
1297
'pycurl does not check HTTP_PROXY for security reasons')
1197
self.proxied_in_env({'HTTP_PROXY': self.proxy_url})
1298
self.overrideEnv('HTTP_PROXY', self.proxy_url)
1299
self.assertProxied()
1199
1301
def test_all_proxy(self):
1200
self.proxied_in_env({'all_proxy': self.proxy_url})
1302
self.overrideEnv('all_proxy', self.proxy_url)
1303
self.assertProxied()
1202
1305
def test_ALL_PROXY(self):
1203
self.proxied_in_env({'ALL_PROXY': self.proxy_url})
1306
self.overrideEnv('ALL_PROXY', self.proxy_url)
1307
self.assertProxied()
1205
1309
def test_http_proxy_with_no_proxy(self):
1206
self.not_proxied_in_env({'http_proxy': self.proxy_url,
1207
'no_proxy': self.no_proxy_host})
1310
self.overrideEnv('no_proxy', self.no_proxy_host)
1311
self.overrideEnv('http_proxy', self.proxy_url)
1312
self.assertNotProxied()
1209
1314
def test_HTTP_PROXY_with_NO_PROXY(self):
1210
1315
if self._testing_pycurl():
1211
1316
raise tests.TestNotApplicable(
1212
1317
'pycurl does not check HTTP_PROXY for security reasons')
1213
self.not_proxied_in_env({'HTTP_PROXY': self.proxy_url,
1214
'NO_PROXY': self.no_proxy_host})
1318
self.overrideEnv('NO_PROXY', self.no_proxy_host)
1319
self.overrideEnv('HTTP_PROXY', self.proxy_url)
1320
self.assertNotProxied()
1216
1322
def test_all_proxy_with_no_proxy(self):
1217
self.not_proxied_in_env({'all_proxy': self.proxy_url,
1218
'no_proxy': self.no_proxy_host})
1323
self.overrideEnv('no_proxy', self.no_proxy_host)
1324
self.overrideEnv('all_proxy', self.proxy_url)
1325
self.assertNotProxied()
1220
1327
def test_ALL_PROXY_with_NO_PROXY(self):
1221
self.not_proxied_in_env({'ALL_PROXY': self.proxy_url,
1222
'NO_PROXY': self.no_proxy_host})
1328
self.overrideEnv('NO_PROXY', self.no_proxy_host)
1329
self.overrideEnv('ALL_PROXY', self.proxy_url)
1330
self.assertNotProxied()
1224
1332
def test_http_proxy_without_scheme(self):
1333
self.overrideEnv('http_proxy', self.server_host_port)
1225
1334
if self._testing_pycurl():
1226
1335
# pycurl *ignores* invalid proxy env variables. If that ever change
1227
1336
# in the future, this test will fail indicating that pycurl do not
1228
1337
# ignore anymore such variables.
1229
self.not_proxied_in_env({'http_proxy': self.server_host_port})
1338
self.assertNotProxied()
1231
self.assertRaises(errors.InvalidURL,
1232
self.proxied_in_env,
1233
{'http_proxy': self.server_host_port})
1340
self.assertRaises(errors.InvalidURL, self.assertProxied)
1236
1343
class TestRanges(http_utils.TestCaseWithWebserver):
1237
1344
"""Test the Range header in GET methods."""
1346
scenarios = multiply_scenarios(
1347
vary_by_http_client_implementation(),
1348
vary_by_http_protocol_version(),
1239
1351
def setUp(self):
1240
1352
http_utils.TestCaseWithWebserver.setUp(self)
1241
1353
self.build_tree_contents([('a', '0123456789')],)
1443
1570
self.get_a, self.old_transport, redirected)
1573
def _setup_authentication_config(**kwargs):
1574
conf = config.AuthenticationConfig()
1575
conf._get_config().update({'httptest': kwargs})
1579
class TestUrllib2AuthHandler(tests.TestCaseWithTransport):
1580
"""Unit tests for glue by which urllib2 asks us for authentication"""
1582
def test_get_user_password_without_port(self):
1583
"""We cope if urllib2 doesn't tell us the port.
1585
See https://bugs.launchpad.net/bzr/+bug/654684
1589
_setup_authentication_config(scheme='http', host='localhost',
1590
user=user, password=password)
1591
handler = _urllib2_wrappers.HTTPAuthHandler()
1592
got_pass = handler.get_user_password(dict(
1599
self.assertEquals((user, password), got_pass)
1446
1602
class TestAuth(http_utils.TestCaseWithWebserver):
1447
1603
"""Test authentication scheme"""
1449
_auth_header = 'Authorization'
1450
_password_prompt_prefix = ''
1451
_username_prompt_prefix = ''
1605
scenarios = multiply_scenarios(
1606
vary_by_http_client_implementation(),
1607
vary_by_http_protocol_version(),
1608
vary_by_http_auth_scheme(),
1455
1611
def setUp(self):
1456
1612
super(TestAuth, self).setUp()
1610
1764
# Only one 'Authentication Required' error should occur
1611
1765
self.assertEqual(1, self.server.auth_required_errors)
1613
def test_user_from_auth_conf(self):
1614
if self._testing_pycurl():
1615
raise tests.TestNotApplicable(
1616
'pycurl does not support authentication.conf')
1619
self.server.add_user(user, password)
1620
# Create a minimal config file with the right password
1621
conf = config.AuthenticationConfig()
1622
conf._get_config().update(
1623
{'httptest': {'scheme': 'http', 'port': self.server.port,
1624
'user': user, 'password': password}})
1626
t = self.get_user_transport(None, None)
1627
# Issue a request to the server to connect
1628
self.assertEqual('contents of a\n', t.get('a').read())
1629
# Only one 'Authentication Required' error should occur
1630
self.assertEqual(1, self.server.auth_required_errors)
1632
1767
def test_changing_nonce(self):
1633
1768
if self._auth_server not in (http_utils.HTTPDigestAuthServer,
1634
1769
http_utils.ProxyDigestAuthServer):
1635
1770
raise tests.TestNotApplicable('HTTP/proxy auth digest only test')
1636
1771
if self._testing_pycurl():
1637
raise tests.KnownFailure(
1638
1773
'pycurl does not handle a nonce change')
1639
1774
self.server.add_user('joe', 'foo')
1640
1775
t = self.get_user_transport('joe', 'foo')
1650
1785
# initial 'who are you' and a second 'who are you' with the new nonce)
1651
1786
self.assertEqual(2, self.server.auth_required_errors)
1788
def test_user_from_auth_conf(self):
1789
if self._testing_pycurl():
1790
raise tests.TestNotApplicable(
1791
'pycurl does not support authentication.conf')
1794
self.server.add_user(user, password)
1795
_setup_authentication_config(scheme='http', port=self.server.port,
1796
user=user, password=password)
1797
t = self.get_user_transport(None, None)
1798
# Issue a request to the server to connect
1799
self.assertEqual('contents of a\n', t.get('a').read())
1800
# Only one 'Authentication Required' error should occur
1801
self.assertEqual(1, self.server.auth_required_errors)
1803
def test_no_credential_leaks_in_log(self):
1804
self.overrideAttr(debug, 'debug_flags', set(['http']))
1806
password = 'very-sensitive-password'
1807
self.server.add_user(user, password)
1808
t = self.get_user_transport(user, password)
1809
# Capture the debug calls to mutter
1812
lines = args[0] % args[1:]
1813
# Some calls output multiple lines, just split them now since we
1814
# care about a single one later.
1815
self.mutters.extend(lines.splitlines())
1816
self.overrideAttr(trace, 'mutter', mutter)
1817
# Issue a request to the server to connect
1818
self.assertEqual(True, t.has('a'))
1819
# Only one 'Authentication Required' error should occur
1820
self.assertEqual(1, self.server.auth_required_errors)
1821
# Since the authentification succeeded, there should be a corresponding
1823
sent_auth_headers = [line for line in self.mutters
1824
if line.startswith('> %s' % (self._auth_header,))]
1825
self.assertLength(1, sent_auth_headers)
1826
self.assertStartsWith(sent_auth_headers[0],
1827
'> %s: <masked>' % (self._auth_header,))
1655
1830
class TestProxyAuth(TestAuth):
1656
"""Test proxy authentication schemes."""
1658
_auth_header = 'Proxy-authorization'
1659
_password_prompt_prefix = 'Proxy '
1660
_username_prompt_prefix = 'Proxy '
1831
"""Test proxy authentication schemes.
1833
This inherits from TestAuth to tweak the setUp and filter some failing
1837
scenarios = multiply_scenarios(
1838
vary_by_http_client_implementation(),
1839
vary_by_http_protocol_version(),
1840
vary_by_http_proxy_auth_scheme(),
1662
1843
def setUp(self):
1663
1844
super(TestProxyAuth, self).setUp()
1665
self.addCleanup(self._restore_env)
1666
1845
# Override the contents to avoid false positives
1667
1846
self.build_tree_contents([('a', 'not proxied contents of a\n'),
1668
1847
('b', 'not proxied contents of b\n'),
1673
1852
def get_user_transport(self, user, password):
1674
self._install_env({'all_proxy': self.get_user_url(user, password)})
1853
self.overrideEnv('all_proxy', self.get_user_url(user, password))
1675
1854
return TestAuth.get_user_transport(self, user, password)
1677
def _install_env(self, env):
1678
for name, value in env.iteritems():
1679
self._old_env[name] = osutils.set_or_unset_env(name, value)
1681
def _restore_env(self):
1682
for name, value in self._old_env.iteritems():
1683
osutils.set_or_unset_env(name, value)
1685
1856
def test_empty_pass(self):
1686
1857
if self._testing_pycurl():
1688
1859
if pycurl.version_info()[1] < '7.16.0':
1689
raise tests.KnownFailure(
1690
1861
'pycurl < 7.16.0 does not handle empty proxy passwords')
1691
1862
super(TestProxyAuth, self).test_empty_pass()
1833
2014
r = t._redirected_to('http://www.example.com/foo',
1834
2015
'http://foo.example.com/foo/subdir')
1835
2016
self.assertIsInstance(r, type(t))
2017
self.assertEquals('http://foo.example.com/foo/subdir/',
1837
2020
def test_redirected_to_same_host_sibling_protocol(self):
1838
2021
t = self._transport('http://www.example.com/foo')
1839
2022
r = t._redirected_to('http://www.example.com/foo',
1840
2023
'https://www.example.com/foo')
1841
2024
self.assertIsInstance(r, type(t))
2025
self.assertEquals('https://www.example.com/foo/',
1843
2028
def test_redirected_to_same_host_different_protocol(self):
1844
2029
t = self._transport('http://www.example.com/foo')
1845
2030
r = t._redirected_to('http://www.example.com/foo',
1846
2031
'ftp://www.example.com/foo')
1847
2032
self.assertNotEquals(type(r), type(t))
2033
self.assertEquals('ftp://www.example.com/foo/', r.external_url())
2035
def test_redirected_to_same_host_specific_implementation(self):
2036
t = self._transport('http://www.example.com/foo')
2037
r = t._redirected_to('http://www.example.com/foo',
2038
'https+urllib://www.example.com/foo')
2039
self.assertEquals('https://www.example.com/foo/', r.external_url())
1849
2041
def test_redirected_to_different_host_same_user(self):
1850
2042
t = self._transport('http://joe@www.example.com/foo')
1851
2043
r = t._redirected_to('http://www.example.com/foo',
1852
2044
'https://foo.example.com/foo')
1853
2045
self.assertIsInstance(r, type(t))
1854
self.assertEqual(t._user, r._user)
2046
self.assertEqual(t._parsed_url.user, r._parsed_url.user)
2047
self.assertEquals('https://joe@foo.example.com/foo/', r.external_url())
1857
2050
class PredefinedRequestHandler(http_server.TestingHTTPRequestHandler):