diff options
Diffstat (limited to 'tests/model_forms')
| -rw-r--r-- | tests/model_forms/tests.py | 21 |
1 files changed, 19 insertions, 2 deletions
diff --git a/tests/model_forms/tests.py b/tests/model_forms/tests.py index 193b5e1fda..78e7399637 100644 --- a/tests/model_forms/tests.py +++ b/tests/model_forms/tests.py @@ -14,8 +14,8 @@ from django.core.validators import ValidationError from django.db import connection, models from django.db.models.query import EmptyQuerySet from django.forms.models import ( - ModelFormMetaclass, construct_instance, fields_for_model, model_to_dict, - modelform_factory, + ModelChoiceIterator, ModelFormMetaclass, construct_instance, + fields_for_model, model_to_dict, modelform_factory, ) from django.template import Context, Template from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature @@ -1573,6 +1573,23 @@ class ModelChoiceFieldTests(TestCase): with self.assertNumQueries(1): template.render(Context({'field': field})) + def test_modelchoicefield_iterator(self): + """ + Iterator defaults to ModelChoiceIterator and can be overridden with + the iterator attribute on a ModelChoiceField subclass. + """ + field = forms.ModelChoiceField(Category.objects.all()) + self.assertIsInstance(field.choices, ModelChoiceIterator) + + class CustomModelChoiceIterator(ModelChoiceIterator): + pass + + class CustomModelChoiceField(forms.ModelChoiceField): + iterator = CustomModelChoiceIterator + + field = CustomModelChoiceField(Category.objects.all()) + self.assertIsInstance(field.choices, CustomModelChoiceIterator) + class ModelMultipleChoiceFieldTests(TestCase): def setUp(self): |
