BTCPrivate
/
python-trezor
mirror of https://github.com/BTCPrivate/python-trezor.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

tests: deepcopy is required when debug_processor is invoked

This commit is contained in:
Tomas Susanka 2018-02-23 16:03:43 +01:00
parent cbd3751bdb
commit 617ccc21d9
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
trezorlib/client.py
View File

@ -937,9 +937,13 @@ class ProtocolMixin(object):
msg = proto.TransactionType()
msg._extend_inputs([current_tx.inputs[res.details.request_index], ])
if debug_processor is not None:
# msg needs to be deep copied so when it's modified
# the other messages stay intact
from copy import deepcopy
msg = deepcopy(msg)
# If debug_processor function is provided,
# pass thru it the request and prepared response.
# This is useful for unit tests, see test_msg_signtx
# This is useful for tests, see test_msg_signtx
msg = debug_processor(res, msg)
res = self.call(proto.TxAck(tx=msg))
@ -953,9 +957,13 @@ class ProtocolMixin(object):
msg._extend_outputs([current_tx.outputs[res.details.request_index], ])
if debug_processor is not None:
# msg needs to be deep copied so when it's modified
# the other messages stay intact
from copy import deepcopy
msg = deepcopy(msg)
# If debug_processor function is provided,
# pass thru it the request and prepared response.
# This is useful for unit tests, see test_msg_signtx
# This is useful for tests, see test_msg_signtx
msg = debug_processor(res, msg)
res = self.call(proto.TxAck(tx=msg))

1
trezorlib/tests/device_tests/test_msg_signtx.py
View File

@ -652,7 +652,6 @@ class TestMsgSigntx(TrezorTest):
run_attack = True
def attack_processor(req, msg):
import sys
global run_attack
if req.details.tx_hash is not None: