129 lines
3.8 KiB
Python
129 lines
3.8 KiB
Python
import pytest
|
|
from ragger.error import ExceptionRAPDU
|
|
from app.client import EthereumClient, StatusWord
|
|
from app.setting import SettingType
|
|
import struct
|
|
|
|
# Values used across all tests
|
|
CHAIN_ID = 1
|
|
NAME = "ledger.eth"
|
|
ADDR = bytes.fromhex("0011223344556677889900112233445566778899")
|
|
KEY_ID = 1
|
|
ALGO_ID = 1
|
|
BIP32_PATH = "m/44'/60'/0'/0/0"
|
|
NONCE = 21
|
|
GAS_PRICE = 13000000000
|
|
GAS_LIMIT = 21000
|
|
AMOUNT = 1.22
|
|
|
|
@pytest.fixture(params=[False, True])
|
|
def verbose(request) -> bool:
|
|
return request.param
|
|
|
|
def common(app_client: EthereumClient) -> int:
|
|
if app_client._client.firmware.device == "nanos":
|
|
pytest.skip("Not supported on LNS")
|
|
return app_client.get_challenge()
|
|
|
|
|
|
def test_send_fund(app_client: EthereumClient, verbose: bool):
|
|
challenge = common(app_client)
|
|
|
|
if verbose:
|
|
app_client.settings_set({
|
|
SettingType.VERBOSE_ENS: True
|
|
})
|
|
|
|
app_client.provide_domain_name(challenge, NAME, ADDR)
|
|
|
|
app_client.send_fund(BIP32_PATH,
|
|
NONCE,
|
|
GAS_PRICE,
|
|
GAS_LIMIT,
|
|
ADDR,
|
|
AMOUNT,
|
|
CHAIN_ID,
|
|
"domain_name_verbose_" + str(verbose))
|
|
|
|
def test_send_fund_wrong_challenge(app_client: EthereumClient):
|
|
caught = False
|
|
challenge = common(app_client)
|
|
|
|
try:
|
|
app_client.provide_domain_name(~challenge & 0xffffffff, NAME, ADDR)
|
|
except ExceptionRAPDU as e:
|
|
assert e.status == StatusWord.INVALID_DATA
|
|
else:
|
|
assert False # An exception should have been raised
|
|
|
|
def test_send_fund_wrong_addr(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
app_client.provide_domain_name(challenge, NAME, ADDR)
|
|
|
|
addr = bytearray(ADDR)
|
|
addr.reverse()
|
|
|
|
app_client.send_fund(BIP32_PATH,
|
|
NONCE,
|
|
GAS_PRICE,
|
|
GAS_LIMIT,
|
|
addr,
|
|
AMOUNT,
|
|
CHAIN_ID,
|
|
"domain_name_wrong_addr")
|
|
|
|
def test_send_fund_non_mainnet(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
app_client.provide_domain_name(challenge, NAME, ADDR)
|
|
|
|
app_client.send_fund(BIP32_PATH,
|
|
NONCE,
|
|
GAS_PRICE,
|
|
GAS_LIMIT,
|
|
ADDR,
|
|
AMOUNT,
|
|
5,
|
|
"domain_name_non_mainnet")
|
|
|
|
def test_send_fund_domain_too_long(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
try:
|
|
app_client.provide_domain_name(challenge, "ledger" + "0"*25 + ".eth", ADDR)
|
|
except ExceptionRAPDU as e:
|
|
assert e.status == StatusWord.INVALID_DATA
|
|
else:
|
|
assert False # An exception should have been raised
|
|
|
|
def test_send_fund_domain_invalid_character(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
try:
|
|
app_client.provide_domain_name(challenge, "l\xe8dger.eth", ADDR)
|
|
except ExceptionRAPDU as e:
|
|
assert e.status == StatusWord.INVALID_DATA
|
|
else:
|
|
assert False # An exception should have been raised
|
|
|
|
def test_send_fund_uppercase(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
try:
|
|
app_client.provide_domain_name(challenge, NAME.upper(), ADDR)
|
|
except ExceptionRAPDU as e:
|
|
assert e.status == StatusWord.INVALID_DATA
|
|
else:
|
|
assert False # An exception should have been raised
|
|
|
|
def test_send_fund_domain_non_ens(app_client: EthereumClient):
|
|
challenge = common(app_client)
|
|
|
|
try:
|
|
app_client.provide_domain_name(challenge, "ledger.hte", ADDR)
|
|
except ExceptionRAPDU as e:
|
|
assert e.status == StatusWord.INVALID_DATA
|
|
else:
|
|
assert False # An exception should have been raised
|