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

Adde debug_processor to sign_tx() for unittest purposes

Browse Source
This commit is contained in:
slush0 2014-04-18 18:56:12 +02:00
parent de1bef9533
commit 54bc7239ce
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
trezorlib/client.py
View File

@ -455,7 +455,7 @@ class ProtocolMixin(object):
return txes return txes
def sign_tx(self, coin_name, inputs, outputs): def sign_tx(self, coin_name, inputs, outputs, debug_processor=None):
start = time.time() start = time.time()
txes = self._prepare_sign_tx(coin_name, inputs, outputs) txes = self._prepare_sign_tx(coin_name, inputs, outputs)
@ -525,6 +525,13 @@ class ProtocolMixin(object):
msg.bin_outputs.extend([current_tx.bin_outputs[res.details.request_index], ]) msg.bin_outputs.extend([current_tx.bin_outputs[res.details.request_index], ])
else: else:
msg.outputs.extend([current_tx.outputs[res.details.request_index], ]) msg.outputs.extend([current_tx.outputs[res.details.request_index], ])
if debug_processor != None:
# If debug_processor function is provided,
# pass thru it the request and prepared response.
# This is useful for unit tests, see test_msg_signtx
msg = debug_processor(res, msg)
res = self.call(proto.TxAck(tx=msg)) res = self.call(proto.TxAck(tx=msg))
continue continue