summaryrefslogtreecommitdiff
path: root/tests/model_fields/test_integerfield.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_fields/test_integerfield.py')
-rw-r--r--tests/model_fields/test_integerfield.py14
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/model_fields/test_integerfield.py b/tests/model_fields/test_integerfield.py
index c0bb7627cf..606d71057a 100644
--- a/tests/model_fields/test_integerfield.py
+++ b/tests/model_fields/test_integerfield.py
@@ -184,6 +184,9 @@ class PositiveIntegerFieldTests(IntegerFieldTests):
class ValidationTests(SimpleTestCase):
+ class Choices(models.IntegerChoices):
+ A = 1
+
def test_integerfield_cleans_valid_string(self):
f = models.IntegerField()
self.assertEqual(f.clean('2', None), 2)
@@ -217,3 +220,14 @@ class ValidationTests(SimpleTestCase):
f = models.IntegerField(choices=((1, 1),))
with self.assertRaises(ValidationError):
f.clean('0', None)
+
+ def test_enum_choices_cleans_valid_string(self):
+ f = models.IntegerField(choices=self.Choices.choices)
+ self.assertEqual(f.clean('1', None), 1)
+
+ def test_enum_choices_invalid_input(self):
+ f = models.IntegerField(choices=self.Choices.choices)
+ with self.assertRaises(ValidationError):
+ f.clean('A', None)
+ with self.assertRaises(ValidationError):
+ f.clean('3', None)