Added file cleanup in FileFieldTests.test_pickle().
@@ -110,13 +110,19 @@ class FileFieldTests(TestCase):
file1 = File(fp, name='test_file.py')
document = Document(myfile='test_file.py')
document.myfile.save('test_file.py', file1)
-
+ try:
dump = pickle.dumps(document)
loaded_document = pickle.loads(dump)
self.assertEqual(document.myfile, loaded_document.myfile)
self.assertEqual(document.myfile.url, loaded_document.myfile.url)
- self.assertEqual(document.myfile.storage, loaded_document.myfile.storage)
- self.assertEqual(document.myfile.instance, loaded_document.myfile.instance)
+ self.assertEqual(
+ document.myfile.storage,
+ loaded_document.myfile.storage,
+ )
+ self.assertEqual(
+ document.myfile.instance,
+ loaded_document.myfile.instance,
+ )
self.assertEqual(document.myfile.field, loaded_document.myfile.field)
myfile_dump = pickle.dumps(document.myfile)
@@ -126,3 +132,5 @@ class FileFieldTests(TestCase):
self.assertEqual(document.myfile.storage, loaded_myfile.storage)
self.assertEqual(document.myfile.instance, loaded_myfile.instance)
self.assertEqual(document.myfile.field, loaded_myfile.field)
+ finally:
+ document.myfile.delete()
