django/tests/regressiontests/bug639/tests.py

"""
Tests for file field behavior, and specifically #639, in which Model.save() gets
called *again* for each FileField. This test will fail if calling an
auto-manipulator's save() method causes Model.save() to be called more than once.
"""

import os
import unittest
from regressiontests.bug639.models import Photo
from django.http import QueryDict
from django.utils.datastructures import MultiValueDict
from django.core.files.uploadedfile import SimpleUploadedFile

class Bug639Test(unittest.TestCase):
        
    def testBug639(self):
        """
        Simulate a file upload and check how many times Model.save() gets called.
        """
        # Grab an image for testing
        img = open(os.path.join(os.path.dirname(__file__), "test.jpg"), "rb").read()
        
        # Fake a request query dict with the file
        qd = QueryDict("title=Testing&image=", mutable=True)
        qd["image_file"] = SimpleUploadedFile('test.jpg', img, 'image/jpeg')

        manip = Photo.AddManipulator()
        manip.do_html2python(qd)
        p = manip.save(qd)
        
        # Check the savecount stored on the object (see the model)
        self.assertEqual(p._savecount, 1)
        
    def tearDown(self):
        """
        Make sure to delete the "uploaded" file to avoid clogging /tmp.
        """
        p = Photo.objects.get()
        os.unlink(p.get_image_filename())