Ticket #9228: sites2.diff

django/contrib/sites/tests.py

@@ -3 +3 @@
 >>> from django.conf import settings
 >>> Site(id=settings.SITE_ID, domain="example.com", name="example.com").save()
 
+>>> # Make sure that save() remove the relevant key from the SITE_CACHE dictionary.
+>>> s = Site.objects.get_current()
+>>> isinstance(s, Site)
+True
+
+>>> s.domain = "example2.com"
+>>> s.name = "example2.com"
+>>> s.save()
+>>> s = Site.objects.get_current()
+>>> s.domain
+'example2.com'
+
+>>> s.name
+'example2.com'
+
 >>> # Make sure that get_current() does not return a deleted Site object.
 >>> s = Site.objects.get_current()
 >>> isinstance(s, Site)

django/contrib/sites/models.py

@@ -42 +42 @@
     def __unicode__(self):
         return self.domain
     
+    def save(self, force_insert=False, force_update=False):
+        super(Site, self).save(force_insert, force_update)
+        del SITE_CACHE[self.pk]
+
     def delete(self):
         pk = self.pk
         super(Site, self).delete()