Ticket #19698: 19698-rebase-and-cleaned.diff

django/contrib/sites/models.py

@@ -1 +1 @@
 from django.db import models
+from django.db.models.signals import pre_save, pre_delete
 from django.utils.translation import ugettext_lazy as _
 from django.utils.encoding import python_2_unicode_compatible
 
@@ -49 +50 @@
     def __str__(self):
         return self.domain
 
-    def save(self, *args, **kwargs):
-        super(Site, self).save(*args, **kwargs)
-        # Cached information will likely be incorrect now.
-        if self.id in SITE_CACHE:
-            del SITE_CACHE[self.id]
-
-    def delete(self):
-        pk = self.pk
-        super(Site, self).delete()
-        try:
-            del SITE_CACHE[pk]
-        except KeyError:
-            pass
-
 
 @python_2_unicode_compatible
 class RequestSite(object):
@@ -96 +83 @@
     else:
         current_site = RequestSite(request)
     return current_site
+
+
+def clear_cache(sender, **kwargs):
+    """
+    Clears the cache (if primed) each time a site is saved or deleted
+    """
+    instance = kwargs['instance']
+    try:
+        del SITE_CACHE[instance.pk]
+    except KeyError:
+        pass
+pre_save.connect(clear_cache)
+pre_delete.connect(clear_cache)

django/contrib/sites/tests.py

@@ -42 +42 @@
         site = Site.objects.get_current()
         self.assertEqual("Example site", site.name)
 
+    def test_delete_all_sites_clears_cache(self):
+        # When all site objects are deleted the cache should also
+        # be cleared and get_current() should raise a DoesNotExist.
+        self.assertIsInstance(Site.objects.get_current(), Site)
+        Site.objects.all().delete()
+        self.assertRaises(Site.DoesNotExist, Site.objects.get_current)
+
     @override_settings(ALLOWED_HOSTS=['example.com'])
     def test_get_current_site(self):
         # Test that the correct Site object is returned