diff --git a/examples/identify_push/identify_push_demo.py b/examples/identify_push/identify_push_demo.py index 9dc98729..31ab4099 100644 --- a/examples/identify_push/identify_push_demo.py +++ b/examples/identify_push/identify_push_demo.py @@ -104,9 +104,11 @@ async def main() -> None: logger.error(f"Error during identify push: {str(e)}") print(f"\nError during identify push: {str(e)}") + if __name__ == "__main__": trio.run(main) + def run_main(): """Non-async entry point for the console script.""" trio.run(main) diff --git a/tests/core/identity/identify_push/test_identify_push.py b/tests/core/identity/identify_push/test_identify_push.py index 1ddca76a..cfceb17a 100644 --- a/tests/core/identity/identify_push/test_identify_push.py +++ b/tests/core/identity/identify_push/test_identify_push.py @@ -1,8 +1,15 @@ import logging import pytest +import multiaddr import trio +from libp2p import ( + new_host, +) +from libp2p.crypto.secp256k1 import ( + create_new_key_pair, +) from libp2p.identity.identify.identify import ( _mk_identify_protobuf, ) @@ -16,21 +23,12 @@ from libp2p.identity.identify_push.identify_push import ( push_identify_to_peer, push_identify_to_peers, ) -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, ) +from tests.utils.factories import ( + host_pair_factory, +) logger = logging.getLogger("libp2p.identity.identify-push-test") @@ -171,7 +169,6 @@ async def test_identify_push_to_peers(security_protocol): ): # Create a third host # Instead of using key_pair, create a new host directly - # Create a new key pair for host_c key_pair_c = create_new_key_pair() @@ -221,7 +218,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. """ - + # Create four hosts to thoroughly test selective pushing async with host_pair_factory(security_protocol=security_protocol) as ( host_a,