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

Refactored tests, added common.BitkeyTest

Browse Source
This commit is contained in:
slush 2013-01-14 14:44:31 +01:00
parent 4aafd2deff
commit 9b4359fe01
3 changed files with 37 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tests/common.py Normal file
View File

@ -0,0 +1,24 @@
import unittest
import config
from bitkeylib.client import BitkeyClient
from bitkeylib.debuglink import DebugLink
from bitkeylib import proto
class BitkeyTest(unittest.TestCase):
def setUp(self):
self.debug_transport = config.DEBUG_TRANSPORT(*config.DEBUG_TRANSPORT_ARGS)
self.transport = config.TRANSPORT(*config.TRANSPORT_ARGS)
self.bitkey = BitkeyClient(self.transport, DebugLink(self.debug_transport), algo=proto.ELECTRUM, debug=True)
self.bitkey.setup_debuglink(button=True, pin_correct=True, otp_correct=True)
self.bitkey.load_device(seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice',
otp=True, pin='1234', spv=True)
print "Setup finished"
print "--------------"
def tearDown(self):
self.debug_transport.close()
self.transport.close()

29
tests/test_basic.py
View File

@ -1,8 +1,6 @@
import unittest
import config
import common
from bitkeylib.client import BitkeyClient
from bitkeylib.debuglink import DebugLink
from bitkeylib import proto
'''
@ -14,25 +12,7 @@ from bitkeylib import proto
'''
class TestBasic(unittest.TestCase):
def setUp(self):
self.debug_transport = config.DEBUG_TRANSPORT(*config.DEBUG_TRANSPORT_ARGS)
self.transport = config.TRANSPORT(*config.TRANSPORT_ARGS)
self.bitkey = BitkeyClient(self.transport, DebugLink(self.debug_transport), algo=proto.ELECTRUM, debug=True)
self.bitkey.setup_debuglink(button=True, pin_correct=True, otp_correct=True)
self.bitkey.load_device(seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice',
otp=True, pin='1234', spv=True)
print "Setup finished"
print "--------------"
def tearDown(self):
self.debug_transport.close()
self.transport.close()
class TestBasic(common.BitkeyTest):
def test_features(self):
features = self.bitkey.call(proto.Initialize(session_id=self.bitkey.session_id))
@ -50,7 +30,10 @@ class TestBasic(unittest.TestCase):
uuid2 = self.bitkey.get_uuid()
# UUID must be longer than 10 characters
self.assertGreater(len(uuid1.UUID), 10)
self.assertGreater(len(uuid1), 10)
# Every resulf of UUID must be the same
self.assertEqual(uuid1, uuid2)
if __name__ == '__main__':
unittest.main()

29
tests/test_signtx.py
View File

@ -1,30 +1,11 @@
import unittest
import config
from bitkeylib.client import BitkeyClient
from bitkeylib.debuglink import DebugLink
from bitkeylib import proto
class TestSignTx(unittest.TestCase):
def setUp(self):
self.debug_transport = config.DEBUG_TRANSPORT(*config.DEBUG_TRANSPORT_ARGS)
self.transport = config.TRANSPORT(*config.TRANSPORT_ARGS)
self.bitkey = BitkeyClient(self.transport, DebugLink(self.debug_transport), algo=proto.ELECTRUM)
self.bitkey.setup_debuglink(pin_correct=True, otp_correct=True)
self.bitkey.load_device(seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice',
otp=True, pin='1234', spv=True, button=True)
print "Setup finished"
print "--------------"
def tearDown(self):
self.debug_transport.close()
self.transport.close()
import common
class TestSignTx(common.BitkeyTest):
'''
def test_signtx(self):
print self.bitkey.sign_tx([], [])
'''
if __name__ == '__main__':
unittest.main()