From b58637b0ead1742b9f4f5914e637701d1c469fb4 Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Mon, 30 Jun 2025 22:08:45 -0400
Subject: [PATCH] Add pynostr workflow test

---
 src/requirements.txt               |  1 +
 src/tests/test_pynostr_workflow.py | 70 ++++++++++++++++++++++++++++++
 2 files changed, 71 insertions(+)
 create mode 100644 src/tests/test_pynostr_workflow.py

diff --git a/src/requirements.txt b/src/requirements.txt
index 74ad6f9..28a26a8 100644
--- a/src/requirements.txt
+++ b/src/requirements.txt
@@ -14,3 +14,4 @@ portalocker>=2.8
 pynostr>=0.6.2
 websocket-client==1.7.0
 
+websockets>=15.0.0
diff --git a/src/tests/test_pynostr_workflow.py b/src/tests/test_pynostr_workflow.py
new file mode 100644
index 0000000..0045975
--- /dev/null
+++ b/src/tests/test_pynostr_workflow.py
@@ -0,0 +1,70 @@
+import asyncio
+import json
+import threading
+import time
+from websocket import create_connection
+
+import websockets
+from nostr.key_manager import KeyManager
+from pynostr.event import Event, EventKind
+
+
+class FakeRelay:
+    def __init__(self):
+        self.events = []
+
+    async def handler(self, ws):
+        async for message in ws:
+            data = json.loads(message)
+            if data[0] == "EVENT":
+                event = data[1]
+                self.events.append(event)
+                await ws.send(json.dumps(["OK", event["id"], True, ""]))
+            elif data[0] == "REQ":
+                sub_id = data[1]
+                for event in self.events:
+                    await ws.send(json.dumps(["EVENT", sub_id, event]))
+                await ws.send(json.dumps(["EOSE", sub_id]))
+
+
+def run_relay(relay, host="localhost", port=8765):
+    async def main():
+        async with websockets.serve(relay.handler, host, port):
+            await asyncio.Future()
+
+    asyncio.run(main())
+
+
+def test_pynostr_send_receive(tmp_path):
+    relay = FakeRelay()
+    thread = threading.Thread(target=run_relay, args=(relay,), daemon=True)
+    thread.start()
+
+    time.sleep(0.5)
+
+    seed = "abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon abandon about"
+    fingerprint = "test"
+    km = KeyManager(seed, fingerprint)
+
+    ws = create_connection("ws://localhost:8765")
+
+    event = Event(kind=EventKind.TEXT_NOTE, content="hello")
+    event.pubkey = km.get_public_key_hex()
+    event.created_at = int(time.time())
+    event.sign(km.get_private_key_hex())
+
+    ws.send(event.to_message())
+    sub_id = "1"
+    ws.send(json.dumps(["REQ", sub_id, {}]))
+
+    received = None
+    while True:
+        msg = json.loads(ws.recv())
+        if msg[0] == "EVENT":
+            received = msg[2]
+        elif msg[0] == "EOSE":
+            break
+    ws.close()
+
+    assert received is not None
+    assert received["content"] == "hello"