arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-01-13 17:00:59 +00:00
Code Issues Releases Wiki Activity

tests: add test for apps.common.coins

Browse Source
This commit is contained in:
Pavol Rusnak 2016-11-15 11:57:18 +01:00
parent f7fe5344b8
commit b09f0eaf4e
No known key found for this signature in database
GPG Key ID: 91F3B339B9A02A3D
3 changed files with 39 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/apps/common/coins.py
View File

@ -90,18 +90,18 @@ def by_shortcut(shortcut):
for c in COINS:
if c.coin_shortcut == shortcut:
return c
raise Exception('Unknown coin shortcut "%s"' % shortcut)
raise ValueError('Unknown coin shortcut "%s"' % shortcut)
def by_name(name):
for c in COINS:
if c.coin_name == name:
return c
raise Exception('Unknown coin name "%s"' % name)
raise ValueError('Unknown coin name "%s"' % name)
def by_address_type(version):
for c in COINS:
if c.address_type == version:
return c
raise Exception('Unknown coin address type %d' % version)
raise ValueError('Unknown coin address type %d' % version)

36
tests/test_apps.common.coins.py Normal file
View File

@ -0,0 +1,36 @@
from common import *
from apps.common import coins
class TestCoins(unittest.TestCase):
def test_coins(self):
ref = [
('BTC', 'Bitcoin', 0),
('TEST', 'Testnet', 111),
('NMC', 'Namecoin', 52),
('LTC', 'Litecoin', 48),
('DOGE', 'Dogecoin', 30),
('DASH', 'Dash', 76),
('ZEC', 'Zcash', 7352),
('TAZ', 'Zcash Testnet', 7461),
]
for s, n, a in ref:
c1 = coins.by_shortcut(s)
c2 = coins.by_name(n)
c3 = coins.by_address_type(a)
self.assertEqual(c1, c2)
self.assertEqual(c1, c3)
self.assertEqual(c2, c3)
def test_failure(self):
with self.assertRaises(ValueError):
coins.by_shortcut('XXX')
with self.assertRaises(ValueError):
coins.by_name('XXXXX')
with self.assertRaises(ValueError):
coins.by_address_type(1234)
if __name__ == '__main__':
unittest.main()

1
tests/test_apps.common.signtx.py
View File

@ -17,7 +17,6 @@ from trezor.messages import OutputScriptType
from apps.common import coins
from apps.common import signtx
class TestSignTx(unittest.TestCase):
# pylint: disable=C0301