from tests import testmodels
from tortoise.contrib import test
from tortoise.exceptions import IntegrityError


class TestBooleanFields(test.TestCase):
    async def test_empty(self):
        with self.assertRaises(IntegrityError):
            await testmodels.BooleanFields.create()

    async def test_create(self):
        obj0 = await testmodels.BooleanFields.create(boolean=True)
        obj = await testmodels.BooleanFields.get(id=obj0.id)
        self.assertIs(obj.boolean, True)
        self.assertIs(obj.boolean_null, None)
        await obj.save()
        obj2 = await testmodels.BooleanFields.get(id=obj.id)
        self.assertEqual(obj, obj2)

    async def test_update(self):
        obj0 = await testmodels.BooleanFields.create(boolean=False)
        await testmodels.BooleanFields.filter(id=obj0.id).update(boolean=False)
        obj = await testmodels.BooleanFields.get(id=obj0.id)
        self.assertIs(obj.boolean, False)
        self.assertIs(obj.boolean_null, None)

    async def test_values(self):
        obj0 = await testmodels.BooleanFields.create(boolean=True)
        values = await testmodels.BooleanFields.get(id=obj0.id).values("boolean")
        self.assertIs(values["boolean"], True)

    async def test_values_list(self):
        obj0 = await testmodels.BooleanFields.create(boolean=True)
        values = await testmodels.BooleanFields.get(id=obj0.id).values_list("boolean", flat=True)
        self.assertIs(values, True)