def test_title_classes(self):
"""
The template tag should return the correct classes
"""
classes = wagtail_omni_forms_tags.admin_field_classes_for_field(self.form['title']).split()
self.assertEqual(2, len(classes))
self.assertIn('field', classes)
self.assertIn('char_field', classes)
def test_choices_classes(self):
"""
The template tag should return the correct classes
"""
classes = wagtail_omni_forms_tags.admin_field_classes_for_field(self.form['choices']).split()
self.assertEqual(3, len(classes))
self.assertIn('field', classes)
self.assertIn('typed_choice_field', classes)
self.assertIn('select', classes)
def test_age_classes(self):
"""
The template tag should return the correct classes
"""
classes = wagtail_omni_forms_tags.admin_field_classes_for_field(self.form['age']).split()
self.assertEqual(3, len(classes))
self.assertIn('field', classes)
self.assertIn('integer_field', classes)
self.assertIn('number_input', classes)
def test_description_classes(self):
"""
The template tag should return the correct classes
"""
classes = wagtail_omni_forms_tags.admin_field_classes_for_field(self.form['description']).split()
self.assertEqual(3, len(classes))
self.assertIn('field', classes)
self.assertIn('char_field', classes)
self.assertIn('admin_auto_height_text_input', classes)
def test_some_time_classes(self):
"""
The template tag should return the correct classes
"""
classes = wagtail_omni_forms_tags.admin_field_classes_for_field(self.form['some_time']).split()
self.assertEqual(3, len(classes))
self.assertIn('field', classes)
self.assertIn('time_field', classes)
self.assertIn('admin_time_input', classes)
