summaryrefslogtreecommitdiff
path: root/tests/model_forms
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_forms')
-rw-r--r--tests/model_forms/tests.py21
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):