Skip to content

Commit 60d0d7b

Browse files
committed
add unit test for checksum of chunk data
1 parent 24a6c8b commit 60d0d7b

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

tests/unit/asyncio/test_async_appendable_object_writer.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@
1515
import pytest
1616
from unittest import mock
1717

18+
from google_crc32c import Checksum
19+
1820
from google.api_core import exceptions
1921
from google.cloud.storage._experimental.asyncio.async_appendable_object_writer import (
2022
AsyncAppendableObjectWriter,
@@ -452,10 +454,15 @@ async def test_append_sends_data_in_chunks(mock_write_object_stream, mock_client
452454
# First chunk
453455
assert first_call[0][0].write_offset == 100
454456
assert len(first_call[0][0].checksummed_data.content) == _MAX_CHUNK_SIZE_BYTES
455-
457+
assert first_call[0][0].checksummed_data.crc32c == int.from_bytes(
458+
Checksum(data[:_MAX_CHUNK_SIZE_BYTES]).digest()
459+
)
456460
# Second chunk
457461
assert second_call[0][0].write_offset == 100 + _MAX_CHUNK_SIZE_BYTES
458462
assert len(second_call[0][0].checksummed_data.content) == 1
463+
assert second_call[0][0].checksummed_data.crc32c == int.from_bytes(
464+
Checksum(data[_MAX_CHUNK_SIZE_BYTES:]).digest()
465+
)
459466

460467
assert writer.offset == 100 + len(data)
461468
writer.simple_flush.assert_not_awaited()

0 commit comments

Comments
 (0)