diff --git a/tests/test_msg_signtx_taproot.py b/tests/test_msg_signtx_taproot.py index a51f378e..373d6a25 100644 --- a/tests/test_msg_signtx_taproot.py +++ b/tests/test_msg_signtx_taproot.py @@ -87,7 +87,36 @@ def test_send_p2tr(self): script_type=proto_types.PAYTOADDRESS, ) - (signatures, _) = self.client.sign_tx("Bitcoin", [inp1], [out1]) + with self.client: + self.client.set_expected_responses([ + proto.TxRequest( + request_type=proto_types.TXINPUT, + details=proto_types.TxRequestDetailsType( + request_index=0)), + proto.TxRequest( + request_type=proto_types.TXOUTPUT, + details=proto_types.TxRequestDetailsType( + request_index=0)), + proto.ButtonRequest( + code=proto_types.ButtonRequest_ConfirmOutput), + proto.ButtonRequest( + code=proto_types.ButtonRequest_SignTx), + proto.TxRequest( + request_type=proto_types.TXINPUT, + details=proto_types.TxRequestDetailsType( + request_index=0)), + proto.TxRequest( + request_type=proto_types.TXOUTPUT, + details=proto_types.TxRequestDetailsType( + request_index=0)), + proto.TxRequest( + request_type=proto_types.TXINPUT, + details=proto_types.TxRequestDetailsType( + request_index=0)), + proto.TxRequest(request_type=proto_types.TXFINISHED), + ]) + (signatures, _) = self.client.sign_tx( + "Bitcoin", [inp1], [out1]) self.assertEqual(len(signatures), 1) self.assertEqual(hexlify(signatures[0]).decode(), EXPECTED_WITNESS)