arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-14 11:39:03 +00:00
Code Issues Releases Wiki Activity

All current tests passed

Browse Source
This commit is contained in:
slush0 2013-09-13 05:28:29 +02:00
parent d273cdf86b
commit 4530dc3684
3 changed files with 38 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_basic.py
View File

@ -1,7 +1,7 @@
import unittest import unittest
import common import common
from bitkeylib import proto from trezorlib import proto
''' '''
TODO: TODO:
@ -12,29 +12,29 @@ from bitkeylib import proto
''' '''
class TestBasic(common.BitkeyTest): class TestBasic(common.TrezorTest):
def test_features(self): def test_features(self):
features = self.bitkey.call(proto.Initialize()) features = self.client.call(proto.Initialize())
# Result is the same as reported by BitkeyClient class # Result is the same as reported by BitkeyClient class
self.assertEqual(features, self.bitkey.features) self.assertEqual(features, self.client.features)
def test_ping(self): def test_ping(self):
ping = self.bitkey.call(proto.Ping(message='ahoj!')) ping = self.client.call(proto.Ping(message='ahoj!'))
# Ping results in Success(message='Ahoj!') # Ping results in Success(message='Ahoj!')
self.assertEqual(ping, proto.Success(message='ahoj!')) self.assertEqual(ping, proto.Success(message='ahoj!'))
def test_uuid(self): def test_uuid(self):
uuid1 = self.bitkey.get_uuid() uuid1 = self.client.get_serial_number()
self.bitkey.init_device() self.client.init_device()
uuid2 = self.bitkey.get_uuid() uuid2 = self.client.get_serial_number()
# UUID must be longer than 10 characters # UUID must be longer than 10 characters
self.assertEqual(len(uuid1), 9) self.assertEqual(len(uuid1), 12)
# Every resulf of UUID must be the same # Every resulf of UUID must be the same
self.assertEqual(uuid1, uuid2) self.assertEqual(uuid1, uuid2)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

24
tests/test_protect_call.py
View File

@ -1,35 +1,31 @@
import unittest import unittest
import common import common
from bitkeylib.client import CallException, PinException from trezorlib.client import CallException, PinException
from bitkeylib import proto from trezorlib import proto
class TestProtectCall(common.BitkeyTest): class TestProtectCall(common.TrezorTest):
def _some_protected_call(self): def _some_protected_call(self):
# This method perform any call which have protection in the device # This method perform any call which have protection in the device
entropy_len = 10 entropy_len = 10
entropy = self.bitkey.get_entropy(entropy_len) entropy = self.client.get_entropy(entropy_len)
self.assertEqual(len(entropy), entropy_len) self.assertEqual(len(entropy), entropy_len)
def test_no_protection(self): def test_no_protection(self):
self.bitkey.load_device( self.client.load_device(seed=self.mnemonic1, pin='')
seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice',
pin='')
self.assertEqual(self.bitkey.debuglink.read_pin()[0], '') self.assertEqual(self.client.debuglink.read_pin()[0], '')
self._some_protected_call() self._some_protected_call()
def test_pin(self): def test_pin(self):
self.bitkey.load_device( self.client.load_device(seed=self.mnemonic1, pin=self.pin2)
seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice',
pin='2345')
self.assertEqual(self.bitkey.debuglink.read_pin()[0], '2345') self.assertEqual(self.client.debuglink.read_pin()[0], self.pin2)
self._some_protected_call() self._some_protected_call()
def test_incorrect_pin(self): def test_incorrect_pin(self):
self.bitkey.setup_debuglink(button=True, pin_correct=False) self.client.setup_debuglink(button=True, pin_correct=False)
self.assertRaises(PinException, self._some_protected_call) self.assertRaises(PinException, self._some_protected_call)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

36
tests/test_signtx.py
View File

