From 934ee09d5f78761ea6c6377b81ab5a9f974fdb2d Mon Sep 17 00:00:00 2001
From: Yuki Kishimoto <yukikishimoto@protonmail.com>
Date: Sat, 8 Feb 2025 10:05:53 +0100
Subject: [PATCH] pool: add ephemeral event subscription unit test

Signed-off-by: Yuki Kishimoto <yukikishimoto@protonmail.com>
---
 crates/nostr-relay-pool/src/relay/mod.rs | 67 ++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/crates/nostr-relay-pool/src/relay/mod.rs b/crates/nostr-relay-pool/src/relay/mod.rs
index d59c45203..b40581066 100644
--- a/crates/nostr-relay-pool/src/relay/mod.rs
+++ b/crates/nostr-relay-pool/src/relay/mod.rs
@@ -1142,5 +1142,72 @@ mod tests {
         assert!(res.is_ok());
     }
 
+    #[tokio::test]
+    async fn test_subscribe_ephemeral_event() {
+        // Mock relay
+        let mock = MockRelay::run().await.unwrap();
+        let url = RelayUrl::parse(&mock.url()).unwrap();
+
+        // Sender
+        let relay1 = Relay::new(url.clone());
+        relay1
+            .try_connect(Duration::from_millis(500))
+            .await
+            .unwrap();
+
+        // Fetcher
+        let relay2 = Relay::new(url);
+        relay2
+            .try_connect(Duration::from_millis(500))
+            .await
+            .unwrap();
+
+        // Signer
+        let keys = Keys::generate();
+
+        // Event
+        let kind = Kind::Custom(22_222); // Ephemeral kind
+        let event: Event = EventBuilder::new(kind, "").sign_with_keys(&keys).unwrap();
+
+        let event_id: EventId = event.id;
+
+        tokio::spawn(async move {
+            tokio::time::sleep(Duration::from_secs(2)).await;
+            relay1.send_event(&event).await.unwrap();
+        });
+
+        // Subscribe
+        let filter = Filter::new().kind(kind);
+        let sub_id = relay2
+            .subscribe(filter, SubscribeOptions::default())
+            .await
+            .unwrap();
+
+        // Listen for notifications
+        let fut = relay2.handle_notifications(|notification| async {
+            if let RelayNotification::Event {
+                subscription_id,
+                event,
+            } = notification
+            {
+                if subscription_id == sub_id {
+                    if event.id == event_id {
+                        return Ok(true);
+                    } else {
+                        panic!("Unexpected event");
+                    }
+                } else {
+                    panic!("Unexpected subscription ID");
+                }
+            }
+            Ok(false)
+        });
+
+        tokio::time::timeout(Duration::from_secs(5), fut)
+            .await
+            .unwrap()
+            .unwrap();
+    }
+
     // TODO: add negentropy reconciliation test
 }