blob: 22ada56d0c8d15087535c78900c634f40e68eb31 [file] [log] [blame]
Brett Cannon74bfd702003-04-25 09:39:47 +00001"""Regresssion tests for urllib"""
2
Jeremy Hylton1afc1692008-06-18 20:49:58 +00003import urllib.parse
4import urllib.request
guido@google.coma119df92011-03-29 11:41:02 -07005import urllib.error
Georg Brandl24420152008-05-26 16:32:26 +00006import http.client
Barry Warsaw820c1202008-06-12 04:06:45 +00007import email.message
Jeremy Hylton66dc8c52007-08-04 03:42:26 +00008import io
Brett Cannon74bfd702003-04-25 09:39:47 +00009import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +000010from test import support
Brett Cannon74bfd702003-04-25 09:39:47 +000011import os
Senthil Kumaran2d2ea1b2011-04-14 13:16:30 +080012import sys
Georg Brandl5a650a22005-08-26 08:51:34 +000013import tempfile
Jeremy Hylton6102e292000-08-31 15:48:10 +000014
Senthil Kumaranc5c5a142012-01-14 19:09:04 +080015from base64 import b64encode
Georg Brandl2daf6ae2012-02-20 19:54:16 +010016import collections
Senthil Kumaranc5c5a142012-01-14 19:09:04 +080017
Brett Cannon74bfd702003-04-25 09:39:47 +000018def hexescape(char):
19 """Escape char as RFC 2396 specifies"""
20 hex_repr = hex(ord(char))[2:].upper()
21 if len(hex_repr) == 1:
22 hex_repr = "0%s" % hex_repr
23 return "%" + hex_repr
Jeremy Hylton6102e292000-08-31 15:48:10 +000024
Jeremy Hylton1afc1692008-06-18 20:49:58 +000025# Shortcut for testing FancyURLopener
26_urlopener = None
27def urlopen(url, data=None, proxies=None):
28 """urlopen(url [, data]) -> open file-like object"""
29 global _urlopener
30 if proxies is not None:
31 opener = urllib.request.FancyURLopener(proxies=proxies)
32 elif not _urlopener:
33 opener = urllib.request.FancyURLopener()
34 _urlopener = opener
35 else:
36 opener = _urlopener
37 if data is None:
38 return opener.open(url)
39 else:
40 return opener.open(url, data)
41
Senthil Kumarance260142011-11-01 01:35:17 +080042
43class FakeHTTPMixin(object):
44 def fakehttp(self, fakedata):
45 class FakeSocket(io.BytesIO):
46 io_refs = 1
47
Senthil Kumaranc5c5a142012-01-14 19:09:04 +080048 def sendall(self, data):
49 FakeHTTPConnection.buf = data
Senthil Kumarance260142011-11-01 01:35:17 +080050
51 def makefile(self, *args, **kwds):
52 self.io_refs += 1
53 return self
54
55 def read(self, amt=None):
56 if self.closed:
57 return b""
58 return io.BytesIO.read(self, amt)
59
60 def readline(self, length=None):
61 if self.closed:
62 return b""
63 return io.BytesIO.readline(self, length)
64
65 def close(self):
66 self.io_refs -= 1
67 if self.io_refs == 0:
68 io.BytesIO.close(self)
69
70 class FakeHTTPConnection(http.client.HTTPConnection):
Senthil Kumaranc5c5a142012-01-14 19:09:04 +080071
72 # buffer to store data for verification in urlopen tests.
73 buf = None
74
Senthil Kumarance260142011-11-01 01:35:17 +080075 def connect(self):
76 self.sock = FakeSocket(fakedata)
Senthil Kumaranc5c5a142012-01-14 19:09:04 +080077
Senthil Kumarance260142011-11-01 01:35:17 +080078 self._connection_class = http.client.HTTPConnection
79 http.client.HTTPConnection = FakeHTTPConnection
80
81 def unfakehttp(self):
82 http.client.HTTPConnection = self._connection_class
83
84
Brett Cannon74bfd702003-04-25 09:39:47 +000085class urlopen_FileTests(unittest.TestCase):
86 """Test urlopen() opening a temporary file.
Jeremy Hylton6102e292000-08-31 15:48:10 +000087
Brett Cannon74bfd702003-04-25 09:39:47 +000088 Try to test as much functionality as possible so as to cut down on reliance
Andrew M. Kuchlingf1a2f9e2004-06-29 13:07:53 +000089 on connecting to the Net for testing.
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000090
Brett Cannon74bfd702003-04-25 09:39:47 +000091 """
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000092
Brett Cannon74bfd702003-04-25 09:39:47 +000093 def setUp(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +000094 # Create a temp file to use for testing
95 self.text = bytes("test_urllib: %s\n" % self.__class__.__name__,
96 "ascii")
97 f = open(support.TESTFN, 'wb')
Brett Cannon74bfd702003-04-25 09:39:47 +000098 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +000099 f.write(self.text)
Brett Cannon74bfd702003-04-25 09:39:47 +0000100 finally:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000101 f.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000102 self.pathname = support.TESTFN
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000103 self.returned_obj = urlopen("file:%s" % self.pathname)
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +0000104
Brett Cannon74bfd702003-04-25 09:39:47 +0000105 def tearDown(self):
106 """Shut down the open object"""
107 self.returned_obj.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000108 os.remove(support.TESTFN)
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +0000109
Brett Cannon74bfd702003-04-25 09:39:47 +0000110 def test_interface(self):
111 # Make sure object returned by urlopen() has the specified methods
112 for attr in ("read", "readline", "readlines", "fileno",
Christian Heimes9bd667a2008-01-20 15:14:11 +0000113 "close", "info", "geturl", "getcode", "__iter__"):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000114 self.assertTrue(hasattr(self.returned_obj, attr),
Brett Cannon74bfd702003-04-25 09:39:47 +0000115 "object returned by urlopen() lacks %s attribute" %
116 attr)
Skip Montanaroe78b92a2001-01-20 20:22:30 +0000117
Brett Cannon74bfd702003-04-25 09:39:47 +0000118 def test_read(self):
119 self.assertEqual(self.text, self.returned_obj.read())
Skip Montanaro080c9972001-01-28 21:12:22 +0000120
Brett Cannon74bfd702003-04-25 09:39:47 +0000121 def test_readline(self):
122 self.assertEqual(self.text, self.returned_obj.readline())
Guido van Rossuma0982942007-07-10 08:30:03 +0000123 self.assertEqual(b'', self.returned_obj.readline(),
Brett Cannon74bfd702003-04-25 09:39:47 +0000124 "calling readline() after exhausting the file did not"
125 " return an empty string")
Skip Montanaro080c9972001-01-28 21:12:22 +0000126
Brett Cannon74bfd702003-04-25 09:39:47 +0000127 def test_readlines(self):
128 lines_list = self.returned_obj.readlines()
129 self.assertEqual(len(lines_list), 1,
130 "readlines() returned the wrong number of lines")
131 self.assertEqual(lines_list[0], self.text,
132 "readlines() returned improper text")
Skip Montanaro080c9972001-01-28 21:12:22 +0000133
Brett Cannon74bfd702003-04-25 09:39:47 +0000134 def test_fileno(self):
135 file_num = self.returned_obj.fileno()
Ezio Melottie9615932010-01-24 19:26:24 +0000136 self.assertIsInstance(file_num, int, "fileno() did not return an int")
Brett Cannon74bfd702003-04-25 09:39:47 +0000137 self.assertEqual(os.read(file_num, len(self.text)), self.text,
138 "Reading on the file descriptor returned by fileno() "
139 "did not return the expected text")
Skip Montanaroe78b92a2001-01-20 20:22:30 +0000140
Brett Cannon74bfd702003-04-25 09:39:47 +0000141 def test_close(self):
Senthil Kumarand91ffca2011-03-19 17:25:27 +0800142 # Test close() by calling it here and then having it be called again
Brett Cannon74bfd702003-04-25 09:39:47 +0000143 # by the tearDown() method for the test
144 self.returned_obj.close()
Skip Montanaro080c9972001-01-28 21:12:22 +0000145
Brett Cannon74bfd702003-04-25 09:39:47 +0000146 def test_info(self):
Ezio Melottie9615932010-01-24 19:26:24 +0000147 self.assertIsInstance(self.returned_obj.info(), email.message.Message)
Skip Montanaroe78b92a2001-01-20 20:22:30 +0000148
Brett Cannon74bfd702003-04-25 09:39:47 +0000149 def test_geturl(self):
150 self.assertEqual(self.returned_obj.geturl(), self.pathname)
Skip Montanaro080c9972001-01-28 21:12:22 +0000151
Christian Heimes9bd667a2008-01-20 15:14:11 +0000152 def test_getcode(self):
Florent Xicluna419e3842010-08-08 16:16:07 +0000153 self.assertIsNone(self.returned_obj.getcode())
Christian Heimes9bd667a2008-01-20 15:14:11 +0000154
Brett Cannon74bfd702003-04-25 09:39:47 +0000155 def test_iter(self):
156 # Test iterator
157 # Don't need to count number of iterations since test would fail the
158 # instant it returned anything beyond the first line from the
Raymond Hettinger038018a2011-06-26 14:29:35 +0200159 # comparison.
160 # Use the iterator in the usual implicit way to test for ticket #4608.
161 for line in self.returned_obj:
Brett Cannon74bfd702003-04-25 09:39:47 +0000162 self.assertEqual(line, self.text)
Skip Montanaro080c9972001-01-28 21:12:22 +0000163
Senthil Kumaran3800ea92012-01-21 11:52:48 +0800164 def test_relativelocalfile(self):
165 self.assertRaises(ValueError,urllib.request.urlopen,'./' + self.pathname)
166
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000167class ProxyTests(unittest.TestCase):
168
169 def setUp(self):
Walter Dörwaldb525e182009-04-26 21:39:21 +0000170 # Records changes to env vars
171 self.env = support.EnvironmentVarGuard()
Benjamin Peterson46a99002010-01-09 18:45:30 +0000172 # Delete all proxy related env vars
Antoine Pitroub3a88b52010-10-14 18:31:39 +0000173 for k in list(os.environ):
Antoine Pitrou8c8f1ac2010-10-14 18:32:54 +0000174 if 'proxy' in k.lower():
Benjamin Peterson46a99002010-01-09 18:45:30 +0000175 self.env.unset(k)
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000176
177 def tearDown(self):
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000178 # Restore all proxy related env vars
Walter Dörwaldb525e182009-04-26 21:39:21 +0000179 self.env.__exit__()
180 del self.env
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000181
182 def test_getproxies_environment_keep_no_proxies(self):
Walter Dörwaldb525e182009-04-26 21:39:21 +0000183 self.env.set('NO_PROXY', 'localhost')
184 proxies = urllib.request.getproxies_environment()
185 # getproxies_environment use lowered case truncated (no '_proxy') keys
Florent Xicluna419e3842010-08-08 16:16:07 +0000186 self.assertEqual('localhost', proxies['no'])
Senthil Kumaran89976f12011-08-06 12:27:40 +0800187 # List of no_proxies with space.
188 self.env.set('NO_PROXY', 'localhost, anotherdomain.com, newdomain.com')
189 self.assertTrue(urllib.request.proxy_bypass_environment('anotherdomain.com'))
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000190
Senthil Kumarance260142011-11-01 01:35:17 +0800191class urlopen_HttpTests(unittest.TestCase, FakeHTTPMixin):
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000192 """Test urlopen() opening a fake http connection."""
193
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000194 def check_read(self, ver):
195 self.fakehttp(b"HTTP/" + ver + b" 200 OK\r\n\r\nHello!")
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000196 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000197 fp = urlopen("http://python.org/")
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000198 self.assertEqual(fp.readline(), b"Hello!")
199 self.assertEqual(fp.readline(), b"")
Christian Heimes9bd667a2008-01-20 15:14:11 +0000200 self.assertEqual(fp.geturl(), 'http://python.org/')
201 self.assertEqual(fp.getcode(), 200)
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000202 finally:
203 self.unfakehttp()
204
Senthil Kumaran26430412011-04-13 07:01:19 +0800205 def test_url_fragment(self):
206 # Issue #11703: geturl() omits fragments in the original URL.
207 url = 'http://docs.python.org/library/urllib.html#OK'
Senthil Kumaranb17abb12011-04-13 07:22:29 +0800208 self.fakehttp(b"HTTP/1.1 200 OK\r\n\r\nHello!")
Senthil Kumaran26430412011-04-13 07:01:19 +0800209 try:
210 fp = urllib.request.urlopen(url)
211 self.assertEqual(fp.geturl(), url)
212 finally:
213 self.unfakehttp()
214
Senthil Kumarand91ffca2011-03-19 17:25:27 +0800215 def test_willclose(self):
216 self.fakehttp(b"HTTP/1.1 200 OK\r\n\r\nHello!")
Senthil Kumaranacbaa922011-03-20 05:30:16 +0800217 try:
218 resp = urlopen("http://www.python.org")
219 self.assertTrue(resp.fp.will_close)
220 finally:
221 self.unfakehttp()
Senthil Kumarand91ffca2011-03-19 17:25:27 +0800222
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000223 def test_read_0_9(self):
224 # "0.9" response accepted (but not "simple responses" without
225 # a status line)
226 self.check_read(b"0.9")
227
228 def test_read_1_0(self):
229 self.check_read(b"1.0")
230
231 def test_read_1_1(self):
232 self.check_read(b"1.1")
233
Christian Heimes57dddfb2008-01-02 18:30:52 +0000234 def test_read_bogus(self):
235 # urlopen() should raise IOError for many error codes.
236 self.fakehttp(b'''HTTP/1.1 401 Authentication Required
237Date: Wed, 02 Jan 2008 03:03:54 GMT
238Server: Apache/1.3.33 (Debian GNU/Linux) mod_ssl/2.8.22 OpenSSL/0.9.7e
239Connection: close
240Content-Type: text/html; charset=iso-8859-1
241''')
242 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000243 self.assertRaises(IOError, urlopen, "http://python.org/")
Christian Heimes57dddfb2008-01-02 18:30:52 +0000244 finally:
245 self.unfakehttp()
246
guido@google.coma119df92011-03-29 11:41:02 -0700247 def test_invalid_redirect(self):
248 # urlopen() should raise IOError for many error codes.
249 self.fakehttp(b'''HTTP/1.1 302 Found
250Date: Wed, 02 Jan 2008 03:03:54 GMT
251Server: Apache/1.3.33 (Debian GNU/Linux) mod_ssl/2.8.22 OpenSSL/0.9.7e
252Location: file://guidocomputer.athome.com:/python/license
253Connection: close
254Content-Type: text/html; charset=iso-8859-1
255''')
256 try:
257 self.assertRaises(urllib.error.HTTPError, urlopen,
258 "http://python.org/")
259 finally:
260 self.unfakehttp()
261
Guido van Rossumd8faa362007-04-27 19:54:29 +0000262 def test_empty_socket(self):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000263 # urlopen() raises IOError if the underlying socket does not send any
264 # data. (#1680230)
Christian Heimes57dddfb2008-01-02 18:30:52 +0000265 self.fakehttp(b'')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000266 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000267 self.assertRaises(IOError, urlopen, "http://something")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000268 finally:
269 self.unfakehttp()
270
Senthil Kumarande0eb242010-08-01 17:53:37 +0000271 def test_userpass_inurl(self):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000272 self.fakehttp(b"HTTP/1.0 200 OK\r\n\r\nHello!")
Senthil Kumarande0eb242010-08-01 17:53:37 +0000273 try:
274 fp = urlopen("http://user:pass@python.org/")
275 self.assertEqual(fp.readline(), b"Hello!")
276 self.assertEqual(fp.readline(), b"")
277 self.assertEqual(fp.geturl(), 'http://user:pass@python.org/')
278 self.assertEqual(fp.getcode(), 200)
279 finally:
280 self.unfakehttp()
281
Senthil Kumaranc5c5a142012-01-14 19:09:04 +0800282 def test_userpass_inurl_w_spaces(self):
283 self.fakehttp(b"HTTP/1.0 200 OK\r\n\r\nHello!")
284 try:
285 userpass = "a b:c d"
286 url = "http://{}@python.org/".format(userpass)
287 fakehttp_wrapper = http.client.HTTPConnection
288 authorization = ("Authorization: Basic %s\r\n" %
289 b64encode(userpass.encode("ASCII")).decode("ASCII"))
290 fp = urlopen(url)
291 # The authorization header must be in place
292 self.assertIn(authorization, fakehttp_wrapper.buf.decode("UTF-8"))
293 self.assertEqual(fp.readline(), b"Hello!")
294 self.assertEqual(fp.readline(), b"")
295 # the spaces are quoted in URL so no match
296 self.assertNotEqual(fp.geturl(), url)
297 self.assertEqual(fp.getcode(), 200)
298 finally:
299 self.unfakehttp()
300
Senthil Kumaran38b968b92012-03-14 13:43:53 -0700301 def test_URLopener_deprecation(self):
302 with support.check_warnings(('',DeprecationWarning)):
303 warn = urllib.request.URLopener()
304
Brett Cannon19691362003-04-29 05:08:06 +0000305class urlretrieve_FileTests(unittest.TestCase):
Brett Cannon74bfd702003-04-25 09:39:47 +0000306 """Test urllib.urlretrieve() on local files"""
Skip Montanaro080c9972001-01-28 21:12:22 +0000307
Brett Cannon19691362003-04-29 05:08:06 +0000308 def setUp(self):
Georg Brandl5a650a22005-08-26 08:51:34 +0000309 # Create a list of temporary files. Each item in the list is a file
310 # name (absolute path or relative to the current working directory).
311 # All files in this list will be deleted in the tearDown method. Note,
312 # this only helps to makes sure temporary files get deleted, but it
313 # does nothing about trying to close files that may still be open. It
314 # is the responsibility of the developer to properly close files even
315 # when exceptional conditions occur.
316 self.tempFiles = []
317
Brett Cannon19691362003-04-29 05:08:06 +0000318 # Create a temporary file.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000319 self.registerFileForCleanUp(support.TESTFN)
Guido van Rossuma0982942007-07-10 08:30:03 +0000320 self.text = b'testing urllib.urlretrieve'
Georg Brandl5a650a22005-08-26 08:51:34 +0000321 try:
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000322 FILE = open(support.TESTFN, 'wb')
Georg Brandl5a650a22005-08-26 08:51:34 +0000323 FILE.write(self.text)
324 FILE.close()
325 finally:
326 try: FILE.close()
327 except: pass
Brett Cannon19691362003-04-29 05:08:06 +0000328
329 def tearDown(self):
Georg Brandl5a650a22005-08-26 08:51:34 +0000330 # Delete the temporary files.
331 for each in self.tempFiles:
332 try: os.remove(each)
333 except: pass
334
335 def constructLocalFileUrl(self, filePath):
Victor Stinner6c6f8512010-08-07 10:09:35 +0000336 filePath = os.path.abspath(filePath)
337 try:
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000338 filePath.encode("utf-8")
Victor Stinner6c6f8512010-08-07 10:09:35 +0000339 except UnicodeEncodeError:
340 raise unittest.SkipTest("filePath is not encodable to utf8")
341 return "file://%s" % urllib.request.pathname2url(filePath)
Georg Brandl5a650a22005-08-26 08:51:34 +0000342
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000343 def createNewTempFile(self, data=b""):
Georg Brandl5a650a22005-08-26 08:51:34 +0000344 """Creates a new temporary file containing the specified data,
345 registers the file for deletion during the test fixture tear down, and
346 returns the absolute path of the file."""
347
348 newFd, newFilePath = tempfile.mkstemp()
349 try:
350 self.registerFileForCleanUp(newFilePath)
351 newFile = os.fdopen(newFd, "wb")
352 newFile.write(data)
353 newFile.close()
354 finally:
355 try: newFile.close()
356 except: pass
357 return newFilePath
358
359 def registerFileForCleanUp(self, fileName):
360 self.tempFiles.append(fileName)
Brett Cannon19691362003-04-29 05:08:06 +0000361
362 def test_basic(self):
363 # Make sure that a local file just gets its own location returned and
364 # a headers value is returned.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000365 result = urllib.request.urlretrieve("file:%s" % support.TESTFN)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000366 self.assertEqual(result[0], support.TESTFN)
Ezio Melottie9615932010-01-24 19:26:24 +0000367 self.assertIsInstance(result[1], email.message.Message,
368 "did not get a email.message.Message instance "
369 "as second returned value")
Brett Cannon19691362003-04-29 05:08:06 +0000370
371 def test_copy(self):
372 # Test that setting the filename argument works.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000373 second_temp = "%s.2" % support.TESTFN
Georg Brandl5a650a22005-08-26 08:51:34 +0000374 self.registerFileForCleanUp(second_temp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000375 result = urllib.request.urlretrieve(self.constructLocalFileUrl(
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000376 support.TESTFN), second_temp)
Brett Cannon19691362003-04-29 05:08:06 +0000377 self.assertEqual(second_temp, result[0])
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000378 self.assertTrue(os.path.exists(second_temp), "copy of the file was not "
Brett Cannon19691362003-04-29 05:08:06 +0000379 "made")
Alex Martelli01c77c62006-08-24 02:58:11 +0000380 FILE = open(second_temp, 'rb')
Brett Cannon19691362003-04-29 05:08:06 +0000381 try:
382 text = FILE.read()
Brett Cannon19691362003-04-29 05:08:06 +0000383 FILE.close()
Georg Brandl5a650a22005-08-26 08:51:34 +0000384 finally:
385 try: FILE.close()
386 except: pass
Brett Cannon19691362003-04-29 05:08:06 +0000387 self.assertEqual(self.text, text)
388
389 def test_reporthook(self):
390 # Make sure that the reporthook works.
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700391 def hooktester(block_count, block_read_size, file_size, count_holder=[0]):
392 self.assertIsInstance(block_count, int)
393 self.assertIsInstance(block_read_size, int)
394 self.assertIsInstance(file_size, int)
395 self.assertEqual(block_count, count_holder[0])
Brett Cannon19691362003-04-29 05:08:06 +0000396 count_holder[0] = count_holder[0] + 1
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000397 second_temp = "%s.2" % support.TESTFN
Georg Brandl5a650a22005-08-26 08:51:34 +0000398 self.registerFileForCleanUp(second_temp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000399 urllib.request.urlretrieve(
400 self.constructLocalFileUrl(support.TESTFN),
Georg Brandl5a650a22005-08-26 08:51:34 +0000401 second_temp, hooktester)
402
403 def test_reporthook_0_bytes(self):
404 # Test on zero length file. Should call reporthook only 1 time.
405 report = []
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700406 def hooktester(block_count, block_read_size, file_size, _report=report):
407 _report.append((block_count, block_read_size, file_size))
Georg Brandl5a650a22005-08-26 08:51:34 +0000408 srcFileName = self.createNewTempFile()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000409 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000410 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000411 self.assertEqual(len(report), 1)
412 self.assertEqual(report[0][2], 0)
413
414 def test_reporthook_5_bytes(self):
415 # Test on 5 byte file. Should call reporthook only 2 times (once when
416 # the "network connection" is established and once when the block is
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700417 # read).
Georg Brandl5a650a22005-08-26 08:51:34 +0000418 report = []
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700419 def hooktester(block_count, block_read_size, file_size, _report=report):
420 _report.append((block_count, block_read_size, file_size))
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000421 srcFileName = self.createNewTempFile(b"x" * 5)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000422 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000423 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000424 self.assertEqual(len(report), 2)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700425 self.assertEqual(report[0][1], 0)
426 self.assertEqual(report[1][1], 5)
Georg Brandl5a650a22005-08-26 08:51:34 +0000427
428 def test_reporthook_8193_bytes(self):
429 # Test on 8193 byte file. Should call reporthook only 3 times (once
430 # when the "network connection" is established, once for the next 8192
431 # bytes, and once for the last byte).
432 report = []
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700433 def hooktester(block_count, block_read_size, file_size, _report=report):
434 _report.append((block_count, block_read_size, file_size))
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000435 srcFileName = self.createNewTempFile(b"x" * 8193)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000436 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000437 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000438 self.assertEqual(len(report), 3)
Senthil Kumarane24f96a2012-03-13 19:29:33 -0700439 self.assertEqual(report[0][1], 0)
440 self.assertEqual(report[1][1], 8192)
441 self.assertEqual(report[2][1], 1)
Skip Montanaro080c9972001-01-28 21:12:22 +0000442
Senthil Kumarance260142011-11-01 01:35:17 +0800443
444class urlretrieve_HttpTests(unittest.TestCase, FakeHTTPMixin):
445 """Test urllib.urlretrieve() using fake http connections"""
446
447 def test_short_content_raises_ContentTooShortError(self):
448 self.fakehttp(b'''HTTP/1.1 200 OK
449Date: Wed, 02 Jan 2008 03:03:54 GMT
450Server: Apache/1.3.33 (Debian GNU/Linux) mod_ssl/2.8.22 OpenSSL/0.9.7e
451Connection: close
452Content-Length: 100
453Content-Type: text/html; charset=iso-8859-1
454
455FF
456''')
457
458 def _reporthook(par1, par2, par3):
459 pass
460
461 with self.assertRaises(urllib.error.ContentTooShortError):
462 try:
463 urllib.request.urlretrieve('http://example.com/',
464 reporthook=_reporthook)
465 finally:
466 self.unfakehttp()
467
468 def test_short_content_raises_ContentTooShortError_without_reporthook(self):
469 self.fakehttp(b'''HTTP/1.1 200 OK
470Date: Wed, 02 Jan 2008 03:03:54 GMT
471Server: Apache/1.3.33 (Debian GNU/Linux) mod_ssl/2.8.22 OpenSSL/0.9.7e
472Connection: close
473Content-Length: 100
474Content-Type: text/html; charset=iso-8859-1
475
476FF
477''')
478 with self.assertRaises(urllib.error.ContentTooShortError):
479 try:
480 urllib.request.urlretrieve('http://example.com/')
481 finally:
482 self.unfakehttp()
483
484
Brett Cannon74bfd702003-04-25 09:39:47 +0000485class QuotingTests(unittest.TestCase):
486 """Tests for urllib.quote() and urllib.quote_plus()
Tim Petersc2659cf2003-05-12 20:19:37 +0000487
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000488 According to RFC 2396 (Uniform Resource Identifiers), to escape a
489 character you write it as '%' + <2 character US-ASCII hex value>.
490 The Python code of ``'%' + hex(ord(<character>))[2:]`` escapes a
491 character properly. Case does not matter on the hex letters.
Brett Cannon74bfd702003-04-25 09:39:47 +0000492
493 The various character sets specified are:
Tim Petersc2659cf2003-05-12 20:19:37 +0000494
Brett Cannon74bfd702003-04-25 09:39:47 +0000495 Reserved characters : ";/?:@&=+$,"
496 Have special meaning in URIs and must be escaped if not being used for
497 their special meaning
498 Data characters : letters, digits, and "-_.!~*'()"
499 Unreserved and do not need to be escaped; can be, though, if desired
500 Control characters : 0x00 - 0x1F, 0x7F
501 Have no use in URIs so must be escaped
502 space : 0x20
503 Must be escaped
504 Delimiters : '<>#%"'
505 Must be escaped
506 Unwise : "{}|\^[]`"
507 Must be escaped
Tim Petersc2659cf2003-05-12 20:19:37 +0000508
Brett Cannon74bfd702003-04-25 09:39:47 +0000509 """
510
511 def test_never_quote(self):
512 # Make sure quote() does not quote letters, digits, and "_,.-"
513 do_not_quote = '' .join(["ABCDEFGHIJKLMNOPQRSTUVWXYZ",
514 "abcdefghijklmnopqrstuvwxyz",
515 "0123456789",
516 "_.-"])
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000517 result = urllib.parse.quote(do_not_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000518 self.assertEqual(do_not_quote, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000519 "using quote(): %r != %r" % (do_not_quote, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000520 result = urllib.parse.quote_plus(do_not_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000521 self.assertEqual(do_not_quote, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000522 "using quote_plus(): %r != %r" % (do_not_quote, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000523
524 def test_default_safe(self):
525 # Test '/' is default value for 'safe' parameter
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000526 self.assertEqual(urllib.parse.quote.__defaults__[0], '/')
Brett Cannon74bfd702003-04-25 09:39:47 +0000527
528 def test_safe(self):
529 # Test setting 'safe' parameter does what it should do
530 quote_by_default = "<>"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000531 result = urllib.parse.quote(quote_by_default, safe=quote_by_default)
Brett Cannon74bfd702003-04-25 09:39:47 +0000532 self.assertEqual(quote_by_default, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000533 "using quote(): %r != %r" % (quote_by_default, result))
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000534 result = urllib.parse.quote_plus(quote_by_default,
535 safe=quote_by_default)
Brett Cannon74bfd702003-04-25 09:39:47 +0000536 self.assertEqual(quote_by_default, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000537 "using quote_plus(): %r != %r" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000538 (quote_by_default, result))
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000539 # Safe expressed as bytes rather than str
540 result = urllib.parse.quote(quote_by_default, safe=b"<>")
541 self.assertEqual(quote_by_default, result,
542 "using quote(): %r != %r" % (quote_by_default, result))
543 # "Safe" non-ASCII characters should have no effect
544 # (Since URIs are not allowed to have non-ASCII characters)
545 result = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="\xfc")
546 expect = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="")
547 self.assertEqual(expect, result,
548 "using quote(): %r != %r" %
549 (expect, result))
550 # Same as above, but using a bytes rather than str
551 result = urllib.parse.quote("a\xfcb", encoding="latin-1", safe=b"\xfc")
552 expect = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="")
553 self.assertEqual(expect, result,
554 "using quote(): %r != %r" %
555 (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000556
557 def test_default_quoting(self):
558 # Make sure all characters that should be quoted are by default sans
559 # space (separate test for that).
560 should_quote = [chr(num) for num in range(32)] # For 0x00 - 0x1F
561 should_quote.append('<>#%"{}|\^[]`')
562 should_quote.append(chr(127)) # For 0x7F
563 should_quote = ''.join(should_quote)
564 for char in should_quote:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000565 result = urllib.parse.quote(char)
Brett Cannon74bfd702003-04-25 09:39:47 +0000566 self.assertEqual(hexescape(char), result,
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000567 "using quote(): "
568 "%s should be escaped to %s, not %s" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000569 (char, hexescape(char), result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000570 result = urllib.parse.quote_plus(char)
Brett Cannon74bfd702003-04-25 09:39:47 +0000571 self.assertEqual(hexescape(char), result,
572 "using quote_plus(): "
Tim Petersc2659cf2003-05-12 20:19:37 +0000573 "%s should be escapes to %s, not %s" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000574 (char, hexescape(char), result))
575 del should_quote
576 partial_quote = "ab[]cd"
577 expected = "ab%5B%5Dcd"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000578 result = urllib.parse.quote(partial_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000579 self.assertEqual(expected, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000580 "using quote(): %r != %r" % (expected, result))
Senthil Kumaran305a68e2011-09-13 06:40:27 +0800581 result = urllib.parse.quote_plus(partial_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000582 self.assertEqual(expected, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000583 "using quote_plus(): %r != %r" % (expected, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000584
585 def test_quoting_space(self):
586 # Make sure quote() and quote_plus() handle spaces as specified in
587 # their unique way
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000588 result = urllib.parse.quote(' ')
Brett Cannon74bfd702003-04-25 09:39:47 +0000589 self.assertEqual(result, hexescape(' '),
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000590 "using quote(): %r != %r" % (result, hexescape(' ')))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000591 result = urllib.parse.quote_plus(' ')
Brett Cannon74bfd702003-04-25 09:39:47 +0000592 self.assertEqual(result, '+',
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000593 "using quote_plus(): %r != +" % result)
Brett Cannon74bfd702003-04-25 09:39:47 +0000594 given = "a b cd e f"
595 expect = given.replace(' ', hexescape(' '))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000596 result = urllib.parse.quote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000597 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000598 "using quote(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000599 expect = given.replace(' ', '+')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000600 result = urllib.parse.quote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000601 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000602 "using quote_plus(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000603
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000604 def test_quoting_plus(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000605 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma'),
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000606 'alpha%2Bbeta+gamma')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000607 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma', '+'),
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000608 'alpha+beta+gamma')
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000609 # Test with bytes
610 self.assertEqual(urllib.parse.quote_plus(b'alpha+beta gamma'),
611 'alpha%2Bbeta+gamma')
612 # Test with safe bytes
613 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma', b'+'),
614 'alpha+beta+gamma')
615
616 def test_quote_bytes(self):
617 # Bytes should quote directly to percent-encoded values
618 given = b"\xa2\xd8ab\xff"
619 expect = "%A2%D8ab%FF"
620 result = urllib.parse.quote(given)
621 self.assertEqual(expect, result,
622 "using quote(): %r != %r" % (expect, result))
623 # Encoding argument should raise type error on bytes input
624 self.assertRaises(TypeError, urllib.parse.quote, given,
625 encoding="latin-1")
626 # quote_from_bytes should work the same
627 result = urllib.parse.quote_from_bytes(given)
628 self.assertEqual(expect, result,
629 "using quote_from_bytes(): %r != %r"
630 % (expect, result))
631
632 def test_quote_with_unicode(self):
633 # Characters in Latin-1 range, encoded by default in UTF-8
634 given = "\xa2\xd8ab\xff"
635 expect = "%C2%A2%C3%98ab%C3%BF"
636 result = urllib.parse.quote(given)
637 self.assertEqual(expect, result,
638 "using quote(): %r != %r" % (expect, result))
639 # Characters in Latin-1 range, encoded by with None (default)
640 result = urllib.parse.quote(given, encoding=None, errors=None)
641 self.assertEqual(expect, result,
642 "using quote(): %r != %r" % (expect, result))
643 # Characters in Latin-1 range, encoded with Latin-1
644 given = "\xa2\xd8ab\xff"
645 expect = "%A2%D8ab%FF"
646 result = urllib.parse.quote(given, encoding="latin-1")
647 self.assertEqual(expect, result,
648 "using quote(): %r != %r" % (expect, result))
649 # Characters in BMP, encoded by default in UTF-8
650 given = "\u6f22\u5b57" # "Kanji"
651 expect = "%E6%BC%A2%E5%AD%97"
652 result = urllib.parse.quote(given)
653 self.assertEqual(expect, result,
654 "using quote(): %r != %r" % (expect, result))
655 # Characters in BMP, encoded with Latin-1
656 given = "\u6f22\u5b57"
657 self.assertRaises(UnicodeEncodeError, urllib.parse.quote, given,
658 encoding="latin-1")
659 # Characters in BMP, encoded with Latin-1, with replace error handling
660 given = "\u6f22\u5b57"
661 expect = "%3F%3F" # "??"
662 result = urllib.parse.quote(given, encoding="latin-1",
663 errors="replace")
664 self.assertEqual(expect, result,
665 "using quote(): %r != %r" % (expect, result))
666 # Characters in BMP, Latin-1, with xmlcharref error handling
667 given = "\u6f22\u5b57"
668 expect = "%26%2328450%3B%26%2323383%3B" # "&#28450;&#23383;"
669 result = urllib.parse.quote(given, encoding="latin-1",
670 errors="xmlcharrefreplace")
671 self.assertEqual(expect, result,
672 "using quote(): %r != %r" % (expect, result))
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000673
Georg Brandlfaf41492009-05-26 18:31:11 +0000674 def test_quote_plus_with_unicode(self):
675 # Encoding (latin-1) test for quote_plus
676 given = "\xa2\xd8 \xff"
677 expect = "%A2%D8+%FF"
678 result = urllib.parse.quote_plus(given, encoding="latin-1")
679 self.assertEqual(expect, result,
680 "using quote_plus(): %r != %r" % (expect, result))
681 # Errors test for quote_plus
682 given = "ab\u6f22\u5b57 cd"
683 expect = "ab%3F%3F+cd"
684 result = urllib.parse.quote_plus(given, encoding="latin-1",
685 errors="replace")
686 self.assertEqual(expect, result,
687 "using quote_plus(): %r != %r" % (expect, result))
688
Senthil Kumarand496c4c2010-07-30 19:34:36 +0000689
Brett Cannon74bfd702003-04-25 09:39:47 +0000690class UnquotingTests(unittest.TestCase):
691 """Tests for unquote() and unquote_plus()
Tim Petersc2659cf2003-05-12 20:19:37 +0000692
Brett Cannon74bfd702003-04-25 09:39:47 +0000693 See the doc string for quoting_Tests for details on quoting and such.
694
695 """
696
697 def test_unquoting(self):
698 # Make sure unquoting of all ASCII values works
699 escape_list = []
700 for num in range(128):
701 given = hexescape(chr(num))
702 expect = chr(num)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000703 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000704 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000705 "using unquote(): %r != %r" % (expect, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000706 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000707 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000708 "using unquote_plus(): %r != %r" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000709 (expect, result))
710 escape_list.append(given)
711 escape_string = ''.join(escape_list)
712 del escape_list
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000713 result = urllib.parse.unquote(escape_string)
Brett Cannon74bfd702003-04-25 09:39:47 +0000714 self.assertEqual(result.count('%'), 1,
Brett Cannon74bfd702003-04-25 09:39:47 +0000715 "using unquote(): not all characters escaped: "
716 "%s" % result)
Georg Brandl604ef372010-07-31 08:20:02 +0000717 self.assertRaises((TypeError, AttributeError), urllib.parse.unquote, None)
718 self.assertRaises((TypeError, AttributeError), urllib.parse.unquote, ())
Florent Xicluna62829dc2010-08-14 20:51:58 +0000719 with support.check_warnings(('', BytesWarning), quiet=True):
720 self.assertRaises((TypeError, AttributeError), urllib.parse.unquote, b'')
Brett Cannon74bfd702003-04-25 09:39:47 +0000721
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000722 def test_unquoting_badpercent(self):
723 # Test unquoting on bad percent-escapes
724 given = '%xab'
725 expect = given
726 result = urllib.parse.unquote(given)
727 self.assertEqual(expect, result, "using unquote(): %r != %r"
728 % (expect, result))
729 given = '%x'
730 expect = given
731 result = urllib.parse.unquote(given)
732 self.assertEqual(expect, result, "using unquote(): %r != %r"
733 % (expect, result))
734 given = '%'
735 expect = given
736 result = urllib.parse.unquote(given)
737 self.assertEqual(expect, result, "using unquote(): %r != %r"
738 % (expect, result))
739 # unquote_to_bytes
740 given = '%xab'
741 expect = bytes(given, 'ascii')
742 result = urllib.parse.unquote_to_bytes(given)
743 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
744 % (expect, result))
745 given = '%x'
746 expect = bytes(given, 'ascii')
747 result = urllib.parse.unquote_to_bytes(given)
748 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
749 % (expect, result))
750 given = '%'
751 expect = bytes(given, 'ascii')
752 result = urllib.parse.unquote_to_bytes(given)
753 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
754 % (expect, result))
Georg Brandl604ef372010-07-31 08:20:02 +0000755 self.assertRaises((TypeError, AttributeError), urllib.parse.unquote_to_bytes, None)
756 self.assertRaises((TypeError, AttributeError), urllib.parse.unquote_to_bytes, ())
Senthil Kumaran79e17f62010-07-19 18:17:19 +0000757
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000758 def test_unquoting_mixed_case(self):
759 # Test unquoting on mixed-case hex digits in the percent-escapes
760 given = '%Ab%eA'
761 expect = b'\xab\xea'
762 result = urllib.parse.unquote_to_bytes(given)
763 self.assertEqual(expect, result,
764 "using unquote_to_bytes(): %r != %r"
765 % (expect, result))
766
Brett Cannon74bfd702003-04-25 09:39:47 +0000767 def test_unquoting_parts(self):
768 # Make sure unquoting works when have non-quoted characters
769 # interspersed
770 given = 'ab%sd' % hexescape('c')
771 expect = "abcd"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000772 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000773 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000774 "using quote(): %r != %r" % (expect, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000775 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000776 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000777 "using unquote_plus(): %r != %r" % (expect, result))
Tim Petersc2659cf2003-05-12 20:19:37 +0000778
Brett Cannon74bfd702003-04-25 09:39:47 +0000779 def test_unquoting_plus(self):
780 # Test difference between unquote() and unquote_plus()
781 given = "are+there+spaces..."
782 expect = given
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000783 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000784 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000785 "using unquote(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000786 expect = given.replace('+', ' ')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000787 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000788 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000789 "using unquote_plus(): %r != %r" % (expect, result))
790
791 def test_unquote_to_bytes(self):
792 given = 'br%C3%BCckner_sapporo_20050930.doc'
793 expect = b'br\xc3\xbcckner_sapporo_20050930.doc'
794 result = urllib.parse.unquote_to_bytes(given)
795 self.assertEqual(expect, result,
796 "using unquote_to_bytes(): %r != %r"
797 % (expect, result))
798 # Test on a string with unescaped non-ASCII characters
799 # (Technically an invalid URI; expect those characters to be UTF-8
800 # encoded).
801 result = urllib.parse.unquote_to_bytes("\u6f22%C3%BC")
802 expect = b'\xe6\xbc\xa2\xc3\xbc' # UTF-8 for "\u6f22\u00fc"
803 self.assertEqual(expect, result,
804 "using unquote_to_bytes(): %r != %r"
805 % (expect, result))
806 # Test with a bytes as input
807 given = b'%A2%D8ab%FF'
808 expect = b'\xa2\xd8ab\xff'
809 result = urllib.parse.unquote_to_bytes(given)
810 self.assertEqual(expect, result,
811 "using unquote_to_bytes(): %r != %r"
812 % (expect, result))
813 # Test with a bytes as input, with unescaped non-ASCII bytes
814 # (Technically an invalid URI; expect those bytes to be preserved)
815 given = b'%A2\xd8ab%FF'
816 expect = b'\xa2\xd8ab\xff'
817 result = urllib.parse.unquote_to_bytes(given)
818 self.assertEqual(expect, result,
819 "using unquote_to_bytes(): %r != %r"
820 % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000821
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +0000822 def test_unquote_with_unicode(self):
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000823 # Characters in the Latin-1 range, encoded with UTF-8
824 given = 'br%C3%BCckner_sapporo_20050930.doc'
825 expect = 'br\u00fcckner_sapporo_20050930.doc'
826 result = urllib.parse.unquote(given)
827 self.assertEqual(expect, result,
828 "using unquote(): %r != %r" % (expect, result))
829 # Characters in the Latin-1 range, encoded with None (default)
830 result = urllib.parse.unquote(given, encoding=None, errors=None)
831 self.assertEqual(expect, result,
832 "using unquote(): %r != %r" % (expect, result))
833
834 # Characters in the Latin-1 range, encoded with Latin-1
835 result = urllib.parse.unquote('br%FCckner_sapporo_20050930.doc',
836 encoding="latin-1")
837 expect = 'br\u00fcckner_sapporo_20050930.doc'
838 self.assertEqual(expect, result,
839 "using unquote(): %r != %r" % (expect, result))
840
841 # Characters in BMP, encoded with UTF-8
842 given = "%E6%BC%A2%E5%AD%97"
843 expect = "\u6f22\u5b57" # "Kanji"
844 result = urllib.parse.unquote(given)
845 self.assertEqual(expect, result,
846 "using unquote(): %r != %r" % (expect, result))
847
848 # Decode with UTF-8, invalid sequence
849 given = "%F3%B1"
850 expect = "\ufffd" # Replacement character
851 result = urllib.parse.unquote(given)
852 self.assertEqual(expect, result,
853 "using unquote(): %r != %r" % (expect, result))
854
855 # Decode with UTF-8, invalid sequence, replace errors
856 result = urllib.parse.unquote(given, errors="replace")
857 self.assertEqual(expect, result,
858 "using unquote(): %r != %r" % (expect, result))
859
860 # Decode with UTF-8, invalid sequence, ignoring errors
861 given = "%F3%B1"
862 expect = ""
863 result = urllib.parse.unquote(given, errors="ignore")
864 self.assertEqual(expect, result,
865 "using unquote(): %r != %r" % (expect, result))
866
867 # A mix of non-ASCII and percent-encoded characters, UTF-8
868 result = urllib.parse.unquote("\u6f22%C3%BC")
869 expect = '\u6f22\u00fc'
870 self.assertEqual(expect, result,
871 "using unquote(): %r != %r" % (expect, result))
872
873 # A mix of non-ASCII and percent-encoded characters, Latin-1
874 # (Note, the string contains non-Latin-1-representable characters)
875 result = urllib.parse.unquote("\u6f22%FC", encoding="latin-1")
876 expect = '\u6f22\u00fc'
877 self.assertEqual(expect, result,
878 "using unquote(): %r != %r" % (expect, result))
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +0000879
Brett Cannon74bfd702003-04-25 09:39:47 +0000880class urlencode_Tests(unittest.TestCase):
881 """Tests for urlencode()"""
882
883 def help_inputtype(self, given, test_type):
884 """Helper method for testing different input types.
Tim Petersc2659cf2003-05-12 20:19:37 +0000885
Brett Cannon74bfd702003-04-25 09:39:47 +0000886 'given' must lead to only the pairs:
887 * 1st, 1
888 * 2nd, 2
889 * 3rd, 3
Tim Petersc2659cf2003-05-12 20:19:37 +0000890
Brett Cannon74bfd702003-04-25 09:39:47 +0000891 Test cannot assume anything about order. Docs make no guarantee and
892 have possible dictionary input.
Tim Petersc2659cf2003-05-12 20:19:37 +0000893
Brett Cannon74bfd702003-04-25 09:39:47 +0000894 """
895 expect_somewhere = ["1st=1", "2nd=2", "3rd=3"]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000896 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000897 for expected in expect_somewhere:
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000898 self.assertIn(expected, result,
Brett Cannon74bfd702003-04-25 09:39:47 +0000899 "testing %s: %s not found in %s" %
900 (test_type, expected, result))
901 self.assertEqual(result.count('&'), 2,
902 "testing %s: expected 2 '&'s; got %s" %
903 (test_type, result.count('&')))
904 amp_location = result.index('&')
905 on_amp_left = result[amp_location - 1]
906 on_amp_right = result[amp_location + 1]
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000907 self.assertTrue(on_amp_left.isdigit() and on_amp_right.isdigit(),
Brett Cannon74bfd702003-04-25 09:39:47 +0000908 "testing %s: '&' not located in proper place in %s" %
909 (test_type, result))
910 self.assertEqual(len(result), (5 * 3) + 2, #5 chars per thing and amps
911 "testing %s: "
912 "unexpected number of characters: %s != %s" %
913 (test_type, len(result), (5 * 3) + 2))
914
915 def test_using_mapping(self):
916 # Test passing in a mapping object as an argument.
917 self.help_inputtype({"1st":'1', "2nd":'2', "3rd":'3'},
918 "using dict as input type")
919
920 def test_using_sequence(self):
921 # Test passing in a sequence of two-item sequences as an argument.
922 self.help_inputtype([('1st', '1'), ('2nd', '2'), ('3rd', '3')],
923 "using sequence of two-item tuples as input")
924
925 def test_quoting(self):
926 # Make sure keys and values are quoted using quote_plus()
927 given = {"&":"="}
928 expect = "%s=%s" % (hexescape('&'), hexescape('='))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000929 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000930 self.assertEqual(expect, result)
931 given = {"key name":"A bunch of pluses"}
932 expect = "key+name=A+bunch+of+pluses"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000933 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000934 self.assertEqual(expect, result)
935
936 def test_doseq(self):
937 # Test that passing True for 'doseq' parameter works correctly
938 given = {'sequence':['1', '2', '3']}
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000939 expect = "sequence=%s" % urllib.parse.quote_plus(str(['1', '2', '3']))
940 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000941 self.assertEqual(expect, result)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000942 result = urllib.parse.urlencode(given, True)
Brett Cannon74bfd702003-04-25 09:39:47 +0000943 for value in given["sequence"]:
944 expect = "sequence=%s" % value
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000945 self.assertIn(expect, result)
Brett Cannon74bfd702003-04-25 09:39:47 +0000946 self.assertEqual(result.count('&'), 2,
947 "Expected 2 '&'s, got %s" % result.count('&'))
948
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000949 def test_empty_sequence(self):
950 self.assertEqual("", urllib.parse.urlencode({}))
951 self.assertEqual("", urllib.parse.urlencode([]))
952
953 def test_nonstring_values(self):
954 self.assertEqual("a=1", urllib.parse.urlencode({"a": 1}))
955 self.assertEqual("a=None", urllib.parse.urlencode({"a": None}))
956
957 def test_nonstring_seq_values(self):
958 self.assertEqual("a=1&a=2", urllib.parse.urlencode({"a": [1, 2]}, True))
959 self.assertEqual("a=None&a=a",
960 urllib.parse.urlencode({"a": [None, "a"]}, True))
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100961 data = collections.OrderedDict([("a", 1), ("b", 1)])
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000962 self.assertEqual("a=a&a=b",
Georg Brandl2daf6ae2012-02-20 19:54:16 +0100963 urllib.parse.urlencode({"a": data}, True))
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000964
Senthil Kumarandf022da2010-07-03 17:48:22 +0000965 def test_urlencode_encoding(self):
966 # ASCII encoding. Expect %3F with errors="replace'
967 given = (('\u00a0', '\u00c1'),)
968 expect = '%3F=%3F'
969 result = urllib.parse.urlencode(given, encoding="ASCII", errors="replace")
970 self.assertEqual(expect, result)
971
972 # Default is UTF-8 encoding.
973 given = (('\u00a0', '\u00c1'),)
974 expect = '%C2%A0=%C3%81'
975 result = urllib.parse.urlencode(given)
976 self.assertEqual(expect, result)
977
978 # Latin-1 encoding.
979 given = (('\u00a0', '\u00c1'),)
980 expect = '%A0=%C1'
981 result = urllib.parse.urlencode(given, encoding="latin-1")
982 self.assertEqual(expect, result)
983
984 def test_urlencode_encoding_doseq(self):
985 # ASCII Encoding. Expect %3F with errors="replace'
986 given = (('\u00a0', '\u00c1'),)
987 expect = '%3F=%3F'
988 result = urllib.parse.urlencode(given, doseq=True,
989 encoding="ASCII", errors="replace")
990 self.assertEqual(expect, result)
991
992 # ASCII Encoding. On a sequence of values.
993 given = (("\u00a0", (1, "\u00c1")),)
994 expect = '%3F=1&%3F=%3F'
995 result = urllib.parse.urlencode(given, True,
996 encoding="ASCII", errors="replace")
997 self.assertEqual(expect, result)
998
999 # Utf-8
1000 given = (("\u00a0", "\u00c1"),)
1001 expect = '%C2%A0=%C3%81'
1002 result = urllib.parse.urlencode(given, True)
1003 self.assertEqual(expect, result)
1004
1005 given = (("\u00a0", (42, "\u00c1")),)
1006 expect = '%C2%A0=42&%C2%A0=%C3%81'
1007 result = urllib.parse.urlencode(given, True)
1008 self.assertEqual(expect, result)
1009
1010 # latin-1
1011 given = (("\u00a0", "\u00c1"),)
1012 expect = '%A0=%C1'
1013 result = urllib.parse.urlencode(given, True, encoding="latin-1")
1014 self.assertEqual(expect, result)
1015
1016 given = (("\u00a0", (42, "\u00c1")),)
1017 expect = '%A0=42&%A0=%C1'
1018 result = urllib.parse.urlencode(given, True, encoding="latin-1")
1019 self.assertEqual(expect, result)
1020
1021 def test_urlencode_bytes(self):
1022 given = ((b'\xa0\x24', b'\xc1\x24'),)
1023 expect = '%A0%24=%C1%24'
1024 result = urllib.parse.urlencode(given)
1025 self.assertEqual(expect, result)
1026 result = urllib.parse.urlencode(given, True)
1027 self.assertEqual(expect, result)
1028
1029 # Sequence of values
1030 given = ((b'\xa0\x24', (42, b'\xc1\x24')),)
1031 expect = '%A0%24=42&%A0%24=%C1%24'
1032 result = urllib.parse.urlencode(given, True)
1033 self.assertEqual(expect, result)
1034
1035 def test_urlencode_encoding_safe_parameter(self):
1036
1037 # Send '$' (\x24) as safe character
1038 # Default utf-8 encoding
1039
1040 given = ((b'\xa0\x24', b'\xc1\x24'),)
1041 result = urllib.parse.urlencode(given, safe=":$")
1042 expect = '%A0$=%C1$'
1043 self.assertEqual(expect, result)
1044
1045 given = ((b'\xa0\x24', b'\xc1\x24'),)
1046 result = urllib.parse.urlencode(given, doseq=True, safe=":$")
1047 expect = '%A0$=%C1$'
1048 self.assertEqual(expect, result)
1049
1050 # Safe parameter in sequence
1051 given = ((b'\xa0\x24', (b'\xc1\x24', 0xd, 42)),)
1052 expect = '%A0$=%C1$&%A0$=13&%A0$=42'
1053 result = urllib.parse.urlencode(given, True, safe=":$")
1054 self.assertEqual(expect, result)
1055
1056 # Test all above in latin-1 encoding
1057
1058 given = ((b'\xa0\x24', b'\xc1\x24'),)
1059 result = urllib.parse.urlencode(given, safe=":$",
1060 encoding="latin-1")
1061 expect = '%A0$=%C1$'
1062 self.assertEqual(expect, result)
1063
1064 given = ((b'\xa0\x24', b'\xc1\x24'),)
1065 expect = '%A0$=%C1$'
1066 result = urllib.parse.urlencode(given, doseq=True, safe=":$",
1067 encoding="latin-1")
1068
1069 given = ((b'\xa0\x24', (b'\xc1\x24', 0xd, 42)),)
1070 expect = '%A0$=%C1$&%A0$=13&%A0$=42'
1071 result = urllib.parse.urlencode(given, True, safe=":$",
1072 encoding="latin-1")
1073 self.assertEqual(expect, result)
1074
Brett Cannon74bfd702003-04-25 09:39:47 +00001075class Pathname_Tests(unittest.TestCase):
1076 """Test pathname2url() and url2pathname()"""
1077
1078 def test_basic(self):
1079 # Make sure simple tests pass
1080 expected_path = os.path.join("parts", "of", "a", "path")
1081 expected_url = "parts/of/a/path"
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001082 result = urllib.request.pathname2url(expected_path)
Brett Cannon74bfd702003-04-25 09:39:47 +00001083 self.assertEqual(expected_url, result,
1084 "pathname2url() failed; %s != %s" %
1085 (result, expected_url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001086 result = urllib.request.url2pathname(expected_url)
Brett Cannon74bfd702003-04-25 09:39:47 +00001087 self.assertEqual(expected_path, result,
1088 "url2pathame() failed; %s != %s" %
1089 (result, expected_path))
1090
1091 def test_quoting(self):
1092 # Test automatic quoting and unquoting works for pathnam2url() and
1093 # url2pathname() respectively
1094 given = os.path.join("needs", "quot=ing", "here")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001095 expect = "needs/%s/here" % urllib.parse.quote("quot=ing")
1096 result = urllib.request.pathname2url(given)
Brett Cannon74bfd702003-04-25 09:39:47 +00001097 self.assertEqual(expect, result,
1098 "pathname2url() failed; %s != %s" %
1099 (expect, result))
1100 expect = given
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001101 result = urllib.request.url2pathname(result)
Brett Cannon74bfd702003-04-25 09:39:47 +00001102 self.assertEqual(expect, result,
1103 "url2pathname() failed; %s != %s" %
1104 (expect, result))
1105 given = os.path.join("make sure", "using_quote")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001106 expect = "%s/using_quote" % urllib.parse.quote("make sure")
1107 result = urllib.request.pathname2url(given)
Brett Cannon74bfd702003-04-25 09:39:47 +00001108 self.assertEqual(expect, result,
1109 "pathname2url() failed; %s != %s" %
1110 (expect, result))
1111 given = "make+sure/using_unquote"
1112 expect = os.path.join("make+sure", "using_unquote")
Jeremy Hylton1afc1692008-06-18 20:49:58 +00001113 result = urllib.request.url2pathname(given)
Brett Cannon74bfd702003-04-25 09:39:47 +00001114 self.assertEqual(expect, result,
1115 "url2pathname() failed; %s != %s" %
1116 (expect, result))
Tim Petersc2659cf2003-05-12 20:19:37 +00001117
Senthil Kumaran2d2ea1b2011-04-14 13:16:30 +08001118 @unittest.skipUnless(sys.platform == 'win32',
1119 'test specific to the urllib.url2path function.')
1120 def test_ntpath(self):
1121 given = ('/C:/', '///C:/', '/C|//')
1122 expect = 'C:\\'
1123 for url in given:
1124 result = urllib.request.url2pathname(url)
1125 self.assertEqual(expect, result,
1126 'urllib.request..url2pathname() failed; %s != %s' %
1127 (expect, result))
1128 given = '///C|/path'
1129 expect = 'C:\\path'
1130 result = urllib.request.url2pathname(given)
1131 self.assertEqual(expect, result,
1132 'urllib.request.url2pathname() failed; %s != %s' %
1133 (expect, result))
1134
Senthil Kumaraneaaec272009-03-30 21:54:41 +00001135class Utility_Tests(unittest.TestCase):
1136 """Testcase to test the various utility functions in the urllib."""
1137
1138 def test_splitpasswd(self):
1139 """Some of password examples are not sensible, but it is added to
1140 confirming to RFC2617 and addressing issue4675.
1141 """
1142 self.assertEqual(('user', 'ab'),urllib.parse.splitpasswd('user:ab'))
1143 self.assertEqual(('user', 'a\nb'),urllib.parse.splitpasswd('user:a\nb'))
1144 self.assertEqual(('user', 'a\tb'),urllib.parse.splitpasswd('user:a\tb'))
1145 self.assertEqual(('user', 'a\rb'),urllib.parse.splitpasswd('user:a\rb'))
1146 self.assertEqual(('user', 'a\fb'),urllib.parse.splitpasswd('user:a\fb'))
1147 self.assertEqual(('user', 'a\vb'),urllib.parse.splitpasswd('user:a\vb'))
1148 self.assertEqual(('user', 'a:b'),urllib.parse.splitpasswd('user:a:b'))
Senthil Kumaranc5c5a142012-01-14 19:09:04 +08001149 self.assertEqual(('user', 'a b'),urllib.parse.splitpasswd('user:a b'))
1150 self.assertEqual(('user 2', 'ab'),urllib.parse.splitpasswd('user 2:ab'))
1151 self.assertEqual(('user+1', 'a+b'),urllib.parse.splitpasswd('user+1:a+b'))
Senthil Kumaraneaaec272009-03-30 21:54:41 +00001152
Senthil Kumaran1b7da512011-10-06 00:32:02 +08001153 def test_thishost(self):
1154 """Test the urllib.request.thishost utility function returns a tuple"""
1155 self.assertIsInstance(urllib.request.thishost(), tuple)
1156
Senthil Kumaran690ce9b2009-05-05 18:41:13 +00001157
1158class URLopener_Tests(unittest.TestCase):
1159 """Testcase to test the open method of URLopener class."""
1160
1161 def test_quoted_open(self):
1162 class DummyURLopener(urllib.request.URLopener):
1163 def open_spam(self, url):
1164 return url
1165
1166 self.assertEqual(DummyURLopener().open(
1167 'spam://example/ /'),'//example/%20/')
1168
Senthil Kumaran734f0592010-02-20 22:19:04 +00001169 # test the safe characters are not quoted by urlopen
1170 self.assertEqual(DummyURLopener().open(
1171 "spam://c:|windows%/:=&?~#+!$,;'@()*[]|/path/"),
1172 "//c:|windows%/:=&?~#+!$,;'@()*[]|/path/")
1173
Guido van Rossume7ba4952007-06-06 23:52:48 +00001174# Just commented them out.
1175# Can't really tell why keep failing in windows and sparc.
Ezio Melotti13925002011-03-16 11:05:33 +02001176# Everywhere else they work ok, but on those machines, sometimes
Guido van Rossume7ba4952007-06-06 23:52:48 +00001177# fail in one of the tests, sometimes in other. I have a linux, and
1178# the tests go ok.
1179# If anybody has one of the problematic enviroments, please help!
1180# . Facundo
1181#
1182# def server(evt):
Georg Brandlf78e02b2008-06-10 17:40:04 +00001183# import socket, time
Guido van Rossume7ba4952007-06-06 23:52:48 +00001184# serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
1185# serv.settimeout(3)
1186# serv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
1187# serv.bind(("", 9093))
1188# serv.listen(5)
1189# try:
1190# conn, addr = serv.accept()
1191# conn.send("1 Hola mundo\n")
1192# cantdata = 0
1193# while cantdata < 13:
1194# data = conn.recv(13-cantdata)
1195# cantdata += len(data)
1196# time.sleep(.3)
1197# conn.send("2 No more lines\n")
1198# conn.close()
1199# except socket.timeout:
1200# pass
1201# finally:
1202# serv.close()
1203# evt.set()
1204#
1205# class FTPWrapperTests(unittest.TestCase):
1206#
1207# def setUp(self):
Georg Brandlf78e02b2008-06-10 17:40:04 +00001208# import ftplib, time, threading
Guido van Rossume7ba4952007-06-06 23:52:48 +00001209# ftplib.FTP.port = 9093
1210# self.evt = threading.Event()
1211# threading.Thread(target=server, args=(self.evt,)).start()
1212# time.sleep(.1)
1213#
1214# def tearDown(self):
1215# self.evt.wait()
1216#
1217# def testBasic(self):
1218# # connects
1219# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
Georg Brandlf78e02b2008-06-10 17:40:04 +00001220# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001221#
1222# def testTimeoutNone(self):
Georg Brandlf78e02b2008-06-10 17:40:04 +00001223# # global default timeout is ignored
1224# import socket
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001225# self.assertTrue(socket.getdefaulttimeout() is None)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001226# socket.setdefaulttimeout(30)
1227# try:
1228# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
1229# finally:
Georg Brandlf78e02b2008-06-10 17:40:04 +00001230# socket.setdefaulttimeout(None)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001231# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
Georg Brandlf78e02b2008-06-10 17:40:04 +00001232# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001233#
Georg Brandlf78e02b2008-06-10 17:40:04 +00001234# def testTimeoutDefault(self):
1235# # global default timeout is used
1236# import socket
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001237# self.assertTrue(socket.getdefaulttimeout() is None)
Georg Brandlf78e02b2008-06-10 17:40:04 +00001238# socket.setdefaulttimeout(30)
1239# try:
1240# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
1241# finally:
1242# socket.setdefaulttimeout(None)
1243# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
1244# ftp.close()
1245#
1246# def testTimeoutValue(self):
1247# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [],
1248# timeout=30)
1249# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
1250# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001251
Senthil Kumarande49d642011-10-16 23:54:44 +08001252class RequestTests(unittest.TestCase):
1253 """Unit tests for urllib.request.Request."""
1254
1255 def test_default_values(self):
1256 Request = urllib.request.Request
1257 request = Request("http://www.python.org")
1258 self.assertEqual(request.get_method(), 'GET')
1259 request = Request("http://www.python.org", {})
1260 self.assertEqual(request.get_method(), 'POST')
1261
1262 def test_with_method_arg(self):
1263 Request = urllib.request.Request
1264 request = Request("http://www.python.org", method='HEAD')
1265 self.assertEqual(request.method, 'HEAD')
1266 self.assertEqual(request.get_method(), 'HEAD')
1267 request = Request("http://www.python.org", {}, method='HEAD')
1268 self.assertEqual(request.method, 'HEAD')
1269 self.assertEqual(request.get_method(), 'HEAD')
1270 request = Request("http://www.python.org", method='GET')
1271 self.assertEqual(request.get_method(), 'GET')
1272 request.method = 'HEAD'
1273 self.assertEqual(request.get_method(), 'HEAD')
Skip Montanaro080c9972001-01-28 21:12:22 +00001274
1275
Brett Cannon74bfd702003-04-25 09:39:47 +00001276def test_main():
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001277 support.run_unittest(
Walter Dörwald21d3a322003-05-01 17:45:56 +00001278 urlopen_FileTests,
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001279 urlopen_HttpTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001280 urlretrieve_FileTests,
Senthil Kumarance260142011-11-01 01:35:17 +08001281 urlretrieve_HttpTests,
Benjamin Peterson9bc93512008-09-22 22:10:59 +00001282 ProxyTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001283 QuotingTests,
1284 UnquotingTests,
1285 urlencode_Tests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00001286 Pathname_Tests,
Senthil Kumaraneaaec272009-03-30 21:54:41 +00001287 Utility_Tests,
Senthil Kumaran690ce9b2009-05-05 18:41:13 +00001288 URLopener_Tests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00001289 #FTPWrapperTests,
Senthil Kumarande49d642011-10-16 23:54:44 +08001290 RequestTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001291 )
Brett Cannon74bfd702003-04-25 09:39:47 +00001292
1293
1294
1295if __name__ == '__main__':
1296 test_main()