Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 1 | """Unit tests for code in urllib.response.""" |
| 2 | |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 3 | import socket |
| 4 | import tempfile |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 5 | import urllib.response |
| 6 | import unittest |
| 7 | |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 8 | class TestResponse(unittest.TestCase): |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 9 | |
| 10 | def setUp(self): |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 11 | self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) |
| 12 | self.fp = self.sock.makefile('rb') |
| 13 | self.test_headers = {"Host": "www.python.org", |
| 14 | "Connection": "close"} |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 15 | |
| 16 | def test_with(self): |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 17 | addbase = urllib.response.addbase(self.fp) |
| 18 | |
| 19 | self.assertIsInstance(addbase, tempfile._TemporaryFileWrapper) |
| 20 | |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 21 | def f(): |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 22 | with addbase as spam: |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 23 | pass |
| 24 | self.assertFalse(self.fp.closed) |
| 25 | f() |
| 26 | self.assertTrue(self.fp.closed) |
| 27 | self.assertRaises(ValueError, f) |
| 28 | |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 29 | def test_addclosehook(self): |
| 30 | closehook_called = False |
| 31 | |
| 32 | def closehook(): |
| 33 | nonlocal closehook_called |
| 34 | closehook_called = True |
| 35 | |
| 36 | closehook = urllib.response.addclosehook(self.fp, closehook) |
| 37 | closehook.close() |
| 38 | |
| 39 | self.assertTrue(self.fp.closed) |
| 40 | self.assertTrue(closehook_called) |
| 41 | |
| 42 | def test_addinfo(self): |
| 43 | info = urllib.response.addinfo(self.fp, self.test_headers) |
| 44 | self.assertEqual(info.info(), self.test_headers) |
| 45 | |
| 46 | def test_addinfourl(self): |
| 47 | url = "http://www.python.org" |
| 48 | code = 200 |
| 49 | infourl = urllib.response.addinfourl(self.fp, self.test_headers, |
| 50 | url, code) |
| 51 | self.assertEqual(infourl.info(), self.test_headers) |
| 52 | self.assertEqual(infourl.geturl(), url) |
| 53 | self.assertEqual(infourl.getcode(), code) |
| 54 | |
| 55 | def tearDown(self): |
| 56 | self.sock.close() |
Jeremy Hylton | b476d59 | 2009-03-26 21:34:20 +0000 | [diff] [blame] | 57 | |
| 58 | if __name__ == '__main__': |
Senthil Kumaran | 6117e5d | 2014-04-20 09:41:29 -0700 | [diff] [blame] | 59 | unittest.main() |