Mercurial > cpython-lru_cache-weakref
diff Lib/test/test_httpservers.py @ 65740:40ac45a092b7
Move test_httpservers over to file context managers.
| author | Brett Cannon <bcannon@gmail.com> |
|---|---|
| date | Fri, 29 Oct 2010 23:43:42 +0000 |
| parents | c147f910af70 |
| children | 211858723b8c |
line wrap: on
line diff
--- a/Lib/test/test_httpservers.py Fri Oct 29 23:41:37 2010 +0000 +++ b/Lib/test/test_httpservers.py Fri Oct 29 23:43:42 2010 +0000 @@ -206,9 +206,8 @@ self.data = b'We are the knights who say Ni!' self.tempdir = tempfile.mkdtemp(dir=basetempdir) self.tempdir_name = os.path.basename(self.tempdir) - temp = open(os.path.join(self.tempdir, 'test'), 'wb') - temp.write(self.data) - temp.close() + with open(os.path.join(self.tempdir, 'test'), 'wb') as temp: + temp.write(self.data) def tearDown(self): try: @@ -240,15 +239,15 @@ self.check_status_and_reason(response, 404) response = self.request('/' + 'ThisDoesNotExist' + '/') self.check_status_and_reason(response, 404) - f = open(os.path.join(self.tempdir_name, 'index.html'), 'w') - response = self.request('/' + self.tempdir_name + '/') - self.check_status_and_reason(response, 200) - if os.name == 'posix': - # chmod won't work as expected on Windows platforms - os.chmod(self.tempdir, 0) - response = self.request(self.tempdir_name + '/') - self.check_status_and_reason(response, 404) - os.chmod(self.tempdir, 0o755) + with open(os.path.join(self.tempdir_name, 'index.html'), 'w') as f: + response = self.request('/' + self.tempdir_name + '/') + self.check_status_and_reason(response, 200) + if os.name == 'posix': + # chmod won't work as expected on Windows platforms + os.chmod(self.tempdir, 0) + response = self.request(self.tempdir_name + '/') + self.check_status_and_reason(response, 404) + os.chmod(self.tempdir, 0o755) def test_head(self): response = self.request(
