Tidied code, added replace() method to FileField

This commit is contained in:
Steve Challis
2010-06-03 15:27:21 +08:00
committed by Florian Schlachter
parent 6bfd6c322b
commit 47bfeec115
4 changed files with 40 additions and 11 deletions

View File

@@ -649,7 +649,10 @@ class FieldTest(unittest.TestCase):
self.assertTrue(streamfile == result)
self.assertEquals(result.file.read(), text + more_text)
self.assertEquals(result.file.content_type, content_type)
result.file.delete() # Remove file from GridFS
result.file.delete()
# Ensure deleted file returns None
self.assertTrue(result.file.read() == None)
setfile = SetFile()
setfile.file = text
@@ -658,7 +661,15 @@ class FieldTest(unittest.TestCase):
result = SetFile.objects.first()
self.assertTrue(setfile == result)
self.assertEquals(result.file.read(), text)
result.file.delete() # Remove file from GridFS
# Try replacing file with new one
result.file.replace(more_text)
result.save()
result.validate()
result = SetFile.objects.first()
self.assertTrue(setfile == result)
self.assertEquals(result.file.read(), more_text)
result.file.delete()
PutFile.drop_collection()
StreamFile.drop_collection()