Bitmessage/PyBitmessage-2024-12-13
Archived
2
0
Fork 0
Code Issues 168 Pull Requests 53 Actions Packages Projects Releases 5 Wiki Activity

Add a test for short pubkey coordinates

Browse Source
This commit is contained in:
Lee Miller 2024-01-07 03:20:51 +02:00
parent f88ffccfb0
commit d3644aa354
Signed by untrusted user: lee.miller
GPG Key ID: 4F97A5EA88F4AB63
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/pyelliptic/tests/test_ecc.py
View File

@ -1,5 +1,6 @@
"""Tests for ECC object""" """Tests for ECC object"""
import os
import unittest import unittest
from hashlib import sha512 from hashlib import sha512
@ -30,6 +31,22 @@ class TestECC(unittest.TestCase):
pubkey = eccobj.get_pubkey() pubkey = eccobj.get_pubkey()
self.assertEqual(pubkey[:4], b'\x02\xca\x00\x20') self.assertEqual(pubkey[:4], b'\x02\xca\x00\x20')
def test_short_keys(self):
"""Check formatting of the keys with leading zeroes"""
def sample_key(_):
return os.urandom(32), os.urandom(31), os.urandom(30)
try:
gen_orig = pyelliptic.ECC._generate
pyelliptic.ECC._generate = sample_key
eccobj = pyelliptic.ECC(curve='secp256k1')
pubkey = eccobj.get_pubkey()
self.assertEqual(pubkey[:4], b'\x02\xca\x00\x20')
self.assertEqual(pubkey[36:38], b'\x00\x20')
self.assertEqual(len(pubkey[38:]), 32)
finally:
pyelliptic.ECC._generate = gen_orig
def test_decode_keys(self): def test_decode_keys(self):
"""Check keys decoding""" """Check keys decoding"""
# pylint: disable=protected-access # pylint: disable=protected-access