From fc32be956977f05042f3d764d8b4a4e414abd690 Mon Sep 17 00:00:00 2001 From: Michael Lenzen Date: Thu, 6 Feb 2020 00:17:21 -0600 Subject: [PATCH] Add tests for current Boolean validators --- tests/tests.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/tests/tests.py b/tests/tests.py index 5904066..c03e721 100755 --- a/tests/tests.py +++ b/tests/tests.py @@ -223,6 +223,7 @@ class Course(Model): description = Column(sqla_types.Text, nullable=False) level = Column(sqla_types.Enum('Primary', 'Secondary')) has_prereqs = Column(sqla_types.Boolean, nullable=False) + boolean_nullable = Column(sqla_types.Boolean, nullable=True) started = Column(sqla_types.DateTime, nullable=False) grade = Column(AnotherInteger, nullable=False) @@ -256,9 +257,14 @@ class Student(Model): self.sess = Session() def test_auto_validators(self): + course_form = model_form(self.Course, self.sess)() student_form = model_form(self.Student, self.sess)() assert contains_validator(student_form.dob, Optional) assert contains_validator(student_form.full_name, Required) + assert not contains_validator(course_form.has_prereqs, Optional) + assert contains_validator(course_form.has_prereqs, Required) + assert contains_validator(course_form.boolean_nullable, Optional) + assert not contains_validator(course_form.boolean_nullable, Required) def test_field_args(self): shared = {'full_name': {'validators': [Regexp('test')]}} @@ -303,7 +309,7 @@ def test_convert_basic(self): self.assertRaises(ModelConversionError, model_form, self.Course) form_class = model_form(self.Course, exclude=['students']) form = form_class() - self.assertEqual(len(list(form)), 7) + self.assertEqual(len(list(form)), 8) def test_only(self): desired_fields = ['id', 'cost', 'description'] @@ -318,7 +324,7 @@ def test_no_mro(self): self.assertRaises(ModelConversionError, model_form, self.Course, self.sess, converter=converter) # If we exclude 'grade' everything should continue working F = model_form(self.Course, self.sess, exclude=['grade'], converter=converter) - self.assertEqual(len(list(F())), 7) + self.assertEqual(len(list(F())), 8) class ModelFormColumnDefaultTest(TestCase):