From c9c8eca9d0fc4f560063633697789b72aac32384 Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Thu, 3 Jul 2025 14:11:21 -0400
Subject: [PATCH] Add KeyManager helper test

---
 src/tests/test_key_manager_helpers.py | 29 +++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 src/tests/test_key_manager_helpers.py

diff --git a/src/tests/test_key_manager_helpers.py b/src/tests/test_key_manager_helpers.py
new file mode 100644
index 0000000..b133be8
--- /dev/null
+++ b/src/tests/test_key_manager_helpers.py
@@ -0,0 +1,29 @@
+import pytest
+from bech32 import bech32_encode, convertbits
+
+from nostr.key_manager import KeyManager
+
+
+def test_key_manager_getters(monkeypatch):
+    priv_hex = "1" * 64
+    pub_hex = "2" * 64
+
+    class DummyKeys:
+        def public_key_hex(self):
+            return pub_hex
+
+        def private_key_hex(self):
+            return priv_hex
+
+    monkeypatch.setattr(KeyManager, "initialize_bip85", lambda self: None)
+    monkeypatch.setattr(KeyManager, "generate_nostr_keys", lambda self: DummyKeys())
+
+    km = KeyManager("seed", "fp")
+
+    assert km.get_public_key_hex() == pub_hex
+    assert km.get_private_key_hex() == priv_hex
+
+    expected_npub = bech32_encode(
+        "npub", convertbits(bytes.fromhex(pub_hex), 8, 5, True)
+    )
+    assert km.get_npub() == expected_npub