#!/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"
    )