trezor-firmware/tests/device_tests/cardano/test_sign_message.py

import pytest

from trezorlib import cardano, messages, tools
from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.exceptions import TrezorFailure

from ...common import parametrize_using_common_fixtures

pytestmark = [
    pytest.mark.altcoin,
    pytest.mark.cardano,
    pytest.mark.skip_t1,
]


@parametrize_using_common_fixtures("cardano/sign_message.json")
def test_cardano_sign_message(client: Client, parameters, result):
    response = call_sign_message(client, parameters)
    assert response == _transform_expected_result(result)


@parametrize_using_common_fixtures("cardano/sign_message.failed.json")
def test_cardano_sign_message_failed(client: Client, parameters, result):
    with pytest.raises(TrezorFailure, match=result["error_message"]):
        call_sign_message(client, parameters)


def call_sign_message(
    client: Client,
    parameters,
) -> messages.CardanoSignMessageFinished:
    client.init_device(new_session=True, derive_cardano=True)

    with client:
        return cardano.sign_message(
            client,
            payload=bytes.fromhex(parameters["payload"]),
            hash_payload=parameters["hash_payload"],
            prefer_hex_display=parameters["prefer_hex_display"],
            signing_path=tools.parse_path(parameters["signing_path"]),
            address_parameters=cardano.parse_optional_address_parameters(
                parameters.get("address_parameters")
            ),
            protocol_magic=parameters.get("protocol_magic"),
            network_id=parameters.get("network_id"),
        )


def _transform_expected_result(result: dict) -> messages.CardanoSignMessageFinished:
    return messages.CardanoSignMessageFinished(
        signature=bytes.fromhex(result["signature"]),
        address=bytes.fromhex(result["address"]),
        pub_key=bytes.fromhex(result["pub_key"]),
    )