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

added tests

Browse Source
This commit is contained in:
lla-dane
2025-07-02 19:43:31 +05:30
parent a89ba8ef81
commit 383d7cb722
2 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libp2p/identity/identify_push/identify_push.py
View File

@ -176,7 +176,9 @@ async def push_identify_to_peers(
host: IHost, host: IHost,
peer_ids: set[ID] | None = None, peer_ids: set[ID] | None = None,
observed_multiaddr: Multiaddr | None = None, observed_multiaddr: Multiaddr | None = None,
) -> None: counter: dict[str, int] | None = None,
lock: trio.Lock | None = None,
) -> int: # <-- return the max concurrency
""" """
Push an identify message to multiple peers in parallel. Push an identify message to multiple peers in parallel.
@ -191,3 +193,5 @@ async def push_identify_to_peers(
async with trio.open_nursery() as nursery: async with trio.open_nursery() as nursery:
for peer_id in peer_ids: for peer_id in peer_ids:
nursery.start_soon(limited_push, peer_id) nursery.start_soon(limited_push, peer_id)
return counter["max"] if counter else 0

1
tests/core/identity/identify_push/test_identify_push.py
View File

@ -10,6 +10,7 @@ import trio
from libp2p import ( from libp2p import (
new_host, new_host,
) )
from libp2p.abc import IHost
from libp2p.crypto.secp256k1 import ( from libp2p.crypto.secp256k1 import (
create_new_key_pair, create_new_key_pair,
) )