diff options
| author | Niclas Olofsson <n@niclasolofsson.se> | 2014-07-26 11:06:30 +0200 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2014-07-31 15:29:46 -0400 |
| commit | cdc25ac4747bf5a6cdc2e70461c2d43c54529d35 (patch) | |
| tree | 98987ea8e79363268cd59368b445e0a86dd6b7b6 | |
| parent | 75790808993691d2869db577df6db47ecbef6f5a (diff) | |
Fixed #22808 -- Made ModelMultipleChoiceField validation more robust to invalid data types..
Thanks Mattias Lindvall for the report and inital patch.
| -rw-r--r-- | AUTHORS | 1 | ||||
| -rw-r--r-- | django/forms/models.py | 4 | ||||
| -rw-r--r-- | tests/model_forms/tests.py | 13 |
3 files changed, 16 insertions, 2 deletions
@@ -458,6 +458,7 @@ answer newbie questions, and generally made Django that much better: Todd O'Bryan <toddobryan@mac.com> Alex Ogier <alex.ogier@gmail.com> Joao Oliveira <joaoxsouls@gmail.com> + Niclas Olofsson <n@niclasolofsson.se> Selwin Ong <selwin@ui.co.id> Gerardo Orozco <gerardo.orozco.mosqueda@gmail.com> Christian Oudard <christian.oudard@gmail.com> diff --git a/django/forms/models.py b/django/forms/models.py index 5838ac1631..fb342b3316 100644 --- a/django/forms/models.py +++ b/django/forms/models.py @@ -1179,7 +1179,7 @@ class ModelChoiceField(ChoiceField): try: key = self.to_field_name or 'pk' value = self.queryset.get(**{key: value}) - except (ValueError, self.queryset.model.DoesNotExist): + except (ValueError, TypeError, self.queryset.model.DoesNotExist): raise ValidationError(self.error_messages['invalid_choice'], code='invalid_choice') return value @@ -1227,7 +1227,7 @@ class ModelMultipleChoiceField(ModelChoiceField): for pk in value: try: self.queryset.filter(**{key: pk}) - except ValueError: + except (ValueError, TypeError): raise ValidationError( self.error_messages['invalid_pk_value'], code='invalid_pk_value', diff --git a/tests/model_forms/tests.py b/tests/model_forms/tests.py index 03a498c4cd..3efc724c38 100644 --- a/tests/model_forms/tests.py +++ b/tests/model_forms/tests.py @@ -1391,6 +1391,13 @@ class ModelChoiceFieldTests(TestCase): f.clean(None) with self.assertRaises(ValidationError): f.clean(0) + + # Invalid types that require TypeError to be caught (#22808). + with self.assertRaises(ValidationError): + f.clean([['fail']]) + with self.assertRaises(ValidationError): + f.clean([{'foo': 'bar'}]) + self.assertEqual(f.clean(self.c2.id).name, "It's a test") self.assertEqual(f.clean(self.c3.id).name, 'Third') @@ -1495,6 +1502,12 @@ class ModelMultipleChoiceFieldTests(TestCase): with self.assertRaises(ValidationError): f.clean(['fail']) + # Invalid types that require TypeError to be caught (#22808). + with self.assertRaises(ValidationError): + f.clean([['fail']]) + with self.assertRaises(ValidationError): + f.clean([{'foo': 'bar'}]) + # Add a Category object *after* the ModelMultipleChoiceField has already been # instantiated. This proves clean() checks the database during clean() rather # than caching it at time of instantiation. |
