Function test_three_nodes

p2p/tests/node.rs:161–212

()

#[test(tokio::test(flavor = "multi_thread", worker_threads = 2))]
async fn test_three_nodes() -> Result<()> {
    let mut test_runner_a = TestRunnerBuilder::new().build().await?;
    let mut test_runner_b = TestRunnerBuilder::new().build().await?;
    let mut test_runner_c = TestRunnerBuilder::new().build().await?;

    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);
    let peer_id_c = test_runner_c.client.local_peer_id().await?;
    assert_eq!(test_runner_c.peer_id, peer_id_c);

    // connect a to b
    test_runner_a
        .client
        .connect(peer_id_b, vec![test_runner_b.addr.clone()])
        .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?);

    // connect b to c
    test_runner_b
        .client
        .connect(peer_id_c, vec![test_runner_c.addr.clone()])
        .await?;

    timeout(
        Duration::from_millis(500),
        wait_bi_connected(&mut test_runner_b, &mut test_runner_c),
    )
    .await
    .context("waiting for b and c to connect")?;
    assert!(is_bi_connected(&test_runner_b, &test_runner_c).await?);

    // We expect that a and c find each other through b and become connected
    timeout(
        //This one takes longer as we need recon to run and sync peers
        Duration::from_millis(3_000),
        wait_bi_connected(&mut test_runner_a, &mut test_runner_c),
    )
    .await
    .context("waiting for a and c to connect")?;
    assert!(is_bi_connected(&test_runner_a, &test_runner_c).await?);

    Ok(())
}

// Reports if peers are connected to the other peer in both directions
async fn is_bi_connected(a: &TestRunner, b: &TestRunner) -> Result<bool> {