mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-22 07:28:10 +00:00
feat(tests): Add tests for spending Taproot UTXOs.
This commit is contained in:
Andrew Kozlik
Andrew Kozlik
parent
554888ef84
commit
f0d6e961a9
279
tests/device_tests/test_msg_signtx_taproot.py
Normal file
279
tests/device_tests/test_msg_signtx_taproot.py
Normal file
@ -0,0 +1,279 @@
|
||||
# This file is part of the Trezor project.
|
||||
#
|
||||
# Copyright (C) 2012-2021 SatoshiLabs and contributors
|
||||
#
|
||||
# This library is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Lesser General Public License version 3
|
||||
# as published by the Free Software Foundation.
|
||||
#
|
||||
# This library is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU Lesser General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the License along with this library.
|
||||
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
|
||||
|
||||
import pytest
|
||||
|
||||
from trezorlib import btc, messages
|
||||
from trezorlib.tools import parse_path
|
||||
|
||||
from ..tx_cache import TxCache
|
||||
from .signtx import request_finished, request_input, request_meta, request_output
|
||||
|
||||
B = messages.ButtonRequestType
|
||||
TX_API = TxCache("Testnet")
|
||||
|
||||
TXHASH_7956f1 = bytes.fromhex(
|
||||
"7956f1de3e7362b04115b64a31f0b6822c50dd6c08d78398f392a0ac3f0e357b"
|
||||
)
|
||||
TXHASH_3ac32e = bytes.fromhex(
|
||||
"3ac32e90831d79385eee49d6030a2123cd9d009fe8ffc3d470af9a6a777a119b"
|
||||
)
|
||||
TXHASH_df862e = bytes.fromhex(
|
||||
"df862e31da31ff84addd392f6aa89af18978a398ea258e4901ae72894b66679f"
|
||||
)
|
||||
TXHASH_8c3ea7 = bytes.fromhex(
|
||||
"8c3ea7a10ab6d289119b722ec8c27b70c17c722334ced31a0370d782e4b6775d"
|
||||
)
|
||||
TXHASH_901593 = bytes.fromhex(
|
||||
"901593bed347678d9762fdee728c35dc4ec3cfdc3728a4d72dcaab3751122e85"
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.skip_t1
|
||||
class TestMsgSigntxTaproot:
|
||||
def test_send_p2tr(self, client):
|
||||
inp1 = messages.TxInputType(
|
||||
# tb1pn2d0yjeedavnkd8z8lhm566p0f2utm3lgvxrsdehnl94y34txmts5s7t4c
|
||||
address_n=parse_path("86'/1'/0'/1/0"),
|
||||
amount=4600,
|
||||
prev_hash=TXHASH_7956f1,
|
||||
prev_index=1,
|
||||
script_type=messages.InputScriptType.SPENDTAPROOT,
|
||||
)
|
||||
out1 = messages.TxOutputType(
|
||||
# 86'/1'/1'/0/0
|
||||
address="tb1paxhjl357yzctuf3fe58fcdx6nul026hhh6kyldpfsf3tckj9a3wslqd7zd",
|
||||
amount=4450,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
with client:
|
||||
client.set_expected_responses(
|
||||
[
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_output(0),
|
||||
request_input(0),
|
||||
request_finished(),
|
||||
]
|
||||
)
|
||||
_, serialized_tx = btc.sign_tx(
|
||||
client, "Testnet", [inp1], [out1], prev_txes=TX_API
|
||||
)
|
||||
|
||||
assert (
|
||||
serialized_tx.hex()
|
||||
== "010000000001017b350e3faca092f39883d7086cdd502c82b6f0314ab61541b062733edef156790100000000ffffffff016211000000000000225120e9af2fc69e20b0be2629cd0e9c34da9f3ef56af7beac4fb4298262bc5a45ec5d0140493145b992dacbd7ea579a415efc2cba20c3bf0f7827d1bcf999109c0d11783fe96f91ddb04a889faa17ad21ecc5c81a578009744e95c7e721aff2a5c442916600000000"
|
||||
)
|
||||
|
||||
def test_send_two_with_change(self, client):
|
||||
inp1 = messages.TxInputType(
|
||||
# tb1pswrqtykue8r89t9u4rprjs0gt4qzkdfuursfnvqaa3f2yql07zmq8s8a5u
|
||||
address_n=parse_path("86'/1'/0'/0/0"),
|
||||
amount=6800,
|
||||
prev_hash=TXHASH_df862e,
|
||||
prev_index=0,
|
||||
script_type=messages.InputScriptType.SPENDTAPROOT,
|
||||
)
|
||||
inp2 = messages.TxInputType(
|
||||
# tb1p8tvmvsvhsee73rhym86wt435qrqm92psfsyhy6a3n5gw455znnpqm8wald
|
||||
address_n=parse_path("86'/1'/0'/0/1"),
|
||||
amount=13000,
|
||||
prev_hash=TXHASH_3ac32e,
|
||||
prev_index=1,
|
||||
script_type=messages.InputScriptType.SPENDTAPROOT,
|
||||
)
|
||||
out1 = messages.TxOutputType(
|
||||
# 84'/1'/1'/0/0
|
||||
address="tb1q7r9yvcdgcl6wmtta58yxf29a8kc96jkyxl7y88",
|
||||
amount=15000,
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
)
|
||||
out2 = messages.TxOutputType(
|
||||
# tb1pn2d0yjeedavnkd8z8lhm566p0f2utm3lgvxrsdehnl94y34txmts5s7t4c
|
||||
address_n=parse_path("86'/1'/0'/1/0"),
|
||||
script_type=messages.OutputScriptType.PAYTOTAPROOT,
|
||||
amount=6800 + 13000 - 200 - 15000,
|
||||
)
|
||||
with client:
|
||||
client.set_expected_responses(
|
||||
[
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_output(0),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_finished(),
|
||||
]
|
||||
)
|
||||
_, serialized_tx = btc.sign_tx(
|
||||
client, "Testnet", [inp1, inp2], [out1, out2], prev_txes=TX_API
|
||||
)
|
||||
|
||||
assert (
|
||||
serialized_tx.hex()
|
||||
== "010000000001029f67664b8972ae01498e25ea98a37889f19aa86a2f39ddad84ff31da312e86df0000000000ffffffff9b117a776a9aaf70d4c3ffe89f009dcd23210a03d649ee5e38791d83902ec33a0100000000ffffffff02983a000000000000160014f0ca4661a8c7f4edad7da1c864a8bd3db05d4ac4f8110000000000002251209a9af24b396f593b34e23fefba6b417a55c5ee3f430c3837379fcb5246ab36d70140b51992353d2f99b7b620c0882cb06694996f1b6c7e62a3c1d3036e0f896fbf0b92f3d9aeab94f2454809a501715667345f702c8214693f469225de5f6636b86b01409956e47403278bf76eecbbbc3af0c2731d8347763825248a2e0f39aca5a684a7d5054e7222a1033fb5864a886180f1a8c64adab12433c78298d1f83e4c8f46e100000000"
|
||||
)
|
||||
|
||||
def test_send_mixed(self, client):
|
||||
inp1 = messages.TxInputType(
|
||||
# 2MutHjgAXkqo3jxX2DZWorLAckAnwTxSM9V
|
||||
address_n=parse_path("49'/1'/1'/0/0"),
|
||||
amount=20000,
|
||||
prev_hash=TXHASH_8c3ea7,
|
||||
prev_index=0,
|
||||
script_type=messages.InputScriptType.SPENDP2SHWITNESS,
|
||||
)
|
||||
inp2 = messages.TxInputType(
|
||||
# tb1q7r9yvcdgcl6wmtta58yxf29a8kc96jkyxl7y88
|
||||
address_n=parse_path("84'/1'/1'/0/0"),
|
||||
amount=15000,
|
||||
prev_hash=TXHASH_7956f1,
|
||||
prev_index=0,
|
||||
script_type=messages.InputScriptType.SPENDWITNESS,
|
||||
)
|
||||
inp3 = messages.TxInputType(
|
||||
# tb1paxhjl357yzctuf3fe58fcdx6nul026hhh6kyldpfsf3tckj9a3wslqd7zd
|
||||
address_n=parse_path("86'/1'/1'/0/0"),
|
||||
amount=4450,
|
||||
prev_hash=TXHASH_901593,
|
||||
prev_index=0,
|
||||
script_type=messages.InputScriptType.SPENDTAPROOT,
|
||||
)
|
||||
inp4 = messages.TxInputType(
|
||||
# msUqRgCWS7ryuFcF34EaKTrsTe3xHra128
|
||||
address_n=parse_path("44'/1'/1'/0/0"),
|
||||
amount=10000,
|
||||
prev_hash=TXHASH_3ac32e,
|
||||
prev_index=2,
|
||||
script_type=messages.InputScriptType.SPENDADDRESS,
|
||||
)
|
||||
out1 = messages.TxOutputType(
|
||||
address="tb1q6xnnna3g7lk22h5tn8nlx2ezmndlvuk556w4w3",
|
||||
amount=25000,
|
||||
script_type=messages.OutputScriptType.PAYTOWITNESS,
|
||||
)
|
||||
out2 = messages.TxOutputType(
|
||||
address="mfnMbVFC1rH4p9GNbjkMfrAjyKRLycFAzA",
|
||||
script_type=messages.OutputScriptType.PAYTOADDRESS,
|
||||
amount=7000,
|
||||
)
|
||||
out3 = messages.TxOutputType(
|
||||
address="2MvAG8m2xSf83FgeR4ZpUtaubpLNjAMMoka",
|
||||
amount=6900,
|
||||
script_type=messages.OutputScriptType.PAYTOP2SHWITNESS,
|
||||
)
|
||||
out4 = messages.TxOutputType(
|
||||
op_return_data=b"test of op_return data",
|
||||
amount=0,
|
||||
script_type=messages.OutputScriptType.PAYTOOPRETURN,
|
||||
)
|
||||
out5 = messages.TxOutputType(
|
||||
address="tb1ptgp9w0mm89ms43flw0gkrhyx75gyc6qjhtpf0jmt5sv0dufpnsrsyv9nsz",
|
||||
amount=10000,
|
||||
script_type=messages.OutputScriptType.PAYTOTAPROOT,
|
||||
)
|
||||
|
||||
with client:
|
||||
client.set_expected_responses(
|
||||
[
|
||||
# process inputs
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_input(2),
|
||||
request_input(3),
|
||||
# approve outputs
|
||||
request_output(0),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(1),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(2),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(3),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
request_output(4),
|
||||
messages.ButtonRequest(code=B.ConfirmOutput),
|
||||
messages.ButtonRequest(code=B.SignTx),
|
||||
# verify inputs
|
||||
request_input(0),
|
||||
request_meta(TXHASH_8c3ea7),
|
||||
request_input(0, TXHASH_8c3ea7),
|
||||
request_output(0, TXHASH_8c3ea7),
|
||||
request_output(1, TXHASH_8c3ea7),
|
||||
request_input(1),
|
||||
request_meta(TXHASH_7956f1),
|
||||
request_input(0, TXHASH_7956f1),
|
||||
request_input(1, TXHASH_7956f1),
|
||||
request_output(0, TXHASH_7956f1),
|
||||
request_output(1, TXHASH_7956f1),
|
||||
request_input(2),
|
||||
request_meta(TXHASH_901593),
|
||||
request_input(0, TXHASH_901593),
|
||||
request_output(0, TXHASH_901593),
|
||||
request_input(3),
|
||||
request_meta(TXHASH_3ac32e),
|
||||
request_input(0, TXHASH_3ac32e),
|
||||
request_output(0, TXHASH_3ac32e),
|
||||
request_output(1, TXHASH_3ac32e),
|
||||
request_output(2, TXHASH_3ac32e),
|
||||
# serialize (segwit) inputs
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_input(2),
|
||||
# serialize and sign legacy input
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_input(2),
|
||||
request_input(3),
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
request_output(2),
|
||||
request_output(3),
|
||||
request_output(4),
|
||||
# serialize outputs
|
||||
request_output(0),
|
||||
request_output(1),
|
||||
request_output(2),
|
||||
request_output(3),
|
||||
request_output(4),
|
||||
# sign segwit inputs
|
||||
request_input(0),
|
||||
request_input(1),
|
||||
request_input(2),
|
||||
request_finished(),
|
||||
]
|
||||
)
|
||||
_, serialized_tx = btc.sign_tx(
|
||||
client,
|
||||
"Testnet",
|
||||
[inp1, inp2, inp3, inp4],
|
||||
[out1, out2, out3, out4, out5],
|
||||
prev_txes=TX_API,
|
||||
)
|
||||
|
||||
assert (
|
||||
serialized_tx.hex()
|
||||
== "010000000001045d77b6e482d770031ad3ce3423727cc1707bc2c82e729b1189d2b60aa1a73e8c0000000017160014a33c6e24c99e108b97bc411e7e9ef31e9d5d6164ffffffff7b350e3faca092f39883d7086cdd502c82b6f0314ab61541b062733edef156790000000000ffffffff852e125137abca2dd7a42837dccfc34edc358c72eefd62978d6747d3be9315900000000000ffffffff9b117a776a9aaf70d4c3ffe89f009dcd23210a03d649ee5e38791d83902ec33a020000006b483045022100f6bd64136839b49822cf7e2050bc5c91346fc18b5cf97a945d4fd6c502f712d002207d1859e66d218f705b704f3cfca0c75410349bb1f50623f4fc2d09d5d8df0a3f012103bae960983f83e28fcb8f0e5f3dc1f1297b9f9636612fd0835b768e1b7275fb9dffffffff05a861000000000000160014d1a739f628f7eca55e8b99e7f32b22dcdbf672d4581b0000000000001976a91402e9b094fd98e2a26e805894eb78f7ff3fef199b88acf41a00000000000017a9141ff816cbeb74817050de585ceb2c772ebf71147a870000000000000000186a1674657374206f66206f705f72657475726e206461746110270000000000002251205a02573f7b39770ac53f73d161dc86f5104c6812bac297cb6ba418f6f1219c070247304402205fae7fa2b5141548593d5623ce5bd82ee18dfc751c243526039c91848efd603702200febfbe3467a68c599245ff89055514f26e146c79b58d932ced2325e6dad1b1a0121021630971f20fa349ba940a6ba3706884c41579cd760c89901374358db5dd545b90247304402201b21212100c84207697cebb852374669c382ed97cbd08afbbdfe1b302802161602206b32b2140d094cf5b7e758135961c95478c8e82fea0df30f56ccee284b79eaea012103f6b2377d52960a6094ec158cf19dcf9e33b3da4798c2302aa5806483ed4187ae01404a81e4b7f55d6d4a26923c5e2daf3cc86ed6030f83ea6e7bb16d7b81b988b34585be21a64ab45ddcc2fb9f17be2dfeff6b22cf943bc3fc8f125a7f463af428ed0000000000"
|
||||
)
|
||||
@ -0,0 +1,26 @@
|
||||
{
|
||||
"bin_outputs": [
|
||||
{
|
||||
"amount": 48534,
|
||||
"script_pubkey": "0014dacad798894330a1743b940708fd8aded9f6727e"
|
||||
},
|
||||
{
|
||||
"amount": 13000,
|
||||
"script_pubkey": "51203ad9b641978673e88ee4d9f4e5d63400c1b2a8304c09726bb19d10ead2829cc2"
|
||||
},
|
||||
{
|
||||
"amount": 10000,
|
||||
"script_pubkey": "76a914833af1a25e63456bf032ba032cc2788122f44d5288ac"
|
||||
}
|
||||
],
|
||||
"inputs": [
|
||||
{
|
||||
"prev_hash": "ec16dc5a539c5d60001a7471c37dbb0b5294c289c77df8bd07870b30d73e2231",
|
||||
"prev_index": 1,
|
||||
"script_sig": "",
|
||||
"sequence": 4294967293
|
||||
}
|
||||
],
|
||||
"lock_time": 2100847,
|
||||
"version": 2
|
||||
}
|
||||
@ -0,0 +1,28 @@
|
||||
{
|
||||
"bin_outputs": [
|
||||
{
|
||||
"amount": 15000,
|
||||
"script_pubkey": "0014f0ca4661a8c7f4edad7da1c864a8bd3db05d4ac4"
|
||||
},
|
||||
{
|
||||
"amount": 4600,
|
||||
"script_pubkey": "51209a9af24b396f593b34e23fefba6b417a55c5ee3f430c3837379fcb5246ab36d7"
|
||||
}
|
||||
],
|
||||
"inputs": [
|
||||
{
|
||||
"prev_hash": "df862e31da31ff84addd392f6aa89af18978a398ea258e4901ae72894b66679f",
|
||||
"prev_index": 0,
|
||||
"script_sig": "",
|
||||
"sequence": 4294967295
|
||||
},
|
||||
{
|
||||
"prev_hash": "3ac32e90831d79385eee49d6030a2123cd9d009fe8ffc3d470af9a6a777a119b",
|
||||
"prev_index": 1,
|
||||
"script_sig": "",
|
||||
"sequence": 4294967295
|
||||
}
|
||||
],
|
||||
"lock_time": 0,
|
||||
"version": 1
|
||||
}
|
||||
@ -0,0 +1,22 @@
|
||||
{
|
||||
"bin_outputs": [
|
||||
{
|
||||
"amount": 20000,
|
||||
"script_pubkey": "a9141cf2c55c8ae95acb87c88329553f88504b080e9787"
|
||||
},
|
||||
{
|
||||
"amount": 28392,
|
||||
"script_pubkey": "001426f4e6254f1689cf7a846509dc2ecbaa7f542745"
|
||||
}
|
||||
],
|
||||
"inputs": [
|
||||
{
|
||||
"prev_hash": "3ac32e90831d79385eee49d6030a2123cd9d009fe8ffc3d470af9a6a777a119b",
|
||||
"prev_index": 0,
|
||||
"script_sig": "",
|
||||
"sequence": 4294967293
|
||||
}
|
||||
],
|
||||
"lock_time": 2100861,
|
||||
"version": 2
|
||||
}
|
||||
@ -0,0 +1,18 @@
|
||||
{
|
||||
"bin_outputs": [
|
||||
{
|
||||
"amount": 4450,
|
||||
"script_pubkey": "5120e9af2fc69e20b0be2629cd0e9c34da9f3ef56af7beac4fb4298262bc5a45ec5d"
|
||||
}
|
||||
],
|
||||
"inputs": [
|
||||
{
|
||||
"prev_hash": "7956f1de3e7362b04115b64a31f0b6822c50dd6c08d78398f392a0ac3f0e357b",
|
||||
"prev_index": 1,
|
||||
"script_sig": "",
|
||||
"sequence": 4294967295
|
||||
}
|
||||
],
|
||||
"lock_time": 0,
|
||||
"version": 1
|
||||
}
|
||||
@ -716,6 +716,9 @@
|
||||
"test_msg_signtx_segwit_native.py-test_send_p2sh": "a61bf07f8f254b6c8386ae5462ea3688b307abce563a61c9c91f5908b0dcaf2a",
|
||||
"test_msg_signtx_segwit_native.py-test_send_p2sh_change": "8288ac83c0e6d5924780ee3b1278e36063ad6c86a288d0b8abbb52eaaf0cc4e7",
|
||||
"test_msg_signtx_segwit_native.py-test_send_to_taproot": "9a138b34e18e310617cc7d79cf2efa1c180948d1a4d3647f6de1e48d94a441b8",
|
||||
"test_msg_signtx_taproot.py-test_send_mixed": "750648ddd29e1eee28bf2d4f292a131821cdf6413ccafbfed88ee5c8f70841b4",
|
||||
"test_msg_signtx_taproot.py-test_send_p2tr": "3ea24494ae8e004645d31afcd798066a784c44acb613588c407c4717ad9a56d6",
|
||||
"test_msg_signtx_taproot.py-test_send_two_with_change": "b394ee2dcdc7d8b5906b9b4d7f52ac69af7e9e5f93229980f6450a1d4e83bd9b",
|
||||
"test_msg_signtx_zcash.py-test_external_presigned": "08a22a49793201f8387a085a554f006387d1df1b37ed8a34bef7aa79b363b44b",
|
||||
"test_msg_signtx_zcash.py-test_one_one_fee_sapling": "b8aff4ae3c9b417acadf1ede52d7d45870dd4ef1f9251bf8e8f2ebf5280b9f3b",
|
||||
"test_msg_signtx_zcash.py-test_spend_old_versions": "c3f75b5c8ceb63ad2102d270689f55fbf3b204581bec7dcee930f875416e1a07",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user