arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-21 05:48:23 +00:00
Code Issues Releases Wiki Activity
ff7d5c224b
trezor-firmware/core/tests/test_apps.bitcoin.signtx.py

126 lines
7.6 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
refactor(core): fix imports and use new protobuf API in apps
2021-03-23 12:35:27 +00:00
from trezor.messages import SignTx
from trezor.messages import TxAckInput
from trezor.messages import TxAckInputWrapper
from trezor.messages import TxInput
from trezor.messages import TxAckOutput
from trezor.messages import TxAckOutputWrapper
from trezor.messages import TxOutput
from trezor.messages import TxAckPrevMeta
from trezor.messages import PrevTx
from trezor.messages import TxAckPrevInput
from trezor.messages import TxAckPrevInputWrapper
from trezor.messages import PrevInput
from trezor.messages import TxAckPrevOutput
from trezor.messages import TxAckPrevOutputWrapper
from trezor.messages import PrevOutput
from trezor.messages import TxRequest
from trezor.enums.RequestType import TXINPUT, TXOUTPUT, TXMETA, TXFINISHED
from trezor.messages import TxRequestDetailsType
from trezor.messages import TxRequestSerializedType
from trezor.enums import AmountUnit
from trezor.enums import OutputScriptType
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 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 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)),
feat(core): implement amount_unit for AuthorizeCoinJoin and SignTx
2020-11-23 14:01:25 +00:00
helpers.UiConfirmOutput(out1, coin_bitcoin, AmountUnit.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,
feat(core): implement amount_unit for AuthorizeCoinJoin and SignTx
2020-11-23 14:01:25 +00:00
helpers.UiConfirmTotal(380000 + 10000, 10000, coin_bitcoin, AmountUnit.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)
chore(core/bitcoin): Add DecredApprover.
2021-01-30 00:49:18 +00:00
signer = bitcoin.Bitcoin(tx, keychain, coin_bitcoin, None).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