87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
from pathlib import Path
|
|
from typing import Optional
|
|
import pytest
|
|
|
|
from py_ecc.bls import G2ProofOfPossession as bls
|
|
|
|
from staking_deposit.key_handling.key_derivation.path import mnemonic_and_path_to_key
|
|
|
|
from ragger.bip.seed import SPECULOS_MNEMONIC
|
|
from ragger.error import ExceptionRAPDU
|
|
from ragger.firmware import Firmware
|
|
from ragger.backend import BackendInterface
|
|
from ragger.navigator.navigation_scenario import NavigateWithScenario
|
|
from ragger.bip import calculate_public_key_and_chaincode, CurveChoice
|
|
|
|
from client.client import EthAppClient, StatusWord
|
|
import client.response_parser as ResponseParser
|
|
|
|
|
|
@pytest.fixture(name="with_chaincode", params=[True, False])
|
|
def with_chaincode_fixture(request) -> bool:
|
|
return request.param
|
|
|
|
|
|
@pytest.fixture(name="chain", params=[None, 1, 2, 5, 137])
|
|
def chain_fixture(request) -> Optional[int]:
|
|
return request.param
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"path, suffix",
|
|
[
|
|
("m/44'/60'/0'/0/0", "60"),
|
|
("m/44'/700'/1'/0/0", "700")
|
|
],
|
|
)
|
|
def test_get_pk_rejected(backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
default_screenshot_path: Path,
|
|
path,
|
|
suffix):
|
|
app_client = EthAppClient(backend)
|
|
|
|
with pytest.raises(ExceptionRAPDU) as e:
|
|
with app_client.get_public_addr(bip32_path=path):
|
|
scenario_navigator.address_review_reject(default_screenshot_path, f"get_pk_rejected_{suffix}")
|
|
|
|
assert e.value.status == StatusWord.CONDITION_NOT_SATISFIED
|
|
|
|
|
|
def test_get_pk(backend: BackendInterface,
|
|
default_screenshot_path: Path,
|
|
scenario_navigator: NavigateWithScenario,
|
|
with_chaincode: bool,
|
|
chain: Optional[int]):
|
|
app_client = EthAppClient(backend)
|
|
|
|
with app_client.get_public_addr(chaincode=with_chaincode, chain_id=chain):
|
|
scenario_navigator.address_review_approve(default_screenshot_path, f"get_pk_{chain}")
|
|
|
|
pk, _, chaincode = ResponseParser.pk_addr(app_client.response().data, with_chaincode)
|
|
ref_pk, ref_chaincode = calculate_public_key_and_chaincode(curve=CurveChoice.Secp256k1,
|
|
path="m/44'/60'/0'/0/0")
|
|
assert pk.hex() == ref_pk
|
|
if with_chaincode:
|
|
assert chaincode.hex() == ref_chaincode
|
|
|
|
|
|
def test_get_eth2_pk(firmware: Firmware,
|
|
backend: BackendInterface,
|
|
scenario_navigator: NavigateWithScenario,
|
|
test_name: str,
|
|
default_screenshot_path: Path):
|
|
|
|
app_client = EthAppClient(backend)
|
|
|
|
path="m/12381/3600/0/0"
|
|
with app_client.get_eth2_public_addr(bip32_path=path):
|
|
scenario_navigator.address_review_approve(default_screenshot_path, test_name)
|
|
|
|
pk = app_client.response().data
|
|
ref_pk = bls.SkToPk(mnemonic_and_path_to_key(SPECULOS_MNEMONIC, path))
|
|
if firmware.name in ("stax", "flex"):
|
|
pk = pk[1:49]
|
|
|
|
assert pk == ref_pk
|