blob: 23492fb3bb72385473a482703dab076ee77ed4a2 [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
Georg Brandl24420152008-05-26 16:32:26 +00005import http.client
Barry Warsaw820c1202008-06-12 04:06:45 +00006import email.message
Jeremy Hylton66dc8c52007-08-04 03:42:26 +00007import io
Brett Cannon74bfd702003-04-25 09:39:47 +00008import unittest
Benjamin Petersonee8712c2008-05-20 21:35:26 +00009from test import support
Brett Cannon74bfd702003-04-25 09:39:47 +000010import os
Georg Brandl5a650a22005-08-26 08:51:34 +000011import tempfile
Jeremy Hylton6102e292000-08-31 15:48:10 +000012
Brett Cannon74bfd702003-04-25 09:39:47 +000013def hexescape(char):
14 """Escape char as RFC 2396 specifies"""
15 hex_repr = hex(ord(char))[2:].upper()
16 if len(hex_repr) == 1:
17 hex_repr = "0%s" % hex_repr
18 return "%" + hex_repr
Jeremy Hylton6102e292000-08-31 15:48:10 +000019
Jeremy Hylton1afc1692008-06-18 20:49:58 +000020# Shortcut for testing FancyURLopener
21_urlopener = None
22def urlopen(url, data=None, proxies=None):
23 """urlopen(url [, data]) -> open file-like object"""
24 global _urlopener
25 if proxies is not None:
26 opener = urllib.request.FancyURLopener(proxies=proxies)
27 elif not _urlopener:
28 opener = urllib.request.FancyURLopener()
29 _urlopener = opener
30 else:
31 opener = _urlopener
32 if data is None:
33 return opener.open(url)
34 else:
35 return opener.open(url, data)
36
Brett Cannon74bfd702003-04-25 09:39:47 +000037class urlopen_FileTests(unittest.TestCase):
38 """Test urlopen() opening a temporary file.
Jeremy Hylton6102e292000-08-31 15:48:10 +000039
Brett Cannon74bfd702003-04-25 09:39:47 +000040 Try to test as much functionality as possible so as to cut down on reliance
Andrew M. Kuchlingf1a2f9e2004-06-29 13:07:53 +000041 on connecting to the Net for testing.
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000042
Brett Cannon74bfd702003-04-25 09:39:47 +000043 """
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000044
Brett Cannon74bfd702003-04-25 09:39:47 +000045 def setUp(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +000046 # Create a temp file to use for testing
47 self.text = bytes("test_urllib: %s\n" % self.__class__.__name__,
48 "ascii")
49 f = open(support.TESTFN, 'wb')
Brett Cannon74bfd702003-04-25 09:39:47 +000050 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +000051 f.write(self.text)
Brett Cannon74bfd702003-04-25 09:39:47 +000052 finally:
Jeremy Hylton1afc1692008-06-18 20:49:58 +000053 f.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000054 self.pathname = support.TESTFN
Jeremy Hylton1afc1692008-06-18 20:49:58 +000055 self.returned_obj = urlopen("file:%s" % self.pathname)
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000056
Brett Cannon74bfd702003-04-25 09:39:47 +000057 def tearDown(self):
58 """Shut down the open object"""
59 self.returned_obj.close()
Benjamin Petersonee8712c2008-05-20 21:35:26 +000060 os.remove(support.TESTFN)
Jeremy Hylton7ae51bf2000-09-14 16:59:07 +000061
Brett Cannon74bfd702003-04-25 09:39:47 +000062 def test_interface(self):
63 # Make sure object returned by urlopen() has the specified methods
64 for attr in ("read", "readline", "readlines", "fileno",
Christian Heimes9bd667a2008-01-20 15:14:11 +000065 "close", "info", "geturl", "getcode", "__iter__"):
Benjamin Petersonc9c0f202009-06-30 23:06:06 +000066 self.assertTrue(hasattr(self.returned_obj, attr),
Brett Cannon74bfd702003-04-25 09:39:47 +000067 "object returned by urlopen() lacks %s attribute" %
68 attr)
Skip Montanaroe78b92a2001-01-20 20:22:30 +000069
Brett Cannon74bfd702003-04-25 09:39:47 +000070 def test_read(self):
71 self.assertEqual(self.text, self.returned_obj.read())
Skip Montanaro080c9972001-01-28 21:12:22 +000072
Brett Cannon74bfd702003-04-25 09:39:47 +000073 def test_readline(self):
74 self.assertEqual(self.text, self.returned_obj.readline())
Guido van Rossuma0982942007-07-10 08:30:03 +000075 self.assertEqual(b'', self.returned_obj.readline(),
Brett Cannon74bfd702003-04-25 09:39:47 +000076 "calling readline() after exhausting the file did not"
77 " return an empty string")
Skip Montanaro080c9972001-01-28 21:12:22 +000078
Brett Cannon74bfd702003-04-25 09:39:47 +000079 def test_readlines(self):
80 lines_list = self.returned_obj.readlines()
81 self.assertEqual(len(lines_list), 1,
82 "readlines() returned the wrong number of lines")
83 self.assertEqual(lines_list[0], self.text,
84 "readlines() returned improper text")
Skip Montanaro080c9972001-01-28 21:12:22 +000085
Brett Cannon74bfd702003-04-25 09:39:47 +000086 def test_fileno(self):
87 file_num = self.returned_obj.fileno()
Ezio Melottie9615932010-01-24 19:26:24 +000088 self.assertIsInstance(file_num, int, "fileno() did not return an int")
Brett Cannon74bfd702003-04-25 09:39:47 +000089 self.assertEqual(os.read(file_num, len(self.text)), self.text,
90 "Reading on the file descriptor returned by fileno() "
91 "did not return the expected text")
Skip Montanaroe78b92a2001-01-20 20:22:30 +000092
Brett Cannon74bfd702003-04-25 09:39:47 +000093 def test_close(self):
94 # Test close() by calling it hear and then having it be called again
95 # by the tearDown() method for the test
96 self.returned_obj.close()
Skip Montanaro080c9972001-01-28 21:12:22 +000097
Brett Cannon74bfd702003-04-25 09:39:47 +000098 def test_info(self):
Ezio Melottie9615932010-01-24 19:26:24 +000099 self.assertIsInstance(self.returned_obj.info(), email.message.Message)
Skip Montanaroe78b92a2001-01-20 20:22:30 +0000100
Brett Cannon74bfd702003-04-25 09:39:47 +0000101 def test_geturl(self):
102 self.assertEqual(self.returned_obj.geturl(), self.pathname)
Skip Montanaro080c9972001-01-28 21:12:22 +0000103
Christian Heimes9bd667a2008-01-20 15:14:11 +0000104 def test_getcode(self):
105 self.assertEqual(self.returned_obj.getcode(), None)
106
Brett Cannon74bfd702003-04-25 09:39:47 +0000107 def test_iter(self):
108 # Test iterator
109 # Don't need to count number of iterations since test would fail the
110 # instant it returned anything beyond the first line from the
111 # comparison
112 for line in self.returned_obj.__iter__():
113 self.assertEqual(line, self.text)
Skip Montanaro080c9972001-01-28 21:12:22 +0000114
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000115class ProxyTests(unittest.TestCase):
116
117 def setUp(self):
Walter Dörwaldb525e182009-04-26 21:39:21 +0000118 # Records changes to env vars
119 self.env = support.EnvironmentVarGuard()
Benjamin Peterson46a99002010-01-09 18:45:30 +0000120 # Delete all proxy related env vars
121 for k in os.environ.keys():
122 if 'proxy' in k.lower():
123 self.env.unset(k)
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000124
125 def tearDown(self):
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000126 # Restore all proxy related env vars
Walter Dörwaldb525e182009-04-26 21:39:21 +0000127 self.env.__exit__()
128 del self.env
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000129
130 def test_getproxies_environment_keep_no_proxies(self):
Walter Dörwaldb525e182009-04-26 21:39:21 +0000131 self.env.set('NO_PROXY', 'localhost')
132 proxies = urllib.request.getproxies_environment()
133 # getproxies_environment use lowered case truncated (no '_proxy') keys
134 self.assertEquals('localhost', proxies['no'])
Benjamin Peterson9bc93512008-09-22 22:10:59 +0000135
136
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000137class urlopen_HttpTests(unittest.TestCase):
138 """Test urlopen() opening a fake http connection."""
139
140 def fakehttp(self, fakedata):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000141 class FakeSocket(io.BytesIO):
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000142 def sendall(self, str): pass
Nick Coghlan598c3a82009-02-08 04:01:00 +0000143 def makefile(self, *args, **kwds):
144 return self
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000145 def read(self, amt=None):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000146 if self.closed: return b""
147 return io.BytesIO.read(self, amt)
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000148 def readline(self, length=None):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000149 if self.closed: return b""
150 return io.BytesIO.readline(self, length)
Georg Brandl24420152008-05-26 16:32:26 +0000151 class FakeHTTPConnection(http.client.HTTPConnection):
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000152 def connect(self):
153 self.sock = FakeSocket(fakedata)
Georg Brandl24420152008-05-26 16:32:26 +0000154 self._connection_class = http.client.HTTPConnection
155 http.client.HTTPConnection = FakeHTTPConnection
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000156
157 def unfakehttp(self):
Georg Brandl24420152008-05-26 16:32:26 +0000158 http.client.HTTPConnection = self._connection_class
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000159
160 def test_read(self):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000161 self.fakehttp(b"Hello!")
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000162 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000163 fp = urlopen("http://python.org/")
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000164 self.assertEqual(fp.readline(), b"Hello!")
165 self.assertEqual(fp.readline(), b"")
Christian Heimes9bd667a2008-01-20 15:14:11 +0000166 self.assertEqual(fp.geturl(), 'http://python.org/')
167 self.assertEqual(fp.getcode(), 200)
Hye-Shik Chang39aef792004-06-05 13:30:56 +0000168 finally:
169 self.unfakehttp()
170
Christian Heimes57dddfb2008-01-02 18:30:52 +0000171 def test_read_bogus(self):
172 # urlopen() should raise IOError for many error codes.
173 self.fakehttp(b'''HTTP/1.1 401 Authentication Required
174Date: Wed, 02 Jan 2008 03:03:54 GMT
175Server: Apache/1.3.33 (Debian GNU/Linux) mod_ssl/2.8.22 OpenSSL/0.9.7e
176Connection: close
177Content-Type: text/html; charset=iso-8859-1
178''')
179 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000180 self.assertRaises(IOError, urlopen, "http://python.org/")
Christian Heimes57dddfb2008-01-02 18:30:52 +0000181 finally:
182 self.unfakehttp()
183
Guido van Rossumd8faa362007-04-27 19:54:29 +0000184 def test_empty_socket(self):
Jeremy Hylton66dc8c52007-08-04 03:42:26 +0000185 # urlopen() raises IOError if the underlying socket does not send any
186 # data. (#1680230)
Christian Heimes57dddfb2008-01-02 18:30:52 +0000187 self.fakehttp(b'')
Guido van Rossumd8faa362007-04-27 19:54:29 +0000188 try:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000189 self.assertRaises(IOError, urlopen, "http://something")
Guido van Rossumd8faa362007-04-27 19:54:29 +0000190 finally:
191 self.unfakehttp()
192
Brett Cannon19691362003-04-29 05:08:06 +0000193class urlretrieve_FileTests(unittest.TestCase):
Brett Cannon74bfd702003-04-25 09:39:47 +0000194 """Test urllib.urlretrieve() on local files"""
Skip Montanaro080c9972001-01-28 21:12:22 +0000195
Brett Cannon19691362003-04-29 05:08:06 +0000196 def setUp(self):
Georg Brandl5a650a22005-08-26 08:51:34 +0000197 # Create a list of temporary files. Each item in the list is a file
198 # name (absolute path or relative to the current working directory).
199 # All files in this list will be deleted in the tearDown method. Note,
200 # this only helps to makes sure temporary files get deleted, but it
201 # does nothing about trying to close files that may still be open. It
202 # is the responsibility of the developer to properly close files even
203 # when exceptional conditions occur.
204 self.tempFiles = []
205
Brett Cannon19691362003-04-29 05:08:06 +0000206 # Create a temporary file.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000207 self.registerFileForCleanUp(support.TESTFN)
Guido van Rossuma0982942007-07-10 08:30:03 +0000208 self.text = b'testing urllib.urlretrieve'
Georg Brandl5a650a22005-08-26 08:51:34 +0000209 try:
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000210 FILE = open(support.TESTFN, 'wb')
Georg Brandl5a650a22005-08-26 08:51:34 +0000211 FILE.write(self.text)
212 FILE.close()
213 finally:
214 try: FILE.close()
215 except: pass
Brett Cannon19691362003-04-29 05:08:06 +0000216
217 def tearDown(self):
Georg Brandl5a650a22005-08-26 08:51:34 +0000218 # Delete the temporary files.
219 for each in self.tempFiles:
220 try: os.remove(each)
221 except: pass
222
223 def constructLocalFileUrl(self, filePath):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000224 return "file://%s" % urllib.request.pathname2url(
225 os.path.abspath(filePath))
Georg Brandl5a650a22005-08-26 08:51:34 +0000226
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000227 def createNewTempFile(self, data=b""):
Georg Brandl5a650a22005-08-26 08:51:34 +0000228 """Creates a new temporary file containing the specified data,
229 registers the file for deletion during the test fixture tear down, and
230 returns the absolute path of the file."""
231
232 newFd, newFilePath = tempfile.mkstemp()
233 try:
234 self.registerFileForCleanUp(newFilePath)
235 newFile = os.fdopen(newFd, "wb")
236 newFile.write(data)
237 newFile.close()
238 finally:
239 try: newFile.close()
240 except: pass
241 return newFilePath
242
243 def registerFileForCleanUp(self, fileName):
244 self.tempFiles.append(fileName)
Brett Cannon19691362003-04-29 05:08:06 +0000245
246 def test_basic(self):
247 # Make sure that a local file just gets its own location returned and
248 # a headers value is returned.
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000249 result = urllib.request.urlretrieve("file:%s" % support.TESTFN)
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000250 self.assertEqual(result[0], support.TESTFN)
Ezio Melottie9615932010-01-24 19:26:24 +0000251 self.assertIsInstance(result[1], email.message.Message,
252 "did not get a email.message.Message instance "
253 "as second returned value")
Brett Cannon19691362003-04-29 05:08:06 +0000254
255 def test_copy(self):
256 # Test that setting the filename argument works.
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000257 second_temp = "%s.2" % support.TESTFN
Georg Brandl5a650a22005-08-26 08:51:34 +0000258 self.registerFileForCleanUp(second_temp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000259 result = urllib.request.urlretrieve(self.constructLocalFileUrl(
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000260 support.TESTFN), second_temp)
Brett Cannon19691362003-04-29 05:08:06 +0000261 self.assertEqual(second_temp, result[0])
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000262 self.assertTrue(os.path.exists(second_temp), "copy of the file was not "
Brett Cannon19691362003-04-29 05:08:06 +0000263 "made")
Alex Martelli01c77c62006-08-24 02:58:11 +0000264 FILE = open(second_temp, 'rb')
Brett Cannon19691362003-04-29 05:08:06 +0000265 try:
266 text = FILE.read()
Brett Cannon19691362003-04-29 05:08:06 +0000267 FILE.close()
Georg Brandl5a650a22005-08-26 08:51:34 +0000268 finally:
269 try: FILE.close()
270 except: pass
Brett Cannon19691362003-04-29 05:08:06 +0000271 self.assertEqual(self.text, text)
272
273 def test_reporthook(self):
274 # Make sure that the reporthook works.
275 def hooktester(count, block_size, total_size, count_holder=[0]):
Ezio Melottie9615932010-01-24 19:26:24 +0000276 self.assertIsInstance(count, int)
277 self.assertIsInstance(block_size, int)
278 self.assertIsInstance(total_size, int)
Brett Cannon19691362003-04-29 05:08:06 +0000279 self.assertEqual(count, count_holder[0])
280 count_holder[0] = count_holder[0] + 1
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000281 second_temp = "%s.2" % support.TESTFN
Georg Brandl5a650a22005-08-26 08:51:34 +0000282 self.registerFileForCleanUp(second_temp)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000283 urllib.request.urlretrieve(
284 self.constructLocalFileUrl(support.TESTFN),
Georg Brandl5a650a22005-08-26 08:51:34 +0000285 second_temp, hooktester)
286
287 def test_reporthook_0_bytes(self):
288 # Test on zero length file. Should call reporthook only 1 time.
289 report = []
290 def hooktester(count, block_size, total_size, _report=report):
291 _report.append((count, block_size, total_size))
292 srcFileName = self.createNewTempFile()
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000293 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000294 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000295 self.assertEqual(len(report), 1)
296 self.assertEqual(report[0][2], 0)
297
298 def test_reporthook_5_bytes(self):
299 # Test on 5 byte file. Should call reporthook only 2 times (once when
300 # the "network connection" is established and once when the block is
301 # read). Since the block size is 8192 bytes, only one block read is
302 # required to read the entire file.
303 report = []
304 def hooktester(count, block_size, total_size, _report=report):
305 _report.append((count, block_size, total_size))
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000306 srcFileName = self.createNewTempFile(b"x" * 5)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000307 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000308 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000309 self.assertEqual(len(report), 2)
310 self.assertEqual(report[0][1], 8192)
311 self.assertEqual(report[0][2], 5)
312
313 def test_reporthook_8193_bytes(self):
314 # Test on 8193 byte file. Should call reporthook only 3 times (once
315 # when the "network connection" is established, once for the next 8192
316 # bytes, and once for the last byte).
317 report = []
318 def hooktester(count, block_size, total_size, _report=report):
319 _report.append((count, block_size, total_size))
Guido van Rossum70d0dda2007-08-29 01:53:26 +0000320 srcFileName = self.createNewTempFile(b"x" * 8193)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000321 urllib.request.urlretrieve(self.constructLocalFileUrl(srcFileName),
Benjamin Petersonee8712c2008-05-20 21:35:26 +0000322 support.TESTFN, hooktester)
Georg Brandl5a650a22005-08-26 08:51:34 +0000323 self.assertEqual(len(report), 3)
324 self.assertEqual(report[0][1], 8192)
325 self.assertEqual(report[0][2], 8193)
Skip Montanaro080c9972001-01-28 21:12:22 +0000326
Brett Cannon74bfd702003-04-25 09:39:47 +0000327class QuotingTests(unittest.TestCase):
328 """Tests for urllib.quote() and urllib.quote_plus()
Tim Petersc2659cf2003-05-12 20:19:37 +0000329
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000330 According to RFC 2396 (Uniform Resource Identifiers), to escape a
331 character you write it as '%' + <2 character US-ASCII hex value>.
332 The Python code of ``'%' + hex(ord(<character>))[2:]`` escapes a
333 character properly. Case does not matter on the hex letters.
Brett Cannon74bfd702003-04-25 09:39:47 +0000334
335 The various character sets specified are:
Tim Petersc2659cf2003-05-12 20:19:37 +0000336
Brett Cannon74bfd702003-04-25 09:39:47 +0000337 Reserved characters : ";/?:@&=+$,"
338 Have special meaning in URIs and must be escaped if not being used for
339 their special meaning
340 Data characters : letters, digits, and "-_.!~*'()"
341 Unreserved and do not need to be escaped; can be, though, if desired
342 Control characters : 0x00 - 0x1F, 0x7F
343 Have no use in URIs so must be escaped
344 space : 0x20
345 Must be escaped
346 Delimiters : '<>#%"'
347 Must be escaped
348 Unwise : "{}|\^[]`"
349 Must be escaped
Tim Petersc2659cf2003-05-12 20:19:37 +0000350
Brett Cannon74bfd702003-04-25 09:39:47 +0000351 """
352
353 def test_never_quote(self):
354 # Make sure quote() does not quote letters, digits, and "_,.-"
355 do_not_quote = '' .join(["ABCDEFGHIJKLMNOPQRSTUVWXYZ",
356 "abcdefghijklmnopqrstuvwxyz",
357 "0123456789",
358 "_.-"])
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000359 result = urllib.parse.quote(do_not_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000360 self.assertEqual(do_not_quote, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000361 "using quote(): %r != %r" % (do_not_quote, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000362 result = urllib.parse.quote_plus(do_not_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000363 self.assertEqual(do_not_quote, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000364 "using quote_plus(): %r != %r" % (do_not_quote, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000365
366 def test_default_safe(self):
367 # Test '/' is default value for 'safe' parameter
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000368 self.assertEqual(urllib.parse.quote.__defaults__[0], '/')
Brett Cannon74bfd702003-04-25 09:39:47 +0000369
370 def test_safe(self):
371 # Test setting 'safe' parameter does what it should do
372 quote_by_default = "<>"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000373 result = urllib.parse.quote(quote_by_default, safe=quote_by_default)
Brett Cannon74bfd702003-04-25 09:39:47 +0000374 self.assertEqual(quote_by_default, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000375 "using quote(): %r != %r" % (quote_by_default, result))
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000376 result = urllib.parse.quote_plus(quote_by_default,
377 safe=quote_by_default)
Brett Cannon74bfd702003-04-25 09:39:47 +0000378 self.assertEqual(quote_by_default, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000379 "using quote_plus(): %r != %r" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000380 (quote_by_default, result))
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000381 # Safe expressed as bytes rather than str
382 result = urllib.parse.quote(quote_by_default, safe=b"<>")
383 self.assertEqual(quote_by_default, result,
384 "using quote(): %r != %r" % (quote_by_default, result))
385 # "Safe" non-ASCII characters should have no effect
386 # (Since URIs are not allowed to have non-ASCII characters)
387 result = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="\xfc")
388 expect = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="")
389 self.assertEqual(expect, result,
390 "using quote(): %r != %r" %
391 (expect, result))
392 # Same as above, but using a bytes rather than str
393 result = urllib.parse.quote("a\xfcb", encoding="latin-1", safe=b"\xfc")
394 expect = urllib.parse.quote("a\xfcb", encoding="latin-1", safe="")
395 self.assertEqual(expect, result,
396 "using quote(): %r != %r" %
397 (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000398
399 def test_default_quoting(self):
400 # Make sure all characters that should be quoted are by default sans
401 # space (separate test for that).
402 should_quote = [chr(num) for num in range(32)] # For 0x00 - 0x1F
403 should_quote.append('<>#%"{}|\^[]`')
404 should_quote.append(chr(127)) # For 0x7F
405 should_quote = ''.join(should_quote)
406 for char in should_quote:
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000407 result = urllib.parse.quote(char)
Brett Cannon74bfd702003-04-25 09:39:47 +0000408 self.assertEqual(hexescape(char), result,
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000409 "using quote(): "
410 "%s should be escaped to %s, not %s" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000411 (char, hexescape(char), result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000412 result = urllib.parse.quote_plus(char)
Brett Cannon74bfd702003-04-25 09:39:47 +0000413 self.assertEqual(hexescape(char), result,
414 "using quote_plus(): "
Tim Petersc2659cf2003-05-12 20:19:37 +0000415 "%s should be escapes to %s, not %s" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000416 (char, hexescape(char), result))
417 del should_quote
418 partial_quote = "ab[]cd"
419 expected = "ab%5B%5Dcd"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000420 result = urllib.parse.quote(partial_quote)
Brett Cannon74bfd702003-04-25 09:39:47 +0000421 self.assertEqual(expected, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000422 "using quote(): %r != %r" % (expected, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000423 self.assertEqual(expected, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000424 "using quote_plus(): %r != %r" % (expected, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000425
426 def test_quoting_space(self):
427 # Make sure quote() and quote_plus() handle spaces as specified in
428 # their unique way
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000429 result = urllib.parse.quote(' ')
Brett Cannon74bfd702003-04-25 09:39:47 +0000430 self.assertEqual(result, hexescape(' '),
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000431 "using quote(): %r != %r" % (result, hexescape(' ')))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000432 result = urllib.parse.quote_plus(' ')
Brett Cannon74bfd702003-04-25 09:39:47 +0000433 self.assertEqual(result, '+',
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000434 "using quote_plus(): %r != +" % result)
Brett Cannon74bfd702003-04-25 09:39:47 +0000435 given = "a b cd e f"
436 expect = given.replace(' ', hexescape(' '))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000437 result = urllib.parse.quote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000438 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000439 "using quote(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000440 expect = given.replace(' ', '+')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000441 result = urllib.parse.quote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000442 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000443 "using quote_plus(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000444
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000445 def test_quoting_plus(self):
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000446 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma'),
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000447 'alpha%2Bbeta+gamma')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000448 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma', '+'),
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000449 'alpha+beta+gamma')
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000450 # Test with bytes
451 self.assertEqual(urllib.parse.quote_plus(b'alpha+beta gamma'),
452 'alpha%2Bbeta+gamma')
453 # Test with safe bytes
454 self.assertEqual(urllib.parse.quote_plus('alpha+beta gamma', b'+'),
455 'alpha+beta+gamma')
456
457 def test_quote_bytes(self):
458 # Bytes should quote directly to percent-encoded values
459 given = b"\xa2\xd8ab\xff"
460 expect = "%A2%D8ab%FF"
461 result = urllib.parse.quote(given)
462 self.assertEqual(expect, result,
463 "using quote(): %r != %r" % (expect, result))
464 # Encoding argument should raise type error on bytes input
465 self.assertRaises(TypeError, urllib.parse.quote, given,
466 encoding="latin-1")
467 # quote_from_bytes should work the same
468 result = urllib.parse.quote_from_bytes(given)
469 self.assertEqual(expect, result,
470 "using quote_from_bytes(): %r != %r"
471 % (expect, result))
472
473 def test_quote_with_unicode(self):
474 # Characters in Latin-1 range, encoded by default in UTF-8
475 given = "\xa2\xd8ab\xff"
476 expect = "%C2%A2%C3%98ab%C3%BF"
477 result = urllib.parse.quote(given)
478 self.assertEqual(expect, result,
479 "using quote(): %r != %r" % (expect, result))
480 # Characters in Latin-1 range, encoded by with None (default)
481 result = urllib.parse.quote(given, encoding=None, errors=None)
482 self.assertEqual(expect, result,
483 "using quote(): %r != %r" % (expect, result))
484 # Characters in Latin-1 range, encoded with Latin-1
485 given = "\xa2\xd8ab\xff"
486 expect = "%A2%D8ab%FF"
487 result = urllib.parse.quote(given, encoding="latin-1")
488 self.assertEqual(expect, result,
489 "using quote(): %r != %r" % (expect, result))
490 # Characters in BMP, encoded by default in UTF-8
491 given = "\u6f22\u5b57" # "Kanji"
492 expect = "%E6%BC%A2%E5%AD%97"
493 result = urllib.parse.quote(given)
494 self.assertEqual(expect, result,
495 "using quote(): %r != %r" % (expect, result))
496 # Characters in BMP, encoded with Latin-1
497 given = "\u6f22\u5b57"
498 self.assertRaises(UnicodeEncodeError, urllib.parse.quote, given,
499 encoding="latin-1")
500 # Characters in BMP, encoded with Latin-1, with replace error handling
501 given = "\u6f22\u5b57"
502 expect = "%3F%3F" # "??"
503 result = urllib.parse.quote(given, encoding="latin-1",
504 errors="replace")
505 self.assertEqual(expect, result,
506 "using quote(): %r != %r" % (expect, result))
507 # Characters in BMP, Latin-1, with xmlcharref error handling
508 given = "\u6f22\u5b57"
509 expect = "%26%2328450%3B%26%2323383%3B" # "&#28450;&#23383;"
510 result = urllib.parse.quote(given, encoding="latin-1",
511 errors="xmlcharrefreplace")
512 self.assertEqual(expect, result,
513 "using quote(): %r != %r" % (expect, result))
Raymond Hettinger2bdec7b2005-09-10 14:30:09 +0000514
Georg Brandlfaf41492009-05-26 18:31:11 +0000515 def test_quote_plus_with_unicode(self):
516 # Encoding (latin-1) test for quote_plus
517 given = "\xa2\xd8 \xff"
518 expect = "%A2%D8+%FF"
519 result = urllib.parse.quote_plus(given, encoding="latin-1")
520 self.assertEqual(expect, result,
521 "using quote_plus(): %r != %r" % (expect, result))
522 # Errors test for quote_plus
523 given = "ab\u6f22\u5b57 cd"
524 expect = "ab%3F%3F+cd"
525 result = urllib.parse.quote_plus(given, encoding="latin-1",
526 errors="replace")
527 self.assertEqual(expect, result,
528 "using quote_plus(): %r != %r" % (expect, result))
529
Senthil Kumarand496c4c2010-07-30 19:34:36 +0000530
Brett Cannon74bfd702003-04-25 09:39:47 +0000531class UnquotingTests(unittest.TestCase):
532 """Tests for unquote() and unquote_plus()
Tim Petersc2659cf2003-05-12 20:19:37 +0000533
Brett Cannon74bfd702003-04-25 09:39:47 +0000534 See the doc string for quoting_Tests for details on quoting and such.
535
536 """
537
538 def test_unquoting(self):
539 # Make sure unquoting of all ASCII values works
540 escape_list = []
541 for num in range(128):
542 given = hexescape(chr(num))
543 expect = chr(num)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000544 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000545 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000546 "using unquote(): %r != %r" % (expect, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000547 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000548 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000549 "using unquote_plus(): %r != %r" %
Brett Cannon74bfd702003-04-25 09:39:47 +0000550 (expect, result))
551 escape_list.append(given)
552 escape_string = ''.join(escape_list)
553 del escape_list
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000554 result = urllib.parse.unquote(escape_string)
Brett Cannon74bfd702003-04-25 09:39:47 +0000555 self.assertEqual(result.count('%'), 1,
Brett Cannon74bfd702003-04-25 09:39:47 +0000556 "using unquote(): not all characters escaped: "
557 "%s" % result)
Senthil Kumaran79e17f62010-07-19 18:17:19 +0000558 self.assertRaises(TypeError, urllib.parse.unquote, None)
Senthil Kumarand496c4c2010-07-30 19:34:36 +0000559 self.assertRaises(TypeError, urllib.parse.unquote, ())
Brett Cannon74bfd702003-04-25 09:39:47 +0000560
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000561 def test_unquoting_badpercent(self):
562 # Test unquoting on bad percent-escapes
563 given = '%xab'
564 expect = given
565 result = urllib.parse.unquote(given)
566 self.assertEqual(expect, result, "using unquote(): %r != %r"
567 % (expect, result))
568 given = '%x'
569 expect = given
570 result = urllib.parse.unquote(given)
571 self.assertEqual(expect, result, "using unquote(): %r != %r"
572 % (expect, result))
573 given = '%'
574 expect = given
575 result = urllib.parse.unquote(given)
576 self.assertEqual(expect, result, "using unquote(): %r != %r"
577 % (expect, result))
578 # unquote_to_bytes
579 given = '%xab'
580 expect = bytes(given, 'ascii')
581 result = urllib.parse.unquote_to_bytes(given)
582 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
583 % (expect, result))
584 given = '%x'
585 expect = bytes(given, 'ascii')
586 result = urllib.parse.unquote_to_bytes(given)
587 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
588 % (expect, result))
589 given = '%'
590 expect = bytes(given, 'ascii')
591 result = urllib.parse.unquote_to_bytes(given)
592 self.assertEqual(expect, result, "using unquote_to_bytes(): %r != %r"
593 % (expect, result))
Senthil Kumaran79e17f62010-07-19 18:17:19 +0000594 self.assertRaises(TypeError, urllib.parse.unquote_to_bytes, None)
Senthil Kumarand496c4c2010-07-30 19:34:36 +0000595 self.assertRaises(TypeError, urllib.parse.unquote_to_bytes, ())
Senthil Kumaran79e17f62010-07-19 18:17:19 +0000596
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000597 def test_unquoting_mixed_case(self):
598 # Test unquoting on mixed-case hex digits in the percent-escapes
599 given = '%Ab%eA'
600 expect = b'\xab\xea'
601 result = urllib.parse.unquote_to_bytes(given)
602 self.assertEqual(expect, result,
603 "using unquote_to_bytes(): %r != %r"
604 % (expect, result))
605
Brett Cannon74bfd702003-04-25 09:39:47 +0000606 def test_unquoting_parts(self):
607 # Make sure unquoting works when have non-quoted characters
608 # interspersed
609 given = 'ab%sd' % hexescape('c')
610 expect = "abcd"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000611 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000612 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000613 "using quote(): %r != %r" % (expect, result))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000614 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000615 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000616 "using unquote_plus(): %r != %r" % (expect, result))
Tim Petersc2659cf2003-05-12 20:19:37 +0000617
Brett Cannon74bfd702003-04-25 09:39:47 +0000618 def test_unquoting_plus(self):
619 # Test difference between unquote() and unquote_plus()
620 given = "are+there+spaces..."
621 expect = given
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000622 result = urllib.parse.unquote(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000623 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000624 "using unquote(): %r != %r" % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000625 expect = given.replace('+', ' ')
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000626 result = urllib.parse.unquote_plus(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000627 self.assertEqual(expect, result,
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000628 "using unquote_plus(): %r != %r" % (expect, result))
629
630 def test_unquote_to_bytes(self):
631 given = 'br%C3%BCckner_sapporo_20050930.doc'
632 expect = b'br\xc3\xbcckner_sapporo_20050930.doc'
633 result = urllib.parse.unquote_to_bytes(given)
634 self.assertEqual(expect, result,
635 "using unquote_to_bytes(): %r != %r"
636 % (expect, result))
637 # Test on a string with unescaped non-ASCII characters
638 # (Technically an invalid URI; expect those characters to be UTF-8
639 # encoded).
640 result = urllib.parse.unquote_to_bytes("\u6f22%C3%BC")
641 expect = b'\xe6\xbc\xa2\xc3\xbc' # UTF-8 for "\u6f22\u00fc"
642 self.assertEqual(expect, result,
643 "using unquote_to_bytes(): %r != %r"
644 % (expect, result))
645 # Test with a bytes as input
646 given = b'%A2%D8ab%FF'
647 expect = b'\xa2\xd8ab\xff'
648 result = urllib.parse.unquote_to_bytes(given)
649 self.assertEqual(expect, result,
650 "using unquote_to_bytes(): %r != %r"
651 % (expect, result))
652 # Test with a bytes as input, with unescaped non-ASCII bytes
653 # (Technically an invalid URI; expect those bytes to be preserved)
654 given = b'%A2\xd8ab%FF'
655 expect = b'\xa2\xd8ab\xff'
656 result = urllib.parse.unquote_to_bytes(given)
657 self.assertEqual(expect, result,
658 "using unquote_to_bytes(): %r != %r"
659 % (expect, result))
Brett Cannon74bfd702003-04-25 09:39:47 +0000660
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +0000661 def test_unquote_with_unicode(self):
Guido van Rossum52dbbb92008-08-18 21:44:30 +0000662 # Characters in the Latin-1 range, encoded with UTF-8
663 given = 'br%C3%BCckner_sapporo_20050930.doc'
664 expect = 'br\u00fcckner_sapporo_20050930.doc'
665 result = urllib.parse.unquote(given)
666 self.assertEqual(expect, result,
667 "using unquote(): %r != %r" % (expect, result))
668 # Characters in the Latin-1 range, encoded with None (default)
669 result = urllib.parse.unquote(given, encoding=None, errors=None)
670 self.assertEqual(expect, result,
671 "using unquote(): %r != %r" % (expect, result))
672
673 # Characters in the Latin-1 range, encoded with Latin-1
674 result = urllib.parse.unquote('br%FCckner_sapporo_20050930.doc',
675 encoding="latin-1")
676 expect = 'br\u00fcckner_sapporo_20050930.doc'
677 self.assertEqual(expect, result,
678 "using unquote(): %r != %r" % (expect, result))
679
680 # Characters in BMP, encoded with UTF-8
681 given = "%E6%BC%A2%E5%AD%97"
682 expect = "\u6f22\u5b57" # "Kanji"
683 result = urllib.parse.unquote(given)
684 self.assertEqual(expect, result,
685 "using unquote(): %r != %r" % (expect, result))
686
687 # Decode with UTF-8, invalid sequence
688 given = "%F3%B1"
689 expect = "\ufffd" # Replacement character
690 result = urllib.parse.unquote(given)
691 self.assertEqual(expect, result,
692 "using unquote(): %r != %r" % (expect, result))
693
694 # Decode with UTF-8, invalid sequence, replace errors
695 result = urllib.parse.unquote(given, errors="replace")
696 self.assertEqual(expect, result,
697 "using unquote(): %r != %r" % (expect, result))
698
699 # Decode with UTF-8, invalid sequence, ignoring errors
700 given = "%F3%B1"
701 expect = ""
702 result = urllib.parse.unquote(given, errors="ignore")
703 self.assertEqual(expect, result,
704 "using unquote(): %r != %r" % (expect, result))
705
706 # A mix of non-ASCII and percent-encoded characters, UTF-8
707 result = urllib.parse.unquote("\u6f22%C3%BC")
708 expect = '\u6f22\u00fc'
709 self.assertEqual(expect, result,
710 "using unquote(): %r != %r" % (expect, result))
711
712 # A mix of non-ASCII and percent-encoded characters, Latin-1
713 # (Note, the string contains non-Latin-1-representable characters)
714 result = urllib.parse.unquote("\u6f22%FC", encoding="latin-1")
715 expect = '\u6f22\u00fc'
716 self.assertEqual(expect, result,
717 "using unquote(): %r != %r" % (expect, result))
Raymond Hettinger4b0f20d2005-10-15 16:41:53 +0000718
Brett Cannon74bfd702003-04-25 09:39:47 +0000719class urlencode_Tests(unittest.TestCase):
720 """Tests for urlencode()"""
721
722 def help_inputtype(self, given, test_type):
723 """Helper method for testing different input types.
Tim Petersc2659cf2003-05-12 20:19:37 +0000724
Brett Cannon74bfd702003-04-25 09:39:47 +0000725 'given' must lead to only the pairs:
726 * 1st, 1
727 * 2nd, 2
728 * 3rd, 3
Tim Petersc2659cf2003-05-12 20:19:37 +0000729
Brett Cannon74bfd702003-04-25 09:39:47 +0000730 Test cannot assume anything about order. Docs make no guarantee and
731 have possible dictionary input.
Tim Petersc2659cf2003-05-12 20:19:37 +0000732
Brett Cannon74bfd702003-04-25 09:39:47 +0000733 """
734 expect_somewhere = ["1st=1", "2nd=2", "3rd=3"]
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000735 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000736 for expected in expect_somewhere:
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000737 self.assertIn(expected, result,
Brett Cannon74bfd702003-04-25 09:39:47 +0000738 "testing %s: %s not found in %s" %
739 (test_type, expected, result))
740 self.assertEqual(result.count('&'), 2,
741 "testing %s: expected 2 '&'s; got %s" %
742 (test_type, result.count('&')))
743 amp_location = result.index('&')
744 on_amp_left = result[amp_location - 1]
745 on_amp_right = result[amp_location + 1]
Benjamin Petersonc9c0f202009-06-30 23:06:06 +0000746 self.assertTrue(on_amp_left.isdigit() and on_amp_right.isdigit(),
Brett Cannon74bfd702003-04-25 09:39:47 +0000747 "testing %s: '&' not located in proper place in %s" %
748 (test_type, result))
749 self.assertEqual(len(result), (5 * 3) + 2, #5 chars per thing and amps
750 "testing %s: "
751 "unexpected number of characters: %s != %s" %
752 (test_type, len(result), (5 * 3) + 2))
753
754 def test_using_mapping(self):
755 # Test passing in a mapping object as an argument.
756 self.help_inputtype({"1st":'1', "2nd":'2', "3rd":'3'},
757 "using dict as input type")
758
759 def test_using_sequence(self):
760 # Test passing in a sequence of two-item sequences as an argument.
761 self.help_inputtype([('1st', '1'), ('2nd', '2'), ('3rd', '3')],
762 "using sequence of two-item tuples as input")
763
764 def test_quoting(self):
765 # Make sure keys and values are quoted using quote_plus()
766 given = {"&":"="}
767 expect = "%s=%s" % (hexescape('&'), hexescape('='))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000768 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000769 self.assertEqual(expect, result)
770 given = {"key name":"A bunch of pluses"}
771 expect = "key+name=A+bunch+of+pluses"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000772 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000773 self.assertEqual(expect, result)
774
775 def test_doseq(self):
776 # Test that passing True for 'doseq' parameter works correctly
777 given = {'sequence':['1', '2', '3']}
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000778 expect = "sequence=%s" % urllib.parse.quote_plus(str(['1', '2', '3']))
779 result = urllib.parse.urlencode(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000780 self.assertEqual(expect, result)
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000781 result = urllib.parse.urlencode(given, True)
Brett Cannon74bfd702003-04-25 09:39:47 +0000782 for value in given["sequence"]:
783 expect = "sequence=%s" % value
Ezio Melottib58e0bd2010-01-23 15:40:09 +0000784 self.assertIn(expect, result)
Brett Cannon74bfd702003-04-25 09:39:47 +0000785 self.assertEqual(result.count('&'), 2,
786 "Expected 2 '&'s, got %s" % result.count('&'))
787
Jeremy Hylton1ef7c6b2009-03-26 16:57:30 +0000788 def test_empty_sequence(self):
789 self.assertEqual("", urllib.parse.urlencode({}))
790 self.assertEqual("", urllib.parse.urlencode([]))
791
792 def test_nonstring_values(self):
793 self.assertEqual("a=1", urllib.parse.urlencode({"a": 1}))
794 self.assertEqual("a=None", urllib.parse.urlencode({"a": None}))
795
796 def test_nonstring_seq_values(self):
797 self.assertEqual("a=1&a=2", urllib.parse.urlencode({"a": [1, 2]}, True))
798 self.assertEqual("a=None&a=a",
799 urllib.parse.urlencode({"a": [None, "a"]}, True))
800 self.assertEqual("a=a&a=b",
801 urllib.parse.urlencode({"a": {"a": 1, "b": 1}}, True))
802
Senthil Kumarandf022da2010-07-03 17:48:22 +0000803 def test_urlencode_encoding(self):
804 # ASCII encoding. Expect %3F with errors="replace'
805 given = (('\u00a0', '\u00c1'),)
806 expect = '%3F=%3F'
807 result = urllib.parse.urlencode(given, encoding="ASCII", errors="replace")
808 self.assertEqual(expect, result)
809
810 # Default is UTF-8 encoding.
811 given = (('\u00a0', '\u00c1'),)
812 expect = '%C2%A0=%C3%81'
813 result = urllib.parse.urlencode(given)
814 self.assertEqual(expect, result)
815
816 # Latin-1 encoding.
817 given = (('\u00a0', '\u00c1'),)
818 expect = '%A0=%C1'
819 result = urllib.parse.urlencode(given, encoding="latin-1")
820 self.assertEqual(expect, result)
821
822 def test_urlencode_encoding_doseq(self):
823 # ASCII Encoding. Expect %3F with errors="replace'
824 given = (('\u00a0', '\u00c1'),)
825 expect = '%3F=%3F'
826 result = urllib.parse.urlencode(given, doseq=True,
827 encoding="ASCII", errors="replace")
828 self.assertEqual(expect, result)
829
830 # ASCII Encoding. On a sequence of values.
831 given = (("\u00a0", (1, "\u00c1")),)
832 expect = '%3F=1&%3F=%3F'
833 result = urllib.parse.urlencode(given, True,
834 encoding="ASCII", errors="replace")
835 self.assertEqual(expect, result)
836
837 # Utf-8
838 given = (("\u00a0", "\u00c1"),)
839 expect = '%C2%A0=%C3%81'
840 result = urllib.parse.urlencode(given, True)
841 self.assertEqual(expect, result)
842
843 given = (("\u00a0", (42, "\u00c1")),)
844 expect = '%C2%A0=42&%C2%A0=%C3%81'
845 result = urllib.parse.urlencode(given, True)
846 self.assertEqual(expect, result)
847
848 # latin-1
849 given = (("\u00a0", "\u00c1"),)
850 expect = '%A0=%C1'
851 result = urllib.parse.urlencode(given, True, encoding="latin-1")
852 self.assertEqual(expect, result)
853
854 given = (("\u00a0", (42, "\u00c1")),)
855 expect = '%A0=42&%A0=%C1'
856 result = urllib.parse.urlencode(given, True, encoding="latin-1")
857 self.assertEqual(expect, result)
858
859 def test_urlencode_bytes(self):
860 given = ((b'\xa0\x24', b'\xc1\x24'),)
861 expect = '%A0%24=%C1%24'
862 result = urllib.parse.urlencode(given)
863 self.assertEqual(expect, result)
864 result = urllib.parse.urlencode(given, True)
865 self.assertEqual(expect, result)
866
867 # Sequence of values
868 given = ((b'\xa0\x24', (42, b'\xc1\x24')),)
869 expect = '%A0%24=42&%A0%24=%C1%24'
870 result = urllib.parse.urlencode(given, True)
871 self.assertEqual(expect, result)
872
873 def test_urlencode_encoding_safe_parameter(self):
874
875 # Send '$' (\x24) as safe character
876 # Default utf-8 encoding
877
878 given = ((b'\xa0\x24', b'\xc1\x24'),)
879 result = urllib.parse.urlencode(given, safe=":$")
880 expect = '%A0$=%C1$'
881 self.assertEqual(expect, result)
882
883 given = ((b'\xa0\x24', b'\xc1\x24'),)
884 result = urllib.parse.urlencode(given, doseq=True, safe=":$")
885 expect = '%A0$=%C1$'
886 self.assertEqual(expect, result)
887
888 # Safe parameter in sequence
889 given = ((b'\xa0\x24', (b'\xc1\x24', 0xd, 42)),)
890 expect = '%A0$=%C1$&%A0$=13&%A0$=42'
891 result = urllib.parse.urlencode(given, True, safe=":$")
892 self.assertEqual(expect, result)
893
894 # Test all above in latin-1 encoding
895
896 given = ((b'\xa0\x24', b'\xc1\x24'),)
897 result = urllib.parse.urlencode(given, safe=":$",
898 encoding="latin-1")
899 expect = '%A0$=%C1$'
900 self.assertEqual(expect, result)
901
902 given = ((b'\xa0\x24', b'\xc1\x24'),)
903 expect = '%A0$=%C1$'
904 result = urllib.parse.urlencode(given, doseq=True, safe=":$",
905 encoding="latin-1")
906
907 given = ((b'\xa0\x24', (b'\xc1\x24', 0xd, 42)),)
908 expect = '%A0$=%C1$&%A0$=13&%A0$=42'
909 result = urllib.parse.urlencode(given, True, safe=":$",
910 encoding="latin-1")
911 self.assertEqual(expect, result)
912
Brett Cannon74bfd702003-04-25 09:39:47 +0000913class Pathname_Tests(unittest.TestCase):
914 """Test pathname2url() and url2pathname()"""
915
916 def test_basic(self):
917 # Make sure simple tests pass
918 expected_path = os.path.join("parts", "of", "a", "path")
919 expected_url = "parts/of/a/path"
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000920 result = urllib.request.pathname2url(expected_path)
Brett Cannon74bfd702003-04-25 09:39:47 +0000921 self.assertEqual(expected_url, result,
922 "pathname2url() failed; %s != %s" %
923 (result, expected_url))
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000924 result = urllib.request.url2pathname(expected_url)
Brett Cannon74bfd702003-04-25 09:39:47 +0000925 self.assertEqual(expected_path, result,
926 "url2pathame() failed; %s != %s" %
927 (result, expected_path))
928
929 def test_quoting(self):
930 # Test automatic quoting and unquoting works for pathnam2url() and
931 # url2pathname() respectively
932 given = os.path.join("needs", "quot=ing", "here")
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000933 expect = "needs/%s/here" % urllib.parse.quote("quot=ing")
934 result = urllib.request.pathname2url(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000935 self.assertEqual(expect, result,
936 "pathname2url() failed; %s != %s" %
937 (expect, result))
938 expect = given
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000939 result = urllib.request.url2pathname(result)
Brett Cannon74bfd702003-04-25 09:39:47 +0000940 self.assertEqual(expect, result,
941 "url2pathname() failed; %s != %s" %
942 (expect, result))
943 given = os.path.join("make sure", "using_quote")
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000944 expect = "%s/using_quote" % urllib.parse.quote("make sure")
945 result = urllib.request.pathname2url(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000946 self.assertEqual(expect, result,
947 "pathname2url() failed; %s != %s" %
948 (expect, result))
949 given = "make+sure/using_unquote"
950 expect = os.path.join("make+sure", "using_unquote")
Jeremy Hylton1afc1692008-06-18 20:49:58 +0000951 result = urllib.request.url2pathname(given)
Brett Cannon74bfd702003-04-25 09:39:47 +0000952 self.assertEqual(expect, result,
953 "url2pathname() failed; %s != %s" %
954 (expect, result))
Tim Petersc2659cf2003-05-12 20:19:37 +0000955
Senthil Kumaraneaaec272009-03-30 21:54:41 +0000956class Utility_Tests(unittest.TestCase):
957 """Testcase to test the various utility functions in the urllib."""
958
959 def test_splitpasswd(self):
960 """Some of password examples are not sensible, but it is added to
961 confirming to RFC2617 and addressing issue4675.
962 """
963 self.assertEqual(('user', 'ab'),urllib.parse.splitpasswd('user:ab'))
964 self.assertEqual(('user', 'a\nb'),urllib.parse.splitpasswd('user:a\nb'))
965 self.assertEqual(('user', 'a\tb'),urllib.parse.splitpasswd('user:a\tb'))
966 self.assertEqual(('user', 'a\rb'),urllib.parse.splitpasswd('user:a\rb'))
967 self.assertEqual(('user', 'a\fb'),urllib.parse.splitpasswd('user:a\fb'))
968 self.assertEqual(('user', 'a\vb'),urllib.parse.splitpasswd('user:a\vb'))
969 self.assertEqual(('user', 'a:b'),urllib.parse.splitpasswd('user:a:b'))
970
Senthil Kumaran690ce9b2009-05-05 18:41:13 +0000971
972class URLopener_Tests(unittest.TestCase):
973 """Testcase to test the open method of URLopener class."""
974
975 def test_quoted_open(self):
976 class DummyURLopener(urllib.request.URLopener):
977 def open_spam(self, url):
978 return url
979
980 self.assertEqual(DummyURLopener().open(
981 'spam://example/ /'),'//example/%20/')
982
Senthil Kumaran734f0592010-02-20 22:19:04 +0000983 # test the safe characters are not quoted by urlopen
984 self.assertEqual(DummyURLopener().open(
985 "spam://c:|windows%/:=&?~#+!$,;'@()*[]|/path/"),
986 "//c:|windows%/:=&?~#+!$,;'@()*[]|/path/")
987
Guido van Rossume7ba4952007-06-06 23:52:48 +0000988# Just commented them out.
989# Can't really tell why keep failing in windows and sparc.
990# Everywhere else they work ok, but on those machines, someteimes
991# fail in one of the tests, sometimes in other. I have a linux, and
992# the tests go ok.
993# If anybody has one of the problematic enviroments, please help!
994# . Facundo
995#
996# def server(evt):
Georg Brandlf78e02b2008-06-10 17:40:04 +0000997# import socket, time
Guido van Rossume7ba4952007-06-06 23:52:48 +0000998# serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
999# serv.settimeout(3)
1000# serv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
1001# serv.bind(("", 9093))
1002# serv.listen(5)
1003# try:
1004# conn, addr = serv.accept()
1005# conn.send("1 Hola mundo\n")
1006# cantdata = 0
1007# while cantdata < 13:
1008# data = conn.recv(13-cantdata)
1009# cantdata += len(data)
1010# time.sleep(.3)
1011# conn.send("2 No more lines\n")
1012# conn.close()
1013# except socket.timeout:
1014# pass
1015# finally:
1016# serv.close()
1017# evt.set()
1018#
1019# class FTPWrapperTests(unittest.TestCase):
1020#
1021# def setUp(self):
Georg Brandlf78e02b2008-06-10 17:40:04 +00001022# import ftplib, time, threading
Guido van Rossume7ba4952007-06-06 23:52:48 +00001023# ftplib.FTP.port = 9093
1024# self.evt = threading.Event()
1025# threading.Thread(target=server, args=(self.evt,)).start()
1026# time.sleep(.1)
1027#
1028# def tearDown(self):
1029# self.evt.wait()
1030#
1031# def testBasic(self):
1032# # connects
1033# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
Georg Brandlf78e02b2008-06-10 17:40:04 +00001034# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001035#
1036# def testTimeoutNone(self):
Georg Brandlf78e02b2008-06-10 17:40:04 +00001037# # global default timeout is ignored
1038# import socket
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001039# self.assertTrue(socket.getdefaulttimeout() is None)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001040# socket.setdefaulttimeout(30)
1041# try:
1042# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
1043# finally:
Georg Brandlf78e02b2008-06-10 17:40:04 +00001044# socket.setdefaulttimeout(None)
Guido van Rossume7ba4952007-06-06 23:52:48 +00001045# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
Georg Brandlf78e02b2008-06-10 17:40:04 +00001046# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001047#
Georg Brandlf78e02b2008-06-10 17:40:04 +00001048# def testTimeoutDefault(self):
1049# # global default timeout is used
1050# import socket
Benjamin Petersonc9c0f202009-06-30 23:06:06 +00001051# self.assertTrue(socket.getdefaulttimeout() is None)
Georg Brandlf78e02b2008-06-10 17:40:04 +00001052# socket.setdefaulttimeout(30)
1053# try:
1054# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [])
1055# finally:
1056# socket.setdefaulttimeout(None)
1057# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
1058# ftp.close()
1059#
1060# def testTimeoutValue(self):
1061# ftp = urllib.ftpwrapper("myuser", "mypass", "localhost", 9093, [],
1062# timeout=30)
1063# self.assertEqual(ftp.ftp.sock.gettimeout(), 30)
1064# ftp.close()
Guido van Rossume7ba4952007-06-06 23:52:48 +00001065
Skip Montanaro080c9972001-01-28 21:12:22 +00001066
1067
Brett Cannon74bfd702003-04-25 09:39:47 +00001068def test_main():
Benjamin Petersonee8712c2008-05-20 21:35:26 +00001069 support.run_unittest(
Walter Dörwald21d3a322003-05-01 17:45:56 +00001070 urlopen_FileTests,
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001071 urlopen_HttpTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001072 urlretrieve_FileTests,
Benjamin Peterson9bc93512008-09-22 22:10:59 +00001073 ProxyTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001074 QuotingTests,
1075 UnquotingTests,
1076 urlencode_Tests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00001077 Pathname_Tests,
Senthil Kumaraneaaec272009-03-30 21:54:41 +00001078 Utility_Tests,
Senthil Kumaran690ce9b2009-05-05 18:41:13 +00001079 URLopener_Tests,
Guido van Rossume7ba4952007-06-06 23:52:48 +00001080 #FTPWrapperTests,
Walter Dörwald21d3a322003-05-01 17:45:56 +00001081 )
Brett Cannon74bfd702003-04-25 09:39:47 +00001082
1083
1084
1085if __name__ == '__main__':
1086 test_main()