1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 22:18:16 +00:00
trezor-firmware/tests/device_tests/cardano/test_address_public_key.py
2024-11-29 11:12:10 +01:00

113 lines
4.1 KiB
Python

# This file is part of the Trezor project.
#
# Copyright (C) 2012-2019 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.cardano import (
create_address_parameters,
get_address,
get_public_key,
parse_optional_bytes,
)
from trezorlib.debuglink import SessionDebugWrapper as Session
from trezorlib.messages import CardanoAddressType, CardanoDerivationType
from trezorlib.tools import parse_path
from ...common import parametrize_using_common_fixtures
from ...input_flows import InputFlowShowXpubQRCode
pytestmark = [
pytest.mark.altcoin,
pytest.mark.cardano,
pytest.mark.models("core"),
]
@parametrize_using_common_fixtures(
"cardano/get_address_byron.json",
"cardano/get_address_byron.slip39.json",
"cardano/get_base_address.json",
"cardano/get_base_address_with_staking_key_hash.json",
"cardano/get_base_address_with_script_hashes.json",
"cardano/get_enterprise_address.json",
"cardano/get_pointer_address.json",
"cardano/get_reward_address.json",
"cardano/get_base_address.derivations.json",
)
@pytest.mark.parametrize("chunkify", (True, False))
def test_cardano_get_address(session: Session, chunkify: bool, parameters, result):
# session.init_device(new_session=True, derive_cardano=True)
derivation_type = CardanoDerivationType.__members__[
parameters.get("derivation_type", "ICARUS_TREZOR")
]
address = get_address(
session,
address_parameters=create_address_parameters(
address_type=getattr(
CardanoAddressType, parameters["address_type"].upper()
),
address_n=(
parse_path(parameters.get("path")) if "path" in parameters else None
),
address_n_staking=(
parse_path(parameters.get("staking_path"))
if "staking_path" in parameters
else None
),
staking_key_hash=parse_optional_bytes(parameters.get("staking_key_hash")),
block_index=parameters.get("block_index"),
tx_index=parameters.get("tx_index"),
certificate_index=parameters.get("certificate_index"),
script_payment_hash=parse_optional_bytes(
parameters.get("script_payment_hash")
),
script_staking_hash=parse_optional_bytes(
parameters.get("script_staking_hash")
),
),
protocol_magic=parameters["protocol_magic"],
network_id=parameters["network_id"],
show_display=True,
derivation_type=derivation_type,
chunkify=chunkify,
)
assert address == result["expected_address"]
@parametrize_using_common_fixtures(
"cardano/get_public_key.json",
"cardano/get_public_key.slip39.json",
"cardano/get_public_key.derivations.json",
)
def test_cardano_get_public_key(session: Session, parameters, result):
with session, session.client as client:
IF = InputFlowShowXpubQRCode(client, passphrase=False)
client.set_input_flow(IF.get())
# session.init_device(new_session=True, derive_cardano=True)
derivation_type = CardanoDerivationType.__members__[
parameters.get("derivation_type", "ICARUS_TREZOR")
]
key = get_public_key(
session, parse_path(parameters["path"]), derivation_type, show_display=True
)
assert key.node.public_key.hex() == result["public_key"]
assert key.node.chain_code.hex() == result["chain_code"]
assert key.xpub == result["public_key"] + result["chain_code"]