fix(test): update python kv store test

Author: enigbe

bindings/python/src/ldk_node/test_ldk_node.py

@@ -6,10 +6,12 @@
 import re
 import requests
 import threading
+import asyncio
 
 from abc import ABC, abstractmethod
 from typing import List
 
+import ldk_node
 from ldk_node import *
 
 DEFAULT_ESPLORA_SERVER_URL = "http://127.0.0.1:3002"
@@ -18,19 +20,35 @@
 
 class AbstractKvStore(ABC):
     @abstractmethod
-    def list(self, primary_namespace: str,secondary_namespace: str):
+    async def list_async(self, primary_namespace: "str",secondary_namespace: "str") -> "typing.List[str]":
         pass
 
     @abstractmethod
-    def read(self, primary_namespace: str,secondary_namespace: str,key: str):
+    def list_sync(self, primary_namespace: "str",secondary_namespace: "str") -> "typing.List[str]":
         pass
 
     @abstractmethod
-    def remove(self, primary_namespace: str,secondary_namespace: str,key: str,lazy: bool):
+    async def read_async(self, primary_namespace: "str",secondary_namespace: "str",key: "str") -> "typing.List[int]":
         pass
 
     @abstractmethod
-    def write(self, primary_namespace: str,secondary_namespace: str,key: str,buf: List[int]):
+    def read_sync(self, primary_namespace: "str",secondary_namespace: "str",key: "str") -> "typing.List[int]":
+        pass
+
+    @abstractmethod
+    async def remove_async(self, primary_namespace: "str",secondary_namespace: "str",key: "str",lazy: "bool") -> None:
+        pass
+
+    @abstractmethod
+    def remove_sync(self, primary_namespace: "str",secondary_namespace: "str",key: "str",lazy: "bool") -> None:
+        pass
+
+    @abstractmethod
+    async def write_async(self, primary_namespace: "str",secondary_namespace: "str",key: "str",buf: "typing.List[int]") -> None:
+        pass
+
+    @abstractmethod
+    def write_sync(self, primary_namespace: "str",secondary_namespace: "str",key: "str",buf: "typing.List[int]") -> None:
         pass
 
 class TestKvStore(AbstractKvStore):
@@ -40,14 +58,15 @@ def __init__(self, name: str):
         self.storage = {}
         self._lock = threading.Lock()
 
-    def list(self, primary_namespace: str,secondary_namespace: str) -> List[str]:
+    # Sync methods
+    def list_sync(self, primary_namespace: str, secondary_namespace: str) -> List[str]:
         with self._lock:
             namespace_key = (primary_namespace, secondary_namespace)
             if namespace_key in self.storage:
                 return list(self.storage[namespace_key].keys())
             return []
 
-    def read(self, primary_namespace: str,secondary_namespace: str,key: str) -> List[int]:
+    def read_sync(self, primary_namespace: str, secondary_namespace: str, key: str) -> List[int]:
         with self._lock:
             print(f"[{self.name}] READ: {primary_namespace}/{secondary_namespace}/{key}")
             namespace_key = (primary_namespace, secondary_namespace)
@@ -64,7 +83,15 @@ def read(self, primary_namespace: str,secondary_namespace: str,key: str) -> List
             print(f"  -> returning {len(data)} bytes")
             return data
 
-    def remove(self, primary_namespace: str,secondary_namespace: str,key: str,lazy: bool) -> None:
+    def write_sync(self, primary_namespace: str, secondary_namespace: str, key: str, buf: List[int]) -> None:
+        with self._lock:
+            namespace_key = (primary_namespace, secondary_namespace)
+            if namespace_key not in self.storage:
+                self.storage[namespace_key] = {}
+
+            self.storage[namespace_key][key] = buf.copy()
+
+    def remove_sync(self, primary_namespace: str, secondary_namespace: str, key: str, lazy: bool) -> None:
         with self._lock:
             namespace_key = (primary_namespace, secondary_namespace)
             if namespace_key not in self.storage:
@@ -78,13 +105,18 @@ def remove(self, primary_namespace: str,secondary_namespace: str,key: str,lazy:
             if not self.storage[namespace_key]:
                 del self.storage[namespace_key]
 
-    def write(self, primary_namespace: str,secondary_namespace: str,key: str,buf: List[int]) -> None:
-        with self._lock:
-            namespace_key = (primary_namespace, secondary_namespace)
-            if namespace_key not in self.storage:
-                self.storage[namespace_key] = {}
+    # Async methods
+    async def list_async(self, primary_namespace: str, secondary_namespace: str) -> List[str]:
+        return self.list_sync(primary_namespace, secondary_namespace)
 
-            self.storage[namespace_key][key] = buf.copy()
+    async def read_async(self, primary_namespace: str, secondary_namespace: str, key: str) -> List[int]:
+        return self.read_sync(primary_namespace, secondary_namespace, key)
+
+    async def write_async(self, primary_namespace: str, secondary_namespace: str, key: str, buf: List[int]) -> None:
+        self.write_sync(primary_namespace, secondary_namespace, key, buf)
+
+    async def remove_async(self, primary_namespace: str, secondary_namespace: str, key: str, lazy: bool) -> None:
+        self.remove_sync(primary_namespace, secondary_namespace, key, lazy)
 
     def dump(self):
         print(f"\n[{self.name}] Store contents:")
@@ -250,6 +282,17 @@ def setup_node_with_tier_store(tmp_dir, esplora_endpoint, listening_addresses):
         backoff_multiplier=2.0
     )
 
+    # Set event loop for async Python callbacks from Rust
+    loop = asyncio.new_event_loop()
+    
+    def run_loop():
+        asyncio.set_event_loop(loop)
+        loop.run_forever()
+    
+    loop_thread = threading.Thread(target=run_loop, daemon=True)
+    loop_thread.start()
+    ldk_node.uniffi_set_event_loop(loop)
+
     builder = Builder.from_config(config)
     builder.set_storage_dir_path(tmp_dir)
     builder.set_chain_source_esplora(esplora_endpoint, None)