from tests import testmodels

from tortoise.contrib import test

from tortoise.exceptions import ConfigurationError, IntegrityError

from tortoise.fields import BinaryField

class TestBinaryFields(test.TestCase):

async def test_empty(self):

with self.assertRaises(IntegrityError):

await testmodels.BinaryFields.create()

async def test_create(self):

obj0 = await testmodels.BinaryFields.create(binary=bytes(range(256)) * 500)

obj = await testmodels.BinaryFields.get(id=obj0.id)

self.assertEqual(obj.binary, bytes(range(256)) * 500)

self.assertEqual(obj.binary_null, None)

await obj.save()

obj2 = await testmodels.BinaryFields.get(id=obj.id)

self.assertEqual(obj, obj2)

async def test_values(self):

obj0 = await testmodels.BinaryFields.create(

binary=bytes(range(256)), binary_null=bytes(range(255, -1, -1))

)

values = await testmodels.BinaryFields.get(id=obj0.id).values("binary", "binary_null")

self.assertEqual(values["binary"], bytes(range(256)))

self.assertEqual(values["binary_null"], bytes(range(255, -1, -1)))

async def test_values_list(self):

obj0 = await testmodels.BinaryFields.create(binary=bytes(range(256)))

values = await testmodels.BinaryFields.get(id=obj0.id).values_list("binary", flat=True)

self.assertEqual(values, bytes(range(256)))

def test_unique_fail(self):

with self.assertRaisesRegex(ConfigurationError, "can't be indexed"):

BinaryField(unique=True)

def test_index_fail(self):

with self.assertRaisesRegex(ConfigurationError, "can't be indexed"):

BinaryField(index=True)