varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor: Update test_identify_push.py removing duplicated import

Browse Source
This commit is contained in:
acul71
2025-04-29 02:19:26 +02:00
committed by Paul Robinson
parent 5a05ca0124
commit e928e6a8cd
1 changed files with 14 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,6 +20,18 @@ from tests.utils.factories import (
host_pair_factory,
)
import multiaddr
from libp2p import (
new_host,
)
from libp2p.crypto.secp256k1 import (
create_new_key_pair,
)
from libp2p.peer.peerinfo import (
info_from_p2p_addr,
)
logger = logging.getLogger("libp2p.identity.identify-push-test")
@ -159,17 +171,7 @@ async def test_identify_push_to_peers(security_protocol):
):
# Create a third host
# Instead of using key_pair, create a new host directly
import multiaddr
from libp2p import (
new_host,
)
from libp2p.crypto.secp256k1 import (
create_new_key_pair,
)
from libp2p.peer.peerinfo import (
info_from_p2p_addr,
)
# Create a new key pair for host_c
key_pair_c = create_new_key_pair()
@ -187,10 +189,6 @@ async def test_identify_push_to_peers(security_protocol):
await host_c.connect(info_from_p2p_addr(host_b.get_addrs()[0]))
# Push identify information from host_a to all connected peers
from libp2p.identity.identify_push.identify_push import (
push_identify_to_peers,
)
await push_identify_to_peers(host_a)
# Wait a bit for the push to complete
@ -223,18 +221,7 @@ async def test_push_identify_to_peers_with_explicit_params(security_protocol):
This test ensures all parameters of push_identify_to_peers are properly tested.
"""
import multiaddr
from libp2p import (
new_host,
)
from libp2p.crypto.secp256k1 import (
create_new_key_pair,
)
from libp2p.peer.peerinfo import (
info_from_p2p_addr,
)
# Create four hosts to thoroughly test selective pushing
async with host_pair_factory(security_protocol=security_protocol) as (
host_a,