1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-15 19:08:07 +00:00
trezor-firmware/tests/test_apps.ethereum.layout.py

40 lines
1.8 KiB
Python
Raw Normal View History

2017-12-27 11:48:05 +00:00
from common import *
2018-02-09 12:36:08 +00:00
from apps.ethereum.layout import format_ethereum_amount
2017-12-27 11:48:05 +00:00
class TestEthereumLayout(unittest.TestCase):
def test_format(self):
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1).to_bytes(5, 'big'), None, 1)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '1 Wei')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1000).to_bytes(5, 'big'), None, 1)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '1000 Wei')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1000000000000000000).to_bytes(20, 'big'), None, 1)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '1 ETH')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((10000000000000000000).to_bytes(20, 'big'), None, 1)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '10 ETH')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((10000000000000000000).to_bytes(20, 'big'), None, 61)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '10 ETC')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1000000000000000000).to_bytes(20, 'big'), None, 31)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '1 tRSK')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1000000000000000001).to_bytes(20, 'big'), None, 1)
self.assertEqual(text, '1.000000000000000001 ETH')
text = format_ethereum_amount((10000000000000000001).to_bytes(20, 'big'), None, 1)
self.assertEqual(text, '10.000000000000000001 ETH')
text = format_ethereum_amount((10000000000000000001).to_bytes(20, 'big'), None, 61)
self.assertEqual(text, '10.000000000000000001 ETC')
text = format_ethereum_amount((1000000000000000001).to_bytes(20, 'big'), None, 31)
self.assertEqual(text, '1.000000000000000001 tRSK')
2017-12-27 11:48:05 +00:00
# unknown chain
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((1).to_bytes(20, 'big'), None, 9999)
2017-12-27 11:48:05 +00:00
self.assertEqual(text, '1 Wei')
2018-02-09 12:36:08 +00:00
text = format_ethereum_amount((10000000000000000001).to_bytes(20, 'big'), None, 9999)
self.assertEqual(text, '10.000000000000000001 UNKN')
2017-12-27 11:48:05 +00:00
if __name__ == '__main__':
unittest.main()