cleanup

lnbits · Jul 19, 2024 · d04b75c · d04b75c
1 parent 9f69073
commit d04b75c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/bolt11/models/signature.py b/bolt11/models/signature.py
@@ -51,8 +51,9 @@ def recover_public_key(self) -> str:
         if not self.signing_data:
             raise ValueError("No signing data")
 
-        message = bytearray([ord(c) for c in self.hrp]) + self.signing_data
-        key = PublicKey.from_signature_and_message(self.signature_data, message)
+        key = PublicKey.from_signature_and_message(
+            self.signature_data, message(self.hrp, self.signing_data)
+        )
         return key.format(compressed=True).hex()
 
     @property

diff --git a/tests/test_signature.py b/tests/test_signature.py
@@ -7,18 +7,17 @@
 
 
 class TestBolt11Signature:
-    def test_recovers_public_key(self):
+    def test_signature(self):
         signature = Signature.from_private_key(
             hrp="lnbc1",
             private_key=ex["private_key"],
             signing_data=b"1234567890",
         )
         assert signature.recover_public_key() == ex["public_key"]
 
-    def test_signature_verify(self):
-        signature = Signature.from_private_key(
+        signature = Signature.from_signature_data(
             hrp="lnbc1",
-            private_key=ex["private_key"],
             signing_data=b"1234567890",
+            signature_data=signature.signature_data,
         )
         assert signature.verify(ex["public_key"])