Skip to content

Commit

Permalink
fix unit
Browse files Browse the repository at this point in the history
  • Loading branch information
@vgvoleg
vgvoleg committed Dec 16, 2024
1 parent 25e07d7 commit 9f3c04a
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 1 deletion.
13 changes: 13 additions & 0 deletions ydb/_topic_reader/topic_reader_asyncio_test.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -208,6 +208,7 @@ def create_message(
written_at=datetime.datetime(2023, 2, 3, 14, 16),
producer_id="test-producer-id",
data=bytes(),
metadata_items={},
_partition_session=partition_session,
_commit_start_offset=partition_session._next_message_start_commit_offset + offset_delta - 1,
_commit_end_offset=partition_session._next_message_start_commit_offset + offset_delta,
Expand Down Expand Up @@ -251,6 +252,7 @@ def batch_size():
seq_no=message.seqno,
created_at=message.created_at,
data=message.data,
metadata_items={},
uncompresed_size=len(message.data),
message_group_id=message.message_group_id,
)
Expand Down Expand Up @@ -773,6 +775,7 @@ async def test_free_buffer_after_partition_stop(self, stream, stream_reader, par
seq_no=123,
created_at=t,
data=bytes(),
metadata_items={},
uncompresed_size=message_size,
message_group_id="test-message-group",
)
Expand Down Expand Up @@ -853,6 +856,7 @@ def reader_batch_count():
created_at=created_at,
data=data,
uncompresed_size=len(data),
metadata_items={},
message_group_id=message_group_id,
)
],
Expand Down Expand Up @@ -884,6 +888,7 @@ def reader_batch_count():
written_at=written_at,
producer_id=producer_id,
data=data,
metadata_items={},
_partition_session=partition_session,
_commit_start_offset=expected_message_offset,
_commit_end_offset=expected_message_offset + 1,
Expand Down Expand Up @@ -930,6 +935,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
seq_no=3,
created_at=created_at,
data=data,
metadata_items={},
uncompresed_size=len(data),
message_group_id=message_group_id,
)
Expand All @@ -951,6 +957,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
seq_no=2,
created_at=created_at2,
data=data,
metadata_items={},
uncompresed_size=len(data),
message_group_id=message_group_id,
)
Expand All @@ -967,6 +974,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
seq_no=3,
created_at=created_at3,
data=data2,
metadata_items={},
uncompresed_size=len(data2),
message_group_id=message_group_id,
),
Expand All @@ -975,6 +983,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
seq_no=5,
created_at=created_at4,
data=data,
metadata_items={},
uncompresed_size=len(data),
message_group_id=message_group_id2,
),
Expand Down Expand Up @@ -1005,6 +1014,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
written_at=written_at,
producer_id=producer_id,
data=data,
metadata_items={},
_partition_session=partition_session,
_commit_start_offset=partition1_mess1_expected_offset,
_commit_end_offset=partition1_mess1_expected_offset + 1,
Expand All @@ -1025,6 +1035,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
written_at=written_at2,
producer_id=producer_id,
data=data,
metadata_items={},
_partition_session=second_partition_session,
_commit_start_offset=partition2_mess1_expected_offset,
_commit_end_offset=partition2_mess1_expected_offset + 1,
Expand All @@ -1045,6 +1056,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
written_at=written_at2,
producer_id=producer_id2,
data=data2,
metadata_items={},
_partition_session=second_partition_session,
_commit_start_offset=partition2_mess2_expected_offset,
_commit_end_offset=partition2_mess2_expected_offset + 1,
Expand All @@ -1058,6 +1070,7 @@ async def test_read_batches(self, stream_reader, partition_session, second_parti
written_at=written_at2,
producer_id=producer_id,
data=data,
metadata_items={},
_partition_session=second_partition_session,
_commit_start_offset=partition2_mess3_expected_offset,
_commit_end_offset=partition2_mess3_expected_offset + 1,
Expand Down
3 changes: 2 additions & 1 deletion ydb/_topic_writer/topic_writer.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -120,11 +120,12 @@ class InternalMessage(StreamWriteMessage.WriteRequest.MessageData, IToProto):
codec: PublicCodec

def __init__(self, mess: PublicMessage):
metadata_items = mess.metadata_items or {}
super().__init__(
seq_no=mess.seqno,
created_at=mess.created_at,
data=mess.data,
metadata_items=mess.metadata_items,
metadata_items=metadata_items,
uncompressed_size=len(mess.data),
partitioning=None,
)
Expand Down
1 change: 1 addition & 0 deletions ydb/_topic_writer/topic_writer_asyncio_test.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -153,6 +153,7 @@ async def test_write_a_message(self, writer_and_stream: WriterWithMockedStream):
seq_no=1,
created_at=now,
data=data,
metadata_items={},
uncompressed_size=len(data),
partitioning=None,
)
Expand Down

0 comments on commit 9f3c04a

Please sign in to comment.