trezor-firmware/core/tests/test_apps.ethereum.tokens.py

from common import *

if not utils.BITCOIN_ONLY:
    from apps.ethereum import tokens


@unittest.skipUnless(not utils.BITCOIN_ONLY, "altcoin")
class TestEthereumTokens(unittest.TestCase):
    def test_token_by_chain_address(self):

        token = tokens.token_by_chain_address(1, b"\x7f\xc6\x65\x00\xc8\x4a\x76\xad\x7e\x9c\x93\x43\x7b\xfc\x5a\xc3\x3e\x2d\xda\xe9")
        self.assertEqual(token.symbol, 'AAVE')

        # invalid adress, invalid chain
        token = tokens.token_by_chain_address(999, b'\x00\xFF')
        self.assertIs(token, None)

        self.assertEqual(tokens.UNKNOWN_TOKEN.symbol, 'Wei UNKN')
        self.assertEqual(tokens.UNKNOWN_TOKEN.decimals, 0)


if __name__ == '__main__':
    unittest.main()