mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-26 09:28:13 +00:00
113 lines
4.1 KiB
Python
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 TrezorClientDebugLink as Client
|
|
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(client: Client, chunkify: bool, parameters, result):
|
|
client.init_device(new_session=True, derive_cardano=True)
|
|
|
|
derivation_type = CardanoDerivationType.__members__[
|
|
parameters.get("derivation_type", "ICARUS_TREZOR")
|
|
]
|
|
|
|
address = get_address(
|
|
client,
|
|
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(client: Client, parameters, result):
|
|
with client:
|
|
IF = InputFlowShowXpubQRCode(client, passphrase=bool(client.ui.passphrase))
|
|
client.set_input_flow(IF.get())
|
|
client.init_device(new_session=True, derive_cardano=True)
|
|
|
|
derivation_type = CardanoDerivationType.__members__[
|
|
parameters.get("derivation_type", "ICARUS_TREZOR")
|
|
]
|
|
key = get_public_key(
|
|
client, 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"]
|