fix two more unit tests

This commit is contained in:
Pavol Rusnak 2014-04-11 15:17:36 +02:00
parent 2964cc4f9c
commit 7ae8894214
2 changed files with 4 additions and 4 deletions

View File

@ -150,7 +150,7 @@ class TestProtectionLevels(common.TrezorTest):
proto.PassphraseRequest(),
proto.ButtonRequest(code=proto_types.ButtonRequest_ConfirmOutput),
proto.ButtonRequest(code=proto_types.ButtonRequest_SignTx),
proto.TxRequest(finished=True)])
proto.TxRequest(request_type=proto_types.TXFINISHED)])
self.client.simple_sign_tx('Bitcoin', [inp1, ], [out1, ])
# def test_firmware_erase(self):

View File

@ -56,8 +56,8 @@ class TestZeroSig(common.TrezorTest):
)
msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
tx = self.client.call(msg)
siglen = ord(tx.serialized_tx[44])
tx = self.client.call(msg)
siglen = ord(tx.serialized.serialized_tx[44])
# Trezor must strip leading zero from signature
self.assertEqual(siglen, 67)
@ -79,7 +79,7 @@ class TestZeroSig(common.TrezorTest):
msg = self.client._prepare_simple_sign_tx('Bitcoin', [inp1], [out1])
tx = self.client.call(msg)
siglen = ord(tx.serialized_tx[44])
siglen = ord(tx.serialized.serialized_tx[44])
# Trezor must strip leading zero from signature
self.assertEqual(siglen, 66)