e2e_test/
e2e_test.rs

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

e2e_test/
e2e_test.rs
