test_wif(): import from highlevelcrypto and add encoding checks

This commit is contained in:
Dmitri Bogomolov 2021-07-29 21:37:01 +03:00
parent 8942f63d06
commit 659b292357
Signed by untrusted user: g1itch
GPG Key ID: 720A756F18DEED13

View File

@ -2,7 +2,7 @@
import unittest import unittest
from binascii import unhexlify from binascii import unhexlify
from pybitmessage import addresses, shared from pybitmessage import addresses, highlevelcrypto
from .samples import ( from .samples import (
sample_address, sample_daddr3_512, sample_daddr4_512, sample_address, sample_daddr3_512, sample_daddr4_512,
@ -65,9 +65,21 @@ class TestAddresses(unittest.TestCase):
"""Decode WIFs of [chan] bitmessage and check the keys""" """Decode WIFs of [chan] bitmessage and check the keys"""
self.assertEqual( self.assertEqual(
sample_wif_privsigningkey, sample_wif_privsigningkey,
shared.decodeWalletImportFormat( highlevelcrypto.decodeWalletImportFormat(
b'5K42shDERM5g7Kbi3JT5vsAWpXMqRhWZpX835M2pdSoqQQpJMYm')) b'5K42shDERM5g7Kbi3JT5vsAWpXMqRhWZpX835M2pdSoqQQpJMYm'))
self.assertEqual( self.assertEqual(
sample_wif_privencryptionkey, sample_wif_privencryptionkey,
shared.decodeWalletImportFormat( highlevelcrypto.decodeWalletImportFormat(
b'5HwugVWm31gnxtoYcvcK7oywH2ezYTh6Y4tzRxsndAeMi6NHqpA')) b'5HwugVWm31gnxtoYcvcK7oywH2ezYTh6Y4tzRxsndAeMi6NHqpA'))
self.assertEqual(
b'5K42shDERM5g7Kbi3JT5vsAWpXMqRhWZpX835M2pdSoqQQpJMYm',
highlevelcrypto.encodeWalletImportFormat(
sample_wif_privsigningkey))
self.assertEqual(
b'5HwugVWm31gnxtoYcvcK7oywH2ezYTh6Y4tzRxsndAeMi6NHqpA',
highlevelcrypto.encodeWalletImportFormat(
sample_wif_privencryptionkey))
with self.assertRaises(ValueError):
highlevelcrypto.decodeWalletImportFormat(
b'5HwugVWm31gnxtoYcvcK7oywH2ezYTh6Y4tzRxsndAeMi6NHq')