Add a test for short pubkey coordinates

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

View File

@ -1,5 +1,6 @@
"""Tests for ECC object"""
import os
import unittest
from hashlib import sha512
@ -30,6 +31,22 @@ class TestECC(unittest.TestCase):
pubkey = eccobj.get_pubkey()
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):
"""Check keys decoding"""
# pylint: disable=protected-access