mirror of
				https://github.com/django/django.git
				synced 2025-10-29 08:36:09 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			103 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			103 lines
		
	
	
		
			3.8 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| from django.core.exceptions import ValidationError
 | |
| from django.db import models
 | |
| from django.forms import ChoiceField, Form
 | |
| from django.test import SimpleTestCase
 | |
| 
 | |
| from . import FormFieldAssertionsMixin
 | |
| 
 | |
| 
 | |
| class ChoiceFieldTest(FormFieldAssertionsMixin, SimpleTestCase):
 | |
|     def test_choicefield_1(self):
 | |
|         f = ChoiceField(choices=[("1", "One"), ("2", "Two")])
 | |
|         with self.assertRaisesMessage(ValidationError, "'This field is required.'"):
 | |
|             f.clean("")
 | |
|         with self.assertRaisesMessage(ValidationError, "'This field is required.'"):
 | |
|             f.clean(None)
 | |
|         self.assertEqual("1", f.clean(1))
 | |
|         self.assertEqual("1", f.clean("1"))
 | |
|         msg = "'Select a valid choice. 3 is not one of the available choices.'"
 | |
|         with self.assertRaisesMessage(ValidationError, msg):
 | |
|             f.clean("3")
 | |
| 
 | |
|     def test_choicefield_2(self):
 | |
|         f = ChoiceField(choices=[("1", "One"), ("2", "Two")], required=False)
 | |
|         self.assertEqual("", f.clean(""))
 | |
|         self.assertEqual("", f.clean(None))
 | |
|         self.assertEqual("1", f.clean(1))
 | |
|         self.assertEqual("1", f.clean("1"))
 | |
|         msg = "'Select a valid choice. 3 is not one of the available choices.'"
 | |
|         with self.assertRaisesMessage(ValidationError, msg):
 | |
|             f.clean("3")
 | |
| 
 | |
|     def test_choicefield_3(self):
 | |
|         f = ChoiceField(choices=[("J", "John"), ("P", "Paul")])
 | |
|         self.assertEqual("J", f.clean("J"))
 | |
|         msg = "'Select a valid choice. John is not one of the available choices.'"
 | |
|         with self.assertRaisesMessage(ValidationError, msg):
 | |
|             f.clean("John")
 | |
| 
 | |
|     def test_choicefield_4(self):
 | |
|         f = ChoiceField(
 | |
|             choices=[
 | |
|                 ("Numbers", (("1", "One"), ("2", "Two"))),
 | |
|                 ("Letters", (("3", "A"), ("4", "B"))),
 | |
|                 ("5", "Other"),
 | |
|             ]
 | |
|         )
 | |
|         self.assertEqual("1", f.clean(1))
 | |
|         self.assertEqual("1", f.clean("1"))
 | |
|         self.assertEqual("3", f.clean(3))
 | |
|         self.assertEqual("3", f.clean("3"))
 | |
|         self.assertEqual("5", f.clean(5))
 | |
|         self.assertEqual("5", f.clean("5"))
 | |
|         msg = "'Select a valid choice. 6 is not one of the available choices.'"
 | |
|         with self.assertRaisesMessage(ValidationError, msg):
 | |
|             f.clean("6")
 | |
| 
 | |
|     def test_choicefield_choices_default(self):
 | |
|         f = ChoiceField()
 | |
|         self.assertEqual(f.choices, [])
 | |
| 
 | |
|     def test_choicefield_callable(self):
 | |
|         def choices():
 | |
|             return [("J", "John"), ("P", "Paul")]
 | |
| 
 | |
|         f = ChoiceField(choices=choices)
 | |
|         self.assertEqual("J", f.clean("J"))
 | |
| 
 | |
|     def test_choicefield_callable_may_evaluate_to_different_values(self):
 | |
|         choices = []
 | |
| 
 | |
|         def choices_as_callable():
 | |
|             return choices
 | |
| 
 | |
|         class ChoiceFieldForm(Form):
 | |
|             choicefield = ChoiceField(choices=choices_as_callable)
 | |
| 
 | |
|         choices = [("J", "John")]
 | |
|         form = ChoiceFieldForm()
 | |
|         self.assertEqual([("J", "John")], list(form.fields["choicefield"].choices))
 | |
| 
 | |
|         choices = [("P", "Paul")]
 | |
|         form = ChoiceFieldForm()
 | |
|         self.assertEqual([("P", "Paul")], list(form.fields["choicefield"].choices))
 | |
| 
 | |
|     def test_choicefield_disabled(self):
 | |
|         f = ChoiceField(choices=[("J", "John"), ("P", "Paul")], disabled=True)
 | |
|         self.assertWidgetRendersTo(
 | |
|             f,
 | |
|             '<select id="id_f" name="f" disabled><option value="J">John</option>'
 | |
|             '<option value="P">Paul</option></select>',
 | |
|         )
 | |
| 
 | |
|     def test_choicefield_enumeration(self):
 | |
|         class FirstNames(models.TextChoices):
 | |
|             JOHN = "J", "John"
 | |
|             PAUL = "P", "Paul"
 | |
| 
 | |
|         f = ChoiceField(choices=FirstNames.choices)
 | |
|         self.assertEqual(f.clean("J"), "J")
 | |
|         msg = "'Select a valid choice. 3 is not one of the available choices.'"
 | |
|         with self.assertRaisesMessage(ValidationError, msg):
 | |
|             f.clean("3")
 |