trezor-core/tests/test_apps.ethereum.layout.py

68 lines
3.6 KiB
Python

from common import *
from apps.ethereum.layout import format_ethereum_amount
class TestEthereumLayout(unittest.TestCase):
def test_format(self):
text = format_ethereum_amount((1).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1 Wei ETH')
text = format_ethereum_amount((1000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1000 Wei ETH')
text = format_ethereum_amount((1000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1000000 Wei ETH')
text = format_ethereum_amount((10000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '10000000 Wei ETH')
text = format_ethereum_amount((100000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '100000000 Wei ETH')
text = format_ethereum_amount((1000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1000000000 Wei ETH')
text = format_ethereum_amount((10000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.00000001 ETH')
text = format_ethereum_amount((100000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.0000001 ETH')
text = format_ethereum_amount((1000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.000001 ETH')
text = format_ethereum_amount((10000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.00001 ETH')
text = format_ethereum_amount((100000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.0001 ETH')
text = format_ethereum_amount((1000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.001 ETH')
text = format_ethereum_amount((10000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.01 ETH')
text = format_ethereum_amount((100000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '0.1 ETH')
text = format_ethereum_amount((1000000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1 ETH')
text = format_ethereum_amount((10000000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '10 ETH')
text = format_ethereum_amount((100000000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '100 ETH')
text = format_ethereum_amount((1000000000000000000000).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1000 ETH')
text = format_ethereum_amount((1000000000000000000).to_bytes(32, 'big'), None, 61)
self.assertEqual(text, '1 ETC')
text = format_ethereum_amount((1000000000000000000).to_bytes(32, 'big'), None, 31)
self.assertEqual(text, '1 tRSK')
text = format_ethereum_amount((1000000000000000001).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '1.000000000000000001 ETH')
text = format_ethereum_amount((10000000000000000001).to_bytes(32, 'big'), None, 1)
self.assertEqual(text, '10.000000000000000001 ETH')
text = format_ethereum_amount((10000000000000000001).to_bytes(32, 'big'), None, 61)
self.assertEqual(text, '10.000000000000000001 ETC')
text = format_ethereum_amount((1000000000000000001).to_bytes(32, 'big'), None, 31)
self.assertEqual(text, '1.000000000000000001 tRSK')
# unknown chain
text = format_ethereum_amount((1).to_bytes(32, 'big'), None, 9999)
self.assertEqual(text, '1 Wei UNKN')
text = format_ethereum_amount((10000000000000000001).to_bytes(32, 'big'), None, 9999)
self.assertEqual(text, '10.000000000000000001 UNKN')
if __name__ == '__main__':
unittest.main()