Fixed a few ResourceWarnings.

This commit is contained in:
Florian Apolloner 2013-02-24 13:10:42 +01:00
parent e4e1287590
commit e4ee3d8fca
3 changed files with 9 additions and 13 deletions

View File

@ -108,14 +108,12 @@ class FileStorageTests(TestCase):
temp_storage.save('tests/example.txt', ContentFile('some content'))
# Load it as python file object
file_obj = open(temp_storage.path('tests/example.txt'))
# Save it using storage and read its content
temp_storage.save('tests/file_obj', file_obj)
with open(temp_storage.path('tests/example.txt')) as file_obj:
# Save it using storage and read its content
temp_storage.save('tests/file_obj', file_obj)
self.assertTrue(temp_storage.exists('tests/file_obj'))
self.assertEqual(
temp_storage.open('tests/file_obj').read(),
b'some content')
with temp_storage.open('tests/file_obj') as f:
self.assertEqual(f.read(), b'some content')
def test_stringio(self):
@ -127,9 +125,8 @@ class FileStorageTests(TestCase):
# Save it and read written file
temp_storage.save('tests/stringio', output)
self.assertTrue(temp_storage.exists('tests/stringio'))
self.assertEqual(
temp_storage.open('tests/stringio').read(),
b'content')
with temp_storage.open('tests/stringio') as f:
self.assertEqual(f.read(), b'content')

View File

@ -639,7 +639,7 @@ class FileLikeObjectTestCase(LiveServerBase):
f = File(file_like_object)
stored_filename = self.storage.save("remote_file.html", f)
stored_file = self.storage.open(stored_filename)
remote_file = self.urlopen('/example_view/')
self.assertEqual(stored_file.read(), remote_file.read())
with self.storage.open(stored_filename) as stored_file:
self.assertEqual(stored_file.read(), remote_file.read())

View File

@ -4,7 +4,6 @@ import shutil
import subprocess
import sys
import tempfile
import warnings
from django import contrib
from django.utils._os import upath