fix #409 - nullable large binary fields
This commit is contained in:
@ -58,6 +58,21 @@ class LargeBinaryStr(ormar.Model):
|
||||
)
|
||||
|
||||
|
||||
class LargeBinaryNullableStr(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "my_str_blobs2"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
test_binary: str = ormar.LargeBinary(
|
||||
max_length=100000,
|
||||
choices=[blob3, blob4],
|
||||
represent_as_base64_str=True,
|
||||
nullable=True,
|
||||
)
|
||||
|
||||
|
||||
class UUIDSample(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "uuids"
|
||||
@ -231,6 +246,37 @@ async def test_binary_str_column():
|
||||
assert items[1].__dict__["test_binary"] == blob4
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_binary_nullable_str_column():
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
await LargeBinaryNullableStr().save()
|
||||
await LargeBinaryNullableStr.objects.create()
|
||||
items = await LargeBinaryNullableStr.objects.all()
|
||||
assert len(items) == 2
|
||||
|
||||
items[0].test_binary = blob3
|
||||
items[1].test_binary = blob4
|
||||
|
||||
await LargeBinaryNullableStr.objects.bulk_update(items)
|
||||
items = await LargeBinaryNullableStr.objects.all()
|
||||
assert len(items) == 2
|
||||
assert items[0].test_binary == base64.b64encode(blob3).decode()
|
||||
items[0].test_binary = base64.b64encode(blob4).decode()
|
||||
assert items[0].test_binary == base64.b64encode(blob4).decode()
|
||||
assert items[1].test_binary == base64.b64encode(blob4).decode()
|
||||
assert items[1].__dict__["test_binary"] == blob4
|
||||
|
||||
await LargeBinaryNullableStr.objects.bulk_create(
|
||||
[LargeBinaryNullableStr(), LargeBinaryNullableStr(test_binary=blob3)]
|
||||
)
|
||||
items = await LargeBinaryNullableStr.objects.all()
|
||||
assert len(items) == 4
|
||||
await items[0].update(test_binary=blob4)
|
||||
check_item = await LargeBinaryNullableStr.objects.get(id=items[0].id)
|
||||
assert check_item.test_binary == base64.b64encode(blob4).decode()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_uuid_column():
|
||||
async with database:
|
||||
|
||||
Reference in New Issue
Block a user