varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2026-02-12 08:00:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

added msg dropping test

Browse Source
This commit is contained in:
mystical-prog
2024-11-08 19:55:58 +05:30
committed by Paul Robinson
parent 9ba04d0b15
commit a10b8fdb91
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/core/pubsub/test_pubsub.py
View File

@ -535,7 +535,7 @@ async def test_publish_push_msg_is_called(monkeypatch):
@pytest.mark.trio @pytest.mark.trio
async def test_push_msg(monkeypatch): async def test_push_msg(monkeypatch):
async with PubsubFactory.create_batch_with_floodsub(1) as pubsubs_fsub: async with PubsubFactory.create_batch_with_floodsub(2) as pubsubs_fsub:
msg_0 = make_pubsub_msg( msg_0 = make_pubsub_msg(
origin_id=pubsubs_fsub[0].my_id, origin_id=pubsubs_fsub[0].my_id,
topic_ids=[TESTING_TOPIC], topic_ids=[TESTING_TOPIC],
@ -571,6 +571,18 @@ async def test_push_msg(monkeypatch):
await trio.sleep(0.01) await trio.sleep(0.01)
assert not event.is_set() assert not event.is_set()
# Test: `push_msg` a new msg but forwarder as not self, it will be reject.
# `router_publish` is not called then.
msg_0A = make_pubsub_msg(
origin_id=pubsubs_fsub[0].my_id,
topic_ids=[TESTING_TOPIC],
data=TESTING_DATA,
seqno=b"\x33" * 8,
)
await pubsubs_fsub[0].push_msg(pubsubs_fsub[1].my_id, msg_0A)
await trio.sleep(0.01)
assert not event.is_set()
sub = await pubsubs_fsub[0].subscribe(TESTING_TOPIC) sub = await pubsubs_fsub[0].subscribe(TESTING_TOPIC)
# Test: `push_msg` succeeds with another unseen msg. # Test: `push_msg` succeeds with another unseen msg.
msg_1 = make_pubsub_msg( msg_1 = make_pubsub_msg(