from django.core.exceptions import ValidationError

from django.forms import CharField, ComboField, EmailField

from django.test import SimpleTestCase

class ComboFieldTest(SimpleTestCase):

    def test_combofield_1(self):

        f = ComboField(fields=[CharField(max_length=20), EmailField()])

        self.assertEqual("[email protected]", f.clean("[email protected]"))

        with self.assertRaisesMessage(

            ValidationError,

            "'Ensure this value has at most 20 characters (it has 28).'",

        ):

            f.clean("[email protected]")

        with self.assertRaisesMessage(

            ValidationError, "'Enter a valid email address.'"

        ):

            f.clean("not an email")

        with self.assertRaisesMessage(ValidationError, "'This field is required.'"):

            f.clean("")

        with self.assertRaisesMessage(ValidationError, "'This field is required.'"):

            f.clean(None)

    def test_combofield_2(self):

        f = ComboField(fields=[CharField(max_length=20), EmailField()], required=False)

        self.assertEqual("[email protected]", f.clean("[email protected]"))

        with self.assertRaisesMessage(

            ValidationError,

            "'Ensure this value has at most 20 characters (it has 28).'",

        ):

            f.clean("[email protected]")

        with self.assertRaisesMessage(

            ValidationError, "'Enter a valid email address.'"

        ):

            f.clean("not an email")

        self.assertEqual("", f.clean(""))

        self.assertEqual("", f.clean(None))