Set validators

Author: adamchainz

src/django_mysql/models/fields/integer.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from django.core.validators import MaxValueValidator
+from django.core.validators import MinValueValidator
 from django.db.backends.base.base import BaseDatabaseWrapper
 from django.db.models import PositiveSmallIntegerField
 from django.db.models import SmallIntegerField
@@ -8,13 +10,15 @@
 
 class TinyIntegerField(SmallIntegerField):
     description = _("Small integer")
+    default_validators = [MinValueValidator(-128), MaxValueValidator(127)]
 
     def db_type(self, connection: BaseDatabaseWrapper) -> str:
         return "tinyint"
 
 
 class PositiveTinyIntegerField(PositiveSmallIntegerField):
     description = _("Positive small integer")
+    default_validators = [MinValueValidator(0), MaxValueValidator(255)]
 
     def db_type(self, connection: BaseDatabaseWrapper) -> str:
         return "tinyint unsigned"

tests/testapp/test_tinyintegerfield.py

@@ -2,6 +2,8 @@
 
 import pytest
 from django.core.management import call_command
+from django.core.validators import MaxValueValidator
+from django.core.validators import MinValueValidator
 from django.db import connection
 from django.db.utils import DataError
 from django.test import TestCase
@@ -79,3 +81,21 @@ def test_adding_field_with_default(self):
         )
         with connection.cursor() as cursor:
             assert table_name not in table_names(cursor)
+
+
+class TestFormValidation(TestCase):
+    def test_signed_validators(self):
+        validators = TinyIntegerModel._meta.get_field("tiny_signed").validators
+        assert len(validators) == 2
+        assert isinstance(validators[0], MinValueValidator)
+        assert validators[0].limit_value == -128
+        assert isinstance(validators[1], MaxValueValidator)
+        assert validators[1].limit_value == 127
+
+    def test_unsigned_validators(self):
+        validators = TinyIntegerModel._meta.get_field("tiny_unsigned").validators
+        assert len(validators) == 2
+        assert isinstance(validators[0], MinValueValidator)
+        assert validators[0].limit_value == 0
+        assert isinstance(validators[1], MaxValueValidator)
+        assert validators[1].limit_value == 255