Skip to content

Commit

Permalink
Fixed #14102 -- Ensure that fields that have been excluded from a for…
Browse files Browse the repository at this point in the history
…m aren't included in the unique_for_* checks, either. Thanks to Travis Cline for the report and fix.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13598 bcc190cf-cafb-0310-a4f2-bffc1f526a37
  • Loading branch information
freakboy3742 committed Aug 17, 2010
1 parent 810ed2b commit 015d85a
Show file tree
Hide file tree
Showing 4 changed files with 18 additions and 5 deletions.
1 change: 1 addition & 0 deletions AUTHORS
Expand Up @@ -108,6 +108,7 @@ answer newbie questions, and generally made Django that much better:
Michal Chruszcz <troll@pld-linux.org>
Can Burak Çilingir <canburak@cs.bilgi.edu.tr>
Ian Clelland <clelland@gmail.com>
Travis Cline <travis.cline@gmail.com>
Russell Cloran <russell@rucus.net>
colin@owlfish.com
crankycoder@gmail.com
Expand Down
9 changes: 4 additions & 5 deletions django/db/models/base.py
@@ -1,6 +1,5 @@
import types
import sys
import os
from itertools import izip
import django.db.models.manager # Imported to register signal handler.
from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned, FieldError, ValidationError, NON_FIELD_ERRORS
Expand All @@ -16,7 +15,7 @@
from django.utils.translation import ugettext_lazy as _
import django.utils.copycompat as copy
from django.utils.functional import curry, update_wrapper
from django.utils.encoding import smart_str, force_unicode, smart_unicode
from django.utils.encoding import smart_str, force_unicode
from django.utils.text import get_text_list, capfirst
from django.conf import settings

Expand Down Expand Up @@ -744,11 +743,11 @@ def _get_unique_checks(self, exclude=None):
continue
if f.unique:
unique_checks.append((model_class, (name,)))
if f.unique_for_date:
if f.unique_for_date and f.unique_for_date not in exclude:
date_checks.append((model_class, 'date', name, f.unique_for_date))
if f.unique_for_year:
if f.unique_for_year and f.unique_for_year not in exclude:
date_checks.append((model_class, 'year', name, f.unique_for_year))
if f.unique_for_month:
if f.unique_for_month and f.unique_for_month not in exclude:
date_checks.append((model_class, 'month', name, f.unique_for_month))
return unique_checks, date_checks

Expand Down
4 changes: 4 additions & 0 deletions tests/modeltests/model_forms/tests.py
Expand Up @@ -156,6 +156,10 @@ def test_unique_for_date(self):
form = PostForm({'subtitle': "Finally", "title": "Django 1.0 is released",
"slug": "Django 1.0", 'posted': '2008-09-03'}, instance=p)
self.assertTrue(form.is_valid())
form = PostForm({'title': "Django 1.0 is released"})
self.assertFalse(form.is_valid())
self.assertEqual(len(form.errors), 1)
self.assertEqual(form.errors['posted'], [u'This field is required.'])

def test_inherited_unique_for_date(self):
p = Post.objects.create(title="Django 1.0 is released",
Expand Down
9 changes: 9 additions & 0 deletions tests/modeltests/validation/test_unique.py
Expand Up @@ -40,6 +40,15 @@ def test_unique_for_date_gets_picked_up(self):
), m._get_unique_checks()
)

def test_unique_for_date_exclusion(self):
m = UniqueForDateModel()
self.assertEqual((
[(UniqueForDateModel, ('id',))],
[(UniqueForDateModel, 'year', 'count', 'end_date'),
(UniqueForDateModel, 'month', 'order', 'end_date')]
), m._get_unique_checks(exclude='start_date')
)

class PerformUniqueChecksTest(unittest.TestCase):
def setUp(self):
# Set debug to True to gain access to connection.queries.
Expand Down

0 comments on commit 015d85a

Please sign in to comment.