Refactor using of crypto functions #1796

Closed
g1itch wants to merge 14 commits from crypto into v0.6
Showing only changes of commit 803937290f - Show all commits

View File

@ -4,9 +4,10 @@ Test if OpenSSL is working correctly
import unittest import unittest
try: try:
from pyelliptic.ecc import ECC
from pyelliptic.openssl import OpenSSL from pyelliptic.openssl import OpenSSL
except ImportError: except ImportError:
from pybitmessage.pyelliptic import OpenSSL from pybitmessage.pyelliptic import ECC, OpenSSL
try: try:
OpenSSL.BN_bn2binpad OpenSSL.BN_bn2binpad
@ -55,3 +56,10 @@ class TestOpenSSL(unittest.TestCase):
if b.raw != c.raw.rjust(OpenSSL.BN_num_bytes(n), b'\x00'): if b.raw != c.raw.rjust(OpenSSL.BN_num_bytes(n), b'\x00'):
bad += 1 bad += 1
self.assertEqual(bad, 0) self.assertEqual(bad, 0)
def test_random_keys(self):
"""A dummy test for random keys in ECC object"""
eccobj = ECC(curve='secp256k1')
self.assertEqual(len(eccobj.privkey), 32)
pubkey = eccobj.get_pubkey()
self.assertEqual(pubkey[:4], b'\x02\xca\x00\x20')