arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-18 13:38:12 +00:00
Code Issues Releases Wiki Activity
e23e35332d
trezor-firmware/core/tests/test_apps.bitcoin.signtx.py

127 lines
7.9 KiB
Python
Raw Normal View History

wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
from common import *
from trezor.utils import chunks
from trezor.crypto import bip32, bip39
from trezor.messages.SignTx import SignTx
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
from trezor.messages.TxAckInput import TxAckInput
from trezor.messages.TxAckInputWrapper import TxAckInputWrapper
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
from trezor.messages.TxInput import TxInput
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
from trezor.messages.TxAckOutput import TxAckOutput
from trezor.messages.TxAckOutputWrapper import TxAckOutputWrapper
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
from trezor.messages.TxOutput import TxOutput
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
from trezor.messages.TxAckPrevMeta import TxAckPrevMeta
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
from trezor.messages.PrevTx import PrevTx
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
from trezor.messages.TxAckPrevInput import TxAckPrevInput
from trezor.messages.TxAckPrevInputWrapper import TxAckPrevInputWrapper
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
from trezor.messages.PrevInput import PrevInput
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
from trezor.messages.TxAckPrevOutput import TxAckPrevOutput
from trezor.messages.TxAckPrevOutputWrapper import TxAckPrevOutputWrapper
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
from trezor.messages.PrevOutput import PrevOutput
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
from trezor.messages.TxRequest import TxRequest
from trezor.messages.RequestType import TXINPUT, TXOUTPUT, TXMETA, TXFINISHED
from trezor.messages.TxRequestDetailsType import TxRequestDetailsType
from trezor.messages.TxRequestSerializedType import TxRequestSerializedType
from trezor.messages import OutputScriptType
from apps.common import coins
core/tests: fix bitcoin unit tests
2020-07-23 14:20:54 +00:00
from apps.common.keychain import Keychain
chore(core/tests): update unit tests for new APIs, drop unnecessary ones
2020-09-23 10:31:07 +00:00
from apps.bitcoin.keychain import get_schemas_for_coin
core/bitcoin: finalize bitcoin refactor - core/bitcoin: move common files to the app's root - core/bitcoin: use require_confirm instead of confirm - core: move bitcoin unrelated functions from 'bitcoin' to a new 'misc' app - core/bitcoin: use relative imports inside the app - core: rename wallet app to bitcoin - core/wallet: replace SigningErrors and the other exception classes with wire.Errors
2020-05-15 18:35:09 +00:00
from apps.bitcoin.sign_tx import bitcoin, helpers
core/tests: Fix unit tests to work with BasicApprover in sign_tx.
2020-07-17 15:39:43 +00:00
from apps.bitcoin.sign_tx.approvers import BasicApprover
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
EMPTY_SERIALIZED = TxRequestSerializedType(serialized_tx=bytearray())
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
tests: cleanup flake8 warnings
2017-12-16 01:11:26 +00:00
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
class TestSignTx(unittest.TestCase):
# pylint: disable=C0301
def test_one_one_fee(self):
# tx: d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882
# input 0: 0.0039 BTC
coin_bitcoin = coins.by_name('Bitcoin')
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
ptx1 = PrevTx(version=1, lock_time=0, inputs_count=2, outputs_count=1, extra_data_len=0)
pinp1 = PrevInput(script_sig=unhexlify('483045022072ba61305fe7cb542d142b8f3299a7b10f9ea61f6ffaab5dca8142601869d53c0221009a8027ed79eb3b9bc13577ac2853269323434558528c6b6a7e542be46e7e9a820141047a2d177c0f3626fc68c53610b0270fa6156181f46586c679ba6a88b34c6f4874686390b4d92e5769fbb89c8050b984f4ec0b257a0e5c4ff8bd3b035a51709503'),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
prev_hash=unhexlify('c16a03f1cf8f99f6b5297ab614586cacec784c2d259af245909dedb0e39eddcf'),
prev_index=1,
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
sequence=0xffff_ffff)
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
pinp2 = PrevInput(script_sig=unhexlify('48304502200fd63adc8f6cb34359dc6cca9e5458d7ea50376cbd0a74514880735e6d1b8a4c0221008b6ead7fe5fbdab7319d6dfede3a0bc8e2a7c5b5a9301636d1de4aa31a3ee9b101410486ad608470d796236b003635718dfc07c0cac0cfc3bfc3079e4f491b0426f0676e6643a39198e8e7bdaffb94f4b49ea21baa107ec2e237368872836073668214'),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
prev_hash=unhexlify('1ae39a2f8d59670c8fc61179148a8e61e039d0d9e8ab08610cb69b4a19453eaf'),
prev_index=1,
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
sequence=0xffff_ffff)
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
pout1 = PrevOutput(script_pubkey=unhexlify('76a91424a56db43cf6f2b02e838ea493f95d8d6047423188ac'),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
amount=200000 + 200000 - 10000)
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
inp1 = TxInput(address_n=[44 | 0x80000000, 0 | 0x80000000, 0 | 0x80000000, 0, 0],
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
# amount=390000,
prev_hash=unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882'),
prev_index=0,
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
amount=390000,
wallet: tx weight calculator is used to determine fee
2017-11-23 17:40:27 +00:00
multisig=None,
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
sequence=0xffff_ffff)
chore: make tx type names shorter as suggested by @andrewkozlik: TxAckInputType -> TxInput TxAckOutputType -> TxOutput TxAckPrevTxType -> PrevTx TxAckPrevInputType -> PrevInput TxAckPrevOutputType -> PrevOutput
2020-09-23 11:08:26 +00:00
out1 = TxOutput(address='1MJ2tj2ThBE62zXbBYA5ZaN3fdve5CPAz1',
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
amount=390000 - 10000,
script_type=OutputScriptType.PAYTOADDRESS,
tests: multisig field set to None
2018-02-12 11:40:18 +00:00
multisig=None)
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
tx = SignTx(coin_name=None, version=1, lock_time=0, inputs_count=1, outputs_count=1)
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
messages = [
None,
coins: validate derivation paths Based on SLIP-44 ids and other checks. See docs/coins/README for info.
2018-08-30 11:45:09 +00:00
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
TxRequest(request_type=TXINPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
TxRequest(request_type=TXOUTPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
wallet: remove "import *" from sign_tx
2018-11-06 14:45:02 +00:00
helpers.UiConfirmOutput(out1, coin_bitcoin),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
True,
wallet: remove "import *" from sign_tx
2018-11-06 14:45:02 +00:00
helpers.UiConfirmTotal(380000 + 10000, 10000, coin_bitcoin),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
True,
# ButtonRequest(code=ButtonRequest_ConfirmOutput),
# ButtonRequest(code=ButtonRequest_SignTx),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
TxRequest(request_type=TXINPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
TxRequest(request_type=TXMETA, details=TxRequestDetailsType(request_index=None, tx_hash=unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882')), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckPrevMeta(tx=ptx1),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
TxRequest(request_type=TXINPUT, details=TxRequestDetailsType(request_index=0, tx_hash=unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882')), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckPrevInput(tx=TxAckPrevInputWrapper(input=pinp1)),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
TxRequest(request_type=TXINPUT, details=TxRequestDetailsType(request_index=1, tx_hash=unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882')), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckPrevInput(tx=TxAckPrevInputWrapper(input=pinp2)),
core/tests: Fix unit tests for streamimg prev_tx after confirmation.
2020-08-22 15:28:29 +00:00
TxRequest(request_type=TXOUTPUT, details=TxRequestDetailsType(request_index=0, tx_hash=unhexlify('d5f65ee80147b4bcc70b75e4bbf2d7382021b871bd8867ef8fa525ef50864882')), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckPrevOutput(tx=TxAckPrevOutputWrapper(output=pout1)),
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
TxRequest(request_type=TXINPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=TxRequestSerializedType(serialized_tx=unhexlify('0100000001'))),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckInput(tx=TxAckInputWrapper(input=inp1)),
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
TxRequest(request_type=TXOUTPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=EMPTY_SERIALIZED),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
TxRequest(request_type=TXOUTPUT, details=TxRequestDetailsType(request_index=0, tx_hash=None), serialized=TxRequestSerializedType(
signature_index=0,
core/tests: update unit tests for new keychain API
2020-05-14 13:55:21 +00:00
signature=unhexlify('30440220198146fa987da8d78c4c7a471614fceb54d161ede244412f3369f436a7aec386022066bbede7644baa38abbdb4b1f3037f8db225c04e107099b625339a55614c3db3'),
serialized_tx=unhexlify('82488650ef25a58fef6788bd71b8212038d7f2bbe4750bc7bcb44701e85ef6d5000000006a4730440220198146fa987da8d78c4c7a471614fceb54d161ede244412f3369f436a7aec386022066bbede7644baa38abbdb4b1f3037f8db225c04e107099b625339a55614c3db30121027a4cebff51c97c047637cda66838e8b64421a4af6bf8ef3c99717f92d09b3c1dffffffff01'))),
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
TxAckOutput(tx=TxAckOutputWrapper(output=out1)),
core/sign_tx: Updates based on code review.
2020-04-22 19:26:04 +00:00
TxRequest(request_type=TXFINISHED, details=TxRequestDetailsType(), serialized=TxRequestSerializedType(
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
signature_index=None,
signature=None,
core/tests: Fix unit tests after sign_tx refactor.
2020-04-16 12:31:15 +00:00
serialized_tx=unhexlify('60cc0500000000001976a914de9b2a8da088824e8fe51debea566617d851537888ac00000000'),
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
)),
]
seed = bip39.seed('alcohol woman abuse must during monitor noble actual mixed trade anger aisle', '')
chore(core/tests): update unit tests for new APIs, drop unnecessary ones
2020-09-23 10:31:07 +00:00
ns = get_schemas_for_coin(coin_bitcoin)
core/tests: fix bitcoin unit tests
2020-07-23 14:20:54 +00:00
keychain = Keychain(seed, coin_bitcoin.curve_name, ns)
core/tests: Fix unit tests to work with BasicApprover in sign_tx.
2020-07-17 15:39:43 +00:00
approver = BasicApprover(tx, coin_bitcoin)
signer = bitcoin.Bitcoin(tx, keychain, coin_bitcoin, approver).signer()
src/apps/wallet/sign_tx: fix bip115 code in last commit
2018-07-02 12:29:58 +00:00
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
for request, response in chunks(messages, 2):
src/apps/wallet/sign_tx: fix bip115 code in last commit
2018-07-02 12:29:58 +00:00
res = signer.send(request)
chore(core): update unit tests to kwargs usage and new TxAck subtypes
2020-09-16 12:51:19 +00:00
if isinstance(res, tuple):
_, res = res
wallet: use keychain API in signing, fix tests
2018-12-04 13:27:06 +00:00
self.assertEqual(res, response)
src/apps/wallet/sign_tx: fix bip115 code in last commit
2018-07-02 12:29:58 +00:00
wire: refactoring - prefer importing modules instead of module members - session_id is always first argument - prefer much shorter names, don't expect users to import module members - shuffle around session-specific code - reduce allocations
2016-12-08 15:14:47 +00:00
with self.assertRaises(StopIteration):
signer.send(None)
if __name__ == '__main__':
unittest.main()
Reference in New Issue Copy Permalink