Fix boolean field constraint name

shosca · Nov 24, 2018 · f259c0f · f259c0f
1 parent 5e0adea
commit f259c0f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/django_sorcery/db/fields.py b/django_sorcery/db/fields.py
@@ -29,6 +29,7 @@
     "SmallIntegerField",
     "TextField",
     "TimeField",
+    "TimestampField",
     "URLField",
 ]
 
@@ -115,6 +116,11 @@ class BooleanField(Field):
     type_class = sa.Boolean
     form_class = djangofields.BooleanField
 
+    def get_type_kwargs(self, type_class, kwargs):
+        type_kwargs = super(BooleanField, self).get_type_kwargs(type_class, kwargs)
+        type_kwargs["name"] = kwargs.pop("constraint_name", None)
+        return type_kwargs
+
     def get_column_kwargs(self, kwargs):
         column_kwargs = super(BooleanField, self).get_column_kwargs(kwargs)
         column_kwargs["nullable"] = False

diff --git a/tests/db/test_fields.py b/tests/db/test_fields.py
@@ -34,6 +34,9 @@ def test_boolean_field(self):
         form_field = meta.column_info(column).formfield()
         self.assertIsInstance(form_field, djangofields.BooleanField)
 
+        column = fields.BooleanField(constraint_name="test")
+        self.assertEqual(column.type.name, "test")
+
     def test_char_field(self):
         column = fields.CharField()
         self.assertIsInstance(column.type, sa.String)
@@ -164,6 +167,9 @@ def test_nullboolean_field(self):
         self.assertIsInstance(column.type, sa.Boolean)
         self.assertTrue(column.nullable)
 
+        column = fields.NullBooleanField(constraint_name="test")
+        self.assertEqual(column.type.name, "test")
+
         form_field = meta.column_info(column).formfield()
         self.assertIsInstance(form_field, djangofields.NullBooleanField)