diff --git a/opentech/apply/categories/tests/test_blocks.py b/opentech/apply/categories/tests/test_blocks.py index 0e8db56139db755e4d0f0c4979bf98c4bc2ec10b..17ab027e7b470ad02c8df342cf5a061db0d26e9a 100644 --- a/opentech/apply/categories/tests/test_blocks.py +++ b/opentech/apply/categories/tests/test_blocks.py @@ -1,3 +1,4 @@ +from django import forms from django.test import TestCase from opentech.apply.categories.blocks import CategoryQuestionBlock @@ -34,3 +35,11 @@ class TestCategoryQuestionBlock(TestCase): field = self.get_field(**values) self.assertEqual(values['field_label'], field.label) self.assertEqual(values['help_text'], field.help_text) + + def test_multi_select_enabled(self): + field = self.get_field(multi=True) + self.assertTrue(isinstance(field, forms.MultipleChoiceField)) + + def test_multi_select_disabled(self): + field = self.get_field(multi=True) + self.assertTrue(isinstance(field, forms.ChoiceField))