remove obsolete test script and add comprehensive validation tests for bootstrap addresses

Author: sumanjeet0012

tests/discovery/bootstrap/test_bootstrap_script.py

@@ -3,33 +3,18 @@
 Test the full bootstrap discovery integration
 """
 
-import logging
 import secrets
 
 import pytest
 
 from libp2p import new_host
-from libp2p.abc import PeerInfo
 from libp2p.crypto.secp256k1 import create_new_key_pair
-from libp2p.discovery.events.peerDiscovery import peerDiscovery
 from libp2p.host.basic_host import BasicHost
 
-# Configure logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger("bootstrap_test")
-
-
-def on_peer_discovery(peer_info: PeerInfo) -> None:
-    """Handler for peer discovery events."""
-    logger.info(f"🔍 Discovered peer: {peer_info.peer_id}")
-    logger.info(f"   Addresses: {[str(addr) for addr in peer_info.addrs]}")
-
 
 @pytest.mark.trio
 async def test_bootstrap_integration():
     """Test bootstrap integration with new_host"""
-    print("🧪 Testing Bootstrap Integration")
-
     # Test bootstrap addresses
     bootstrap_addrs = [
         "/ip4/104.131.131.82/tcp/4001/p2p/QmaCpDMGvV2BGHeYERUEnRQAwe3N8SznbYGzPwp8qDrq",
@@ -40,18 +25,9 @@ async def test_bootstrap_integration():
     secret = secrets.token_bytes(32)
     key_pair = create_new_key_pair(secret)
 
-    # Register peer discovery handler
-    peerDiscovery.register_peer_discovered_handler(on_peer_discovery)
-
-    print(f"🌐 Testing with {len(bootstrap_addrs)} bootstrap peers")
-
     # Create host with bootstrap
     host = new_host(key_pair=key_pair, bootstrap=bootstrap_addrs)
 
-    print("✅ Successfully created host with bootstrap")
-    print(f"📍 Host peer ID: {host.get_id()}")
-    print("🔗 Bootstrap discovery should process peers when host starts")
-
     # Verify bootstrap discovery is set up (cast to BasicHost for type checking)
     assert isinstance(host, BasicHost), "Host should be a BasicHost instance"
     assert hasattr(host, "bootstrap"), "Host should have bootstrap attribute"
@@ -60,8 +36,6 @@ async def test_bootstrap_integration():
         "Bootstrap addresses should match"
     )
 
-    print("🎉 Bootstrap integration test completed successfully!")
-
 
 def test_bootstrap_no_addresses():
     """Test that bootstrap is not initialized when no addresses provided"""

tests/discovery/bootstrap/test_integration.py

@@ -0,0 +1,35 @@
+#!/usr/bin/env python3
+"""
+Test bootstrap address validation
+"""
+
+import pytest
+from libp2p.discovery.bootstrap.utils import (
+    parse_bootstrap_peer_info,
+    validate_bootstrap_addresses,
+)
+
+def test_validate_addresses():
+    """Test validation with a mix of valid and invalid addresses in one list."""
+    addresses = [
+        # Valid
+        "/ip4/104.131.131.82/tcp/4001/p2p/QmaCpDMGvV2BGHeYERUEnRQAwe3N8SznbYGzPwp8qDrq",
+        "/ip4/104.236.179.241/tcp/4001/p2p/QmSoLPppuBtQSGwKDZT2M73ULpjvfd3aZ6ha4oFGL1KrGM",
+        # Invalid
+        "invalid-address",
+        "/ip4/192.168.1.1/tcp/4001",  # Missing p2p part
+        "",  # Empty
+        "/ip4/127.0.0.1/tcp/4001/p2p/InvalidPeerID",  # Bad peer ID
+    ]
+    valid_expected = [
+        "/ip4/104.131.131.82/tcp/4001/p2p/QmaCpDMGvV2BGHeYERUEnRQAwe3N8SznbYGzPwp8qDrq",
+        "/ip4/104.236.179.241/tcp/4001/p2p/QmSoLPppuBtQSGwKDZT2M73ULpjvfd3aZ6ha4oFGL1KrGM",
+    ]
+    validated = validate_bootstrap_addresses(addresses)
+    assert validated == valid_expected, f"Expected only valid addresses, got: {validated}"
+    for addr in addresses:
+        peer_info = parse_bootstrap_peer_info(addr)
+        if addr in valid_expected:
+            assert peer_info is not None and peer_info.peer_id is not None, f"Should parse valid address: {addr}"
+        else:
+            assert peer_info is None, f"Should not parse invalid address: {addr}"