1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-25 15:58:08 +00:00
trezor-firmware/tests/device_tests/solana/test_sign_tx.py
2023-12-05 22:11:41 +01:00

95 lines
3.3 KiB
Python

# This file is part of the Trezor project.
#
# Copyright (C) 2012-2023 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 messages
from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.solana import sign_tx
from trezorlib.tools import parse_path
from ...common import parametrize_using_common_fixtures
from .construct.instructions import PROGRAMS, UnknownInstruction
from .construct.transaction import Message, RawInstruction
pytestmark = [
pytest.mark.altcoin,
pytest.mark.solana,
pytest.mark.skip_t1,
]
@parametrize_using_common_fixtures(
"solana/sign_tx.system_program.json",
"solana/sign_tx.stake_program.json",
"solana/sign_tx.associated_token_account_program.json",
"solana/sign_tx.memo_program.json",
"solana/sign_tx.compute_budget_program.json",
"solana/sign_tx.token_program.json",
"solana/sign_tx.unknown_instructions.json",
"solana/sign_tx.predefined_transactions.json",
)
def test_solana_sign_tx(client: Client, parameters, result):
client.init_device(new_session=True)
serialized_tx = _serialize_tx(parameters["construct"])
actual_result = sign_tx(
client,
address_n=parse_path(parameters["address"]),
serialized_tx=serialized_tx,
additional_info=messages.SolanaTxAdditionalInfo(
token_accounts_infos=[
messages.SolanaTxTokenAccountInfo(
base_address=token_account["base_address"],
token_program=token_account["token_program"],
token_mint=token_account["token_mint"],
token_account=token_account["token_account"],
)
for token_account in parameters["additional_info"][
"token_accounts_infos"
]
]
)
if "additional_info" in parameters
else None,
)
assert actual_result.signature == bytes.fromhex(result["expected_signature"])
def _serialize_tx(tx_construct):
serialized_instructions = []
for instruction in tx_construct["instructions"]:
program = tx_construct["accounts"][instruction["program_index"]]
builder = PROGRAMS.get(program, UnknownInstruction)
serialized_instruction = builder.build(instruction)
raw_instruction = RawInstruction.parse(serialized_instruction)
serialized_instructions.append(raw_instruction)
message_construct = {
"version": tx_construct["version"],
"header": tx_construct["header"],
"accounts": tx_construct["accounts"],
"blockhash": tx_construct["blockhash"],
"instructions": serialized_instructions,
"luts": tx_construct["luts"],
}
return Message.build(
message_construct,
)