@ -2,7 +2,7 @@ import unittest
import common import common
import binascii import binascii
import bitkeylib.trezor_pb2 as proto import trezorlib.trezor_pb2 as proto
''' '''
./electrum -w ~/.electrum-bitkey mktx 1FQVPnjrbkPWeA8poUoEnX9U3n9DyhAVtv 0.001 ./electrum -w ~/.electrum-bitkey mktx 1FQVPnjrbkPWeA8poUoEnX9U3n9DyhAVtv 0.001
@ -16,7 +16,7 @@ a62964699995f6e018cfbeb7a71a66d4c64fa38875d79ead0a9ac66f59c1c8b3a3ffffffff
0250c30000000000001976a91444ce5c6789b0bb0e8a9ab9a4769fe181cb274c4688aca086 0250c30000000000001976a91444ce5c6789b0bb0e8a9ab9a4769fe181cb274c4688aca086
0100000000001976a9149e03078026388661b197129a43f0f64f88379ce688ac00000000 0100000000001976a9149e03078026388661b197129a43f0f64f88379ce688ac00000000
''' '''
class TestSignTx(common.BitkeyTest): class TestSignTx(common.TrezorTest):
def test_signtx(self): def test_signtx(self):
expected_tx = '01000000012de70f7d6ffed0db70f8882f3fca90db9bb09f0e99bce27468c23d3c994fcd56' \ expected_tx = '01000000012de70f7d6ffed0db70f8882f3fca90db9bb09f0e99bce27468c23d3c994fcd56' \
'010000008b4830450221009b985e14d53cfeed3496846db6ddaa77a0206138d0df4c2ccd3b' \ '010000008b4830450221009b985e14d53cfeed3496846db6ddaa77a0206138d0df4c2ccd3b' \
@ -27,7 +27,7 @@ class TestSignTx(common.BitkeyTest):
'0100000000001976a9149e03078026388661b197129a43f0f64f88379ce688ac00000000' '0100000000001976a9149e03078026388661b197129a43f0f64f88379ce688ac00000000'
inp1 = proto.TxInput(index=0, inp1 = proto.TxInput(index=0,
address_n=[1,0], address_n=[1, 0],
amount=200000, # 0.002 BTC amount=200000, # 0.002 BTC
prev_hash=binascii.unhexlify('56cd4f993c3dc26874e2bc990e9fb09bdb90ca3f2f88f870dbd0fe6f7d0fe72d'), prev_hash=binascii.unhexlify('56cd4f993c3dc26874e2bc990e9fb09bdb90ca3f2f88f870dbd0fe6f7d0fe72d'),
prev_index=1, prev_index=1,
@ -35,7 +35,7 @@ class TestSignTx(common.BitkeyTest):
) )
out1 = proto.TxOutput(index=0, out1 = proto.TxOutput(index=0,
address='17GpAFnkHRjWKePkX4kxHaHy49V8EHTr7i', address='1GnnT11aZeH6QZCtT7EjCvRF3EXHoY3owE',
address_n=[0, 1], address_n=[0, 1],
amount=50000, # 0.0005 BTC amount=50000, # 0.0005 BTC
script_type=proto.PAYTOADDRESS, script_type=proto.PAYTOADDRESS,
@ -50,7 +50,7 @@ class TestSignTx(common.BitkeyTest):
#script_args= #script_args=
) )
print binascii.hexlify(self.bitkey.sign_tx([inp1], [out1, out2])[1]) print binascii.hexlify(self.client.sign_tx([inp1], [out1, out2])[1])
''' '''
def test_workflow(self): def test_workflow(self):
@ -87,11 +87,11 @@ class TestSignTx(common.BitkeyTest):
# Prepare and send initial message # Prepare and send initial message
tx = proto.SignTx() tx = proto.SignTx()
tx.algo = proto.ELECTRUM tx.algo = proto.ELECTRUM
tx.random = self.bitkey._get_local_entropy() tx.random = self.client._get_local_entropy()
tx.inputs_count = 2 tx.inputs_count = 2
tx.outputs_count = 2 tx.outputs_count = 2
res = self.bitkey.call(tx) res = self.client.call(tx)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXINPUT) self.assertEqual(res.request_type, proto.TXINPUT)
self.assertEqual(res.request_index, 0) self.assertEqual(res.request_index, 0)
@ -99,28 +99,28 @@ class TestSignTx(common.BitkeyTest):
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
# FIRST SIGNATURE # FIRST SIGNATURE
res = self.bitkey.call(inp1) res = self.client.call(inp1)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXINPUT) self.assertEqual(res.request_type, proto.TXINPUT)
self.assertEqual(res.request_index, 1) self.assertEqual(res.request_index, 1)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(inp2) res = self.client.call(inp2)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 0) self.assertEqual(res.request_index, 0)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(out1) res = self.client.call(out1)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 1) self.assertEqual(res.request_index, 1)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(out2) res = self.client.call(out2)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXINPUT) self.assertEqual(res.request_type, proto.TXINPUT)
self.assertEqual(res.request_index, 0) self.assertEqual(res.request_index, 0)
@ -129,28 +129,28 @@ class TestSignTx(common.BitkeyTest):
serialized += res.serialized_tx serialized += res.serialized_tx
# SECOND SIGNATURE # SECOND SIGNATURE
res = self.bitkey.call(inp1) res = self.client.call(inp1)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXINPUT) self.assertEqual(res.request_type, proto.TXINPUT)
self.assertEqual(res.request_index, 1) self.assertEqual(res.request_index, 1)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(inp2) res = self.client.call(inp2)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 0) self.assertEqual(res.request_index, 0)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(out1) res = self.client.call(out1)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 1) self.assertEqual(res.request_index, 1)
self.assertEqual(res.signature, '') self.assertEqual(res.signature, '')
self.assertEqual(res.serialized_tx, '') self.assertEqual(res.serialized_tx, '')
res = self.bitkey.call(out2) res = self.client.call(out2)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 0) self.assertEqual(res.request_index, 0)
@ -159,7 +159,7 @@ class TestSignTx(common.BitkeyTest):
serialized += res.serialized_tx serialized += res.serialized_tx
# FINALIZING OUTPUTS # FINALIZING OUTPUTS
res = self.bitkey.call(out1) res = self.client.call(out1)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, 1) self.assertEqual(res.request_index, 1)
@ -167,7 +167,7 @@ class TestSignTx(common.BitkeyTest):
self.assertNotEqual(res.serialized_tx, '') self.assertNotEqual(res.serialized_tx, '')
serialized += res.serialized_tx serialized += res.serialized_tx
res = self.bitkey.call(out2) res = self.client.call(out2)
self.assertIsInstance(res, proto.TxRequest) self.assertIsInstance(res, proto.TxRequest)
self.assertEqual(res.request_type, proto.TXOUTPUT) self.assertEqual(res.request_type, proto.TXOUTPUT)
self.assertEqual(res.request_index, -1) self.assertEqual(res.request_index, -1)
@ -179,4 +179,4 @@ class TestSignTx(common.BitkeyTest):
''' '''
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()