Improved content id search mechanics

2026-07-23 15:28:03 +03:00
parent daea24042c
commit a69e651250
7 changed files with 461 additions and 142 deletions
@@ -0,0 +1,98 @@
+//! Focused integration test: a track published by one node must be
+//! resolvable by content id from another node (the share-link flow).
+
+use std::path::Path;
+use std::time::Duration;
+
+use music_dht::{
+    ItemKind, ItemSpec, MusicDhtConfig, MusicDhtEventReceiver, MusicDhtService, NetworkId,
+};
+
+const TEST_TIMEOUT: Duration = Duration::from_secs(240);
+
+const CONTENT_ID: &str = "b3:661eb31d76ab7cef89a19bb8e978c0eb357ae62395a73fa5539a0efd35110dd9";
+
+type Started = (MusicDhtService, MusicDhtEventReceiver);
+
+async fn start(dir: &Path, network: &str) -> Started {
+    let config = MusicDhtConfig::builder()
+        .data_dir(dir)
+        .network_id(NetworkId::from_name(network))
+        .republish_interval(Duration::from_secs(5))
+        .expire_interval(Duration::from_secs(2))
+        .request_timeout(Duration::from_secs(5))
+        .lookup_timeout(Duration::from_secs(10))
+        .build()
+        .expect("valid config");
+    MusicDhtService::start(config)
+        .await
+        .expect("service starts")
+}
+
+#[tokio::test]
+async fn share_link_content_id_lookup() {
+    tokio::time::timeout(TEST_TIMEOUT, async {
+        let dir_a = tempfile::tempdir().expect("tempdir");
+        let dir_b = tempfile::tempdir().expect("tempdir");
+
+        let (a, _events_a) = start(dir_a.path(), "content-id-test").await;
+        let ticket_a = a.ticket().await.expect("ticket");
+        let (b, _events_b) = start(dir_b.path(), "content-id-test").await;
+        b.connect(ticket_a).await.expect("b connects to a");
+
+        // B publishes one track with a content id and many featured artists
+        // (mirrors the real-world failing share link).
+        let spec = ItemSpec {
+            local_key: "track:1".into(),
+            kind: ItemKind::Track,
+            name: "Ежемесячные".into(),
+            artist_names: vec!["Основной Артист".into()],
+            featured_artist_names: vec![
+                "Pyrokinesis".into(),
+                "Блёв МС".into(),
+                "Артем Татищевский".into(),
+                "WormGanger".into(),
+                "Рудбой".into(),
+                "АнальгиН-56 школа".into(),
+            ],
+            year: Some(2024),
+            release_type: Some("album".into()),
+            release_title: Some("Тестовый релиз".into()),
+            track_number: Some(1),
+            disc_number: Some(1),
+            duration_seconds: Some(200.0),
+            content_id: Some(CONTENT_ID.into()),
+        };
+        let stats = b.sync_library(vec![spec]).await.expect("sync");
+        assert_eq!(stats.added, 1, "track is published");
+        assert_eq!(stats.failed, 0);
+
+        // A resolves the share link by content id.
+        let started = std::time::Instant::now();
+        let found = loop {
+            let outcome = a
+                .search_content_id(CONTENT_ID)
+                .await
+                .expect("content id search");
+            let hit = outcome
+                .network_results
+                .iter()
+                .find(|item| item.kind == ItemKind::Track);
+            if let Some(hit) = hit {
+                break hit.clone();
+            }
+            assert!(
+                started.elapsed() < Duration::from_secs(45),
+                "timed out resolving the content id from the other node"
+            );
+            tokio::time::sleep(Duration::from_millis(500)).await;
+        };
+        assert_eq!(found.content_id.as_deref(), Some(CONTENT_ID));
+        assert_eq!(found.owner, b.endpoint_id());
+
+        a.shutdown().await.expect("shutdown a");
+        b.shutdown().await.expect("shutdown b");
+    })
+    .await
+    .expect("test timed out");
+}