Function test_two_nodes

p2p/tests/node.rs:134–158

()

#[test(tokio::test(flavor = "multi_thread", worker_threads = 2))]
async fn test_two_nodes() -> Result<()> {
    let mut test_runner_a = TestRunnerBuilder::new().build().await?;
    let mut test_runner_b = TestRunnerBuilder::new().build().await?;
    let addrs_b = vec![test_runner_b.addr.clone()];
    debug!(?test_runner_a.peer_id, ?test_runner_b.peer_id, "peer ids");

    let peer_id_a = test_runner_a.client.local_peer_id().await?;
    assert_eq!(test_runner_a.peer_id, peer_id_a);
    let peer_id_b = test_runner_b.client.local_peer_id().await?;
    assert_eq!(test_runner_b.peer_id, peer_id_b);

    // connect
    test_runner_a.client.connect(peer_id_b, addrs_b).await?;

    timeout(
        Duration::from_millis(500),
        wait_bi_connected(&mut test_runner_a, &mut test_runner_b),
    )
    .await
    .context("waiting for a and b to connect")?;

    assert!(is_bi_connected(&test_runner_a, &test_runner_b).await?);

    Ok(())
}

#[test(tokio::test(flavor = "multi_thread", worker_threads = 2))]
async fn test_three_nodes() -> Result<()> {