thirdparty/google_appengine/lib/django/tests/regressiontests/bug639/tests.py
author Pawel Solyga <Pawel.Solyga@gmail.com>
Mon, 01 Jun 2009 22:23:46 +0200
changeset 2376 feec28b50f1b
parent 109 620f9b141567
permissions -rwxr-xr-x
Extend taggable-mixin to support different Tag models. Usage is pretty simple. Tag model is default in Taggable constructor but you can call it with different model like GHOPTaskType that inherits from Tag model. Both Taggable and Tag models have been updated and they don't use hardcoded Tag model anymore and instead use cls of class methods or self.__class__. In case of Taggable it's self.__tag_model.

"""
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

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"] = {
            "filename" : "test.jpg",
            "content-type" : "image/jpeg",
            "content" : img
        }
        
        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())