Skip to content

Commit

Permalink
added ping_service tests
Browse files Browse the repository at this point in the history
  • Loading branch information
mystical-prog committed Oct 26, 2024
1 parent 996e689 commit 1073b3c
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions tests/core/host/test_ping.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
from libp2p.host.ping import (
ID,
PING_LENGTH,
PingService,
)
from libp2p.tools.factories import (
host_pair_factory,
Expand Down Expand Up @@ -47,3 +48,28 @@ async def test_ping_several(security_protocol):
# NOTE: this interval can be `0` for this test.
await trio.sleep(0)
await stream.close()


@pytest.mark.trio
async def test_ping_service_once(security_protocol):
async with host_pair_factory(security_protocol=security_protocol) as (
host_a,
host_b,
):
ping_service = PingService(host_b)
rtts = await ping_service.ping(host_a.get_id())
assert len(rtts) == 1
assert rtts[0] < 10**6


@pytest.mark.trio
async def test_ping_service_several(security_protocol):
async with host_pair_factory(security_protocol=security_protocol) as (
host_a,
host_b,
):
ping_service = PingService(host_b)
rtts = await ping_service.ping(host_a.get_id(), ping_amt=SOME_PING_COUNT)
assert len(rtts) == SOME_PING_COUNT
for rtt in rtts:
assert rtt < 10**6

0 comments on commit 1073b3c

Please sign in to comment.