mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test the full bootstrap discovery integration
|
|
"""
|
|
|
|
import secrets
|
|
|
|
import pytest
|
|
|
|
from libp2p import new_host
|
|
from libp2p.crypto.secp256k1 import create_new_key_pair
|
|
from libp2p.host.basic_host import BasicHost
|
|
|
|
|
|
@pytest.mark.trio
|
|
async def test_bootstrap_integration():
|
|
"""Test bootstrap integration with new_host"""
|
|
# Test bootstrap addresses
|
|
bootstrap_addrs = [
|
|
"/ip4/104.131.131.82/tcp/4001/p2p/QmaCpDMGvV2BGHeYERUEnRQAwe3N8SznbYGzPwp8qDrq",
|
|
"/ip4/104.236.179.241/tcp/4001/p2p/QmSoLPppuBtQSGwKDZT2M73ULpjvfd3aZ6ha4oFGL1KrGM",
|
|
]
|
|
|
|
# Generate key pair
|
|
secret = secrets.token_bytes(32)
|
|
key_pair = create_new_key_pair(secret)
|
|
|
|
# Create host with bootstrap
|
|
host = new_host(key_pair=key_pair, bootstrap=bootstrap_addrs)
|
|
|
|
# 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"
|
|
assert host.bootstrap is not None, "Bootstrap discovery should be initialized"
|
|
assert len(host.bootstrap.bootstrap_addrs) == len(bootstrap_addrs), (
|
|
"Bootstrap addresses should match"
|
|
)
|
|
|
|
|
|
def test_bootstrap_no_addresses():
|
|
"""Test that bootstrap is not initialized when no addresses provided"""
|
|
secret = secrets.token_bytes(32)
|
|
key_pair = create_new_key_pair(secret)
|
|
|
|
# Create host without bootstrap
|
|
host = new_host(key_pair=key_pair)
|
|
|
|
# Verify bootstrap is not initialized
|
|
assert isinstance(host, BasicHost)
|
|
assert not hasattr(host, "bootstrap") or host.bootstrap is None, (
|
|
"Bootstrap should not be initialized when no addresses provided"
|
|
)
|