Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions dimos/protocol/pubsub/test_spec.py
Original file line number Diff line number Diff line change
Expand Up @@ -349,9 +349,9 @@ async def consume_messages() -> None:
def test_high_volume_messages(
pubsub_context: Callable[[], Any], topic: Any, values: list[Any]
) -> None:
"""Test that all 5k messages are received correctly.
Limited to 5k because ros transport cannot handle more.
Might want to have separate expectations per transport later
"""Test that all messages are received correctly under moderate volume.
This is an acceptance test, not a benchmark, so volume is kept low (500)
to avoid flakiness. Might want separate expectations per transport later.
"""
with pubsub_context() as x:
# Create a list to capture received messages
Expand All @@ -366,8 +366,8 @@ def callback(message: Any, topic: Any) -> None:
# Subscribe to the topic
x.subscribe(topic, callback)

# Publish 5000 messages
num_messages = 5000
# Publish 500 messages
num_messages = 500
Comment thread
spomichter marked this conversation as resolved.
for _ in range(num_messages):
x.publish(topic, values[0])

Expand Down
Loading