mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-24 00:18:26 +00:00
75 lines
2.7 KiB
Python
75 lines
2.7 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
|
||
|
|
from trezorlib.messages import CardanoAddressType
|
||
|
|
from trezorlib.tools import parse_path
|
||
|
|
|
||
|
|
from ...common import parametrize_using_common_fixtures
|
||
|
|
|
||
|
|
pytestmark = [
|
||
|
|
pytest.mark.altcoin,
|
||
|
|
pytest.mark.cardano,
|
||
|
|
pytest.mark.skip_t1,
|
||
|
|
pytest.mark.skip_ui,
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
@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_enterprise_address.json",
|
||
|
|
"cardano/get_pointer_address.json",
|
||
|
|
"cardano/get_reward_address.json",
|
||
|
|
)
|
||
|
|
def test_cardano_get_address(client, parameters, result):
|
||
|
|
address = get_address(
|
||
|
|
client,
|
||
|
|
address_parameters=create_address_parameters(
|
||
|
|
address_type=getattr(
|
||
|
|
CardanoAddressType, parameters["address_type"].upper()
|
||
|
|
),
|
||
|
|
address_n=parse_path(parameters["path"]),
|
||
|
|
address_n_staking=parse_path(parameters.get("staking_path"))
|
||
|
|
if "staking_path" in parameters
|
||
|
|
else None,
|
||
|
|
staking_key_hash=bytes.fromhex(parameters.get("staking_key_hash"))
|
||
|
|
if "staking_key_hash" in parameters
|
||
|
|
else None,
|
||
|
|
block_index=parameters.get("block_index"),
|
||
|
|
tx_index=parameters.get("tx_index"),
|
||
|
|
certificate_index=parameters.get("certificate_index"),
|
||
|
|
),
|
||
|
|
protocol_magic=parameters["protocol_magic"],
|
||
|
|
network_id=parameters["network_id"],
|
||
|
|
)
|
||
|
|
assert address == result["expected_address"]
|
||
|
|
|
||
|
|
|
||
|
|
@parametrize_using_common_fixtures(
|
||
|
|
"cardano/get_public_key.json", "cardano/get_public_key.slip39.json"
|
||
|
|
)
|
||
|
|
def test_cardano_get_public_key(client, parameters, result):
|
||
|
|
key = get_public_key(client, parse_path(parameters["path"]))
|
||
|
|
|
||
|
|
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"]
|