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")