e2e_test/
e2e_test.rs

1use anyhow::Result;
2use iroh::{Endpoint, SecretKey};
3use iroh_gossip::net::Gossip;
4
5// Imports from distrubuted-topic-tracker
6use distributed_topic_tracker::{AutoDiscoveryGossip, RecordPublisher, TopicId};
7
8#[tokio::main]
9async fn main() -> Result<()> {
10    // Generate a new random secret key
11    let secret_key = SecretKey::generate(&mut rand::rng());
12    let signing_key = mainline::SigningKey::from_bytes(&secret_key.to_bytes());
13
14    // Set up endpoint with discovery enabled
15    let endpoint = Endpoint::builder()
16        .secret_key(secret_key.clone())
17        .bind()
18        .await?;
19
20    // Initialize gossip with auto-discovery
21    let gossip = Gossip::builder().spawn(endpoint.clone());
22
23    // Set up protocol router
24    let _router = iroh::protocol::Router::builder(endpoint.clone())
25        .accept(iroh_gossip::ALPN, gossip.clone())
26        .spawn();
27
28    let topic_id = TopicId::new("my-iroh-gossip-topic".to_string());
29    let initial_secret = b"my-initial-secret".to_vec();
30
31    let record_publisher = RecordPublisher::new(
32        topic_id.clone(),
33        signing_key.verifying_key(),
34        signing_key.clone(),
35        None,
36        initial_secret,
37    );
38    let (gossip_sender, gossip_receiver) = gossip
39        .subscribe_and_join_with_auto_discovery(record_publisher)
40        .await?
41        .split()
42        .await?;
43
44    tokio::spawn(async move {
45        while let Some(Ok(event)) = gossip_receiver.next().await {
46            println!("event: {event:?}");
47        }
48    });
49
50    tokio::time::sleep(std::time::Duration::from_secs(3)).await;
51    gossip_sender
52        .broadcast(format!("hi from {}", endpoint.id()).into())
53        .await?;
54
55    println!("[joined topic]");
56
57    tokio::time::sleep(std::time::Duration::from_secs(10)).await;
58
59    println!("[finished]");
60
61    // successfully joined
62    // exit with code 0
63    Ok(())
64}
e2e_test/e2e_test.rs

e2e_test/
e2e_test.rs
