diff options
| -rw-r--r-- | django/contrib/sites/models.py | 9 | ||||
| -rw-r--r-- | django/contrib/sites/tests.py | 13 |
2 files changed, 22 insertions, 0 deletions
diff --git a/django/contrib/sites/models.py b/django/contrib/sites/models.py index fb6d0a2b1e..c928a4e852 100644 --- a/django/contrib/sites/models.py +++ b/django/contrib/sites/models.py @@ -44,6 +44,15 @@ class Site(models.Model): def __unicode__(self): return self.domain + def delete(self): + pk = self.pk + super(Site, self).delete() + try: + del(SITE_CACHE[pk]) + except KeyError: + pass + + class RequestSite(object): """ A class that shares the primary interface of Site (i.e., it has diff --git a/django/contrib/sites/tests.py b/django/contrib/sites/tests.py new file mode 100644 index 0000000000..d2ec331eca --- /dev/null +++ b/django/contrib/sites/tests.py @@ -0,0 +1,13 @@ +""" +>>> # Make sure that get_current() does not return a deleted Site object. +>>> from django.contrib.sites.models import Site +>>> s = Site.objects.get_current() +>>> s +<Site: example.com> + +>>> s.delete() +>>> Site.objects.get_current() +Traceback (most recent call last): +... +DoesNotExist: Site matching query does not exist. +""" |