mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-30 03:18:20 +00:00
e073e619c9
isort set to skip the first necessary "from common import *" line. A better solution would be to get rid of the need of this import in the future. [no changelog]
80 lines
2.4 KiB
Python
80 lines
2.4 KiB
Python
from common import * # isort:skip
|
|
|
|
from trezor.wire import ProcessError
|
|
|
|
if not utils.BITCOIN_ONLY:
|
|
from apps.stellar.helpers import address_from_public_key, public_key_from_address
|
|
|
|
|
|
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
|
|
class TestStellarAddress(unittest.TestCase):
|
|
def test_address_to_pubkey(self):
|
|
self.assertEqual(
|
|
public_key_from_address(
|
|
"GBOVKZBEM2YYLOCDCUXJ4IMRKHN4LCJAE7WEAEA2KF562XFAGDBOB64V"
|
|
),
|
|
unhexlify(
|
|
"5d55642466b185b843152e9e219151dbc5892027ec40101a517bed5ca030c2e0"
|
|
),
|
|
)
|
|
|
|
self.assertEqual(
|
|
public_key_from_address(
|
|
"GCN2K2HG53AWX2SP5UHRPMJUUHLJF2XBTGSXROTPWRGAYJCDDP63J2U6"
|
|
),
|
|
unhexlify(
|
|
"9ba568e6eec16bea4fed0f17b134a1d692eae199a578ba6fb44c0c24431bfdb4"
|
|
),
|
|
)
|
|
|
|
def test_pubkey_to_address(self):
|
|
addr = address_from_public_key(
|
|
unhexlify(
|
|
"5d55642466b185b843152e9e219151dbc5892027ec40101a517bed5ca030c2e0"
|
|
)
|
|
)
|
|
self.assertEqual(
|
|
addr, "GBOVKZBEM2YYLOCDCUXJ4IMRKHN4LCJAE7WEAEA2KF562XFAGDBOB64V"
|
|
)
|
|
|
|
addr = address_from_public_key(
|
|
unhexlify(
|
|
"9ba568e6eec16bea4fed0f17b134a1d692eae199a578ba6fb44c0c24431bfdb4"
|
|
)
|
|
)
|
|
self.assertEqual(
|
|
addr, "GCN2K2HG53AWX2SP5UHRPMJUUHLJF2XBTGSXROTPWRGAYJCDDP63J2U6"
|
|
)
|
|
|
|
def test_both(self):
|
|
pubkey = unhexlify(
|
|
"dfcc77d08588601702e02de2dc603f5c5281bea23baa894ae3b3b4778e5bbe40"
|
|
)
|
|
self.assertEqual(
|
|
public_key_from_address(address_from_public_key(pubkey)), pubkey
|
|
)
|
|
|
|
pubkey = unhexlify(
|
|
"53214e6155469c32fb882b1b1d94930d5445a78202867b7ddc6a33ad42ff4464"
|
|
)
|
|
self.assertEqual(
|
|
public_key_from_address(address_from_public_key(pubkey)), pubkey
|
|
)
|
|
|
|
pubkey = unhexlify(
|
|
"5ed4690134e5ef79b290ea1e7a4b8f3b6b3bcf287463c18bfe36baa030e7efbd"
|
|
)
|
|
self.assertEqual(
|
|
public_key_from_address(address_from_public_key(pubkey)), pubkey
|
|
)
|
|
|
|
def test_invalid_address(self):
|
|
with self.assertRaises(ProcessError):
|
|
public_key_from_address(
|
|
"GCN2K2HG53AWX2SP5UHRPMJUUHLJF2XBTGSXROTPWRGAYJCDDP63J2AA"
|
|
) # invalid checksum
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|