from tests import testmodels
from tortoise import fields
from tortoise.contrib import test
from tortoise.exceptions import ConfigurationError, ValidationError


class TestCharFields(test.TestCase):
    def test_max_length_missing(self):
        with self.assertRaisesRegex(
            TypeError, "missing 1 required positional argument: 'max_length'"
        ):
            fields.CharField()  # pylint: disable=E1120

    def test_max_length_bad(self):
        with self.assertRaisesRegex(ConfigurationError, "'max_length' must be >= 1"):
            fields.CharField(max_length=0)

    async def test_empty(self):
        with self.assertRaises(ValidationError):
            await testmodels.CharFields.create()

    async def test_create(self):
        obj0 = await testmodels.CharFields.create(char="moo")
        obj = await testmodels.CharFields.get(id=obj0.id)
        self.assertEqual(obj.char, "moo")
        self.assertEqual(obj.char_null, None)
        await obj.save()
        obj2 = await testmodels.CharFields.get(id=obj.id)
        self.assertEqual(obj, obj2)

    async def test_update(self):
        obj0 = await testmodels.CharFields.create(char="moo")
        await testmodels.CharFields.filter(id=obj0.id).update(char="ba'a")
        obj = await testmodels.CharFields.get(id=obj0.id)
        self.assertEqual(obj.char, "ba'a")
        self.assertEqual(obj.char_null, None)

    async def test_cast(self):
        obj0 = await testmodels.CharFields.create(char=33)
        obj = await testmodels.CharFields.get(id=obj0.id)
        self.assertEqual(obj.char, "33")

    async def test_values(self):
        obj0 = await testmodels.CharFields.create(char="moo")
        values = await testmodels.CharFields.get(id=obj0.id).values("char")
        self.assertEqual(values["char"], "moo")

    async def test_values_list(self):
        obj0 = await testmodels.CharFields.create(char="moo")
        values = await testmodels.CharFields.get(id=obj0.id).values_list("char", flat=True)
        self.assertEqual(values, "moo")