1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-22 14:28:07 +00:00
trezor-firmware/core/tests/test_apps.ethereum.helpers.py
2023-03-24 13:24:46 +01:00

58 lines
2.1 KiB
Python

from common import *
from apps.common.paths import HARDENED
if not utils.BITCOIN_ONLY:
from apps.ethereum.helpers import address_from_bytes
from ethereum_common import make_network
@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
class TestEthereumGetAddress(unittest.TestCase):
def test_address_from_bytes_eip55(self):
# https://github.com/ethereum/EIPs/blob/master/EIPS/eip-55.md
eip55 = [
"0x52908400098527886E0F7030069857D2E4169EE7",
"0x8617E340B3D01FA5F11F306F4090FD50E238070D",
"0xde709f2102306220921060314715629080e2fb77",
"0x27b1fdb04752bbc536007a920d24acb045561c26",
"0x5aAeb6053F3E94C9b9A09f33669435E7Ef1BeAed",
"0xfB6916095ca1df60bB79Ce92cE3Ea74c37c5d359",
"0xdbF03B407c01E7cD3CBea99509d93f8DDDC8C6FB",
"0xD1220A0cf47c7B9Be7A2E6BA89F429762e7b9aDb",
]
for s in eip55:
b = unhexlify(s[2:])
h = address_from_bytes(b)
self.assertEqual(h, s)
def test_address_from_bytes_rskip60(self):
# https://github.com/rsksmart/RSKIPs/blob/master/IPs/RSKIP60.md
rskip60_chain_30 = [
"0x5aaEB6053f3e94c9b9a09f33669435E7ef1bEAeD",
"0xFb6916095cA1Df60bb79ce92cE3EA74c37c5d359",
"0xDBF03B407c01E7CD3cBea99509D93F8Dddc8C6FB",
"0xD1220A0Cf47c7B9BE7a2e6ba89F429762E7B9adB",
]
rskip60_chain_31 = [
"0x5aAeb6053F3e94c9b9A09F33669435E7EF1BEaEd",
"0xFb6916095CA1dF60bb79CE92ce3Ea74C37c5D359",
"0xdbF03B407C01E7cd3cbEa99509D93f8dDDc8C6fB",
"0xd1220a0CF47c7B9Be7A2E6Ba89f429762E7b9adB",
]
n = make_network(chain_id=30)
for s in rskip60_chain_30:
b = unhexlify(s[2:])
h = address_from_bytes(b, n)
self.assertEqual(h, s)
n = make_network(chain_id=31)
for s in rskip60_chain_31:
b = unhexlify(s[2:])
h = address_from_bytes(b, n)
self.assertEqual(h, s)
if __name__ == "__main__":
unittest.main()