arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-27 16:48:09 +00:00
Code Issues Releases Wiki Activity

tests: convert setup_client to a marker

Browse Source
This commit is contained in:
matejcik 2019-08-26 18:26:57 +02:00
parent 1e4cf8d801
commit 4b2235ab78
16 changed files with 51 additions and 69 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
tests/device_tests/conftest.py
View File

@ -14,7 +14,6 @@
# You should have received a copy of the License along with this library. # 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>. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
import functools
import os import os
import pytest import pytest
@ -58,39 +57,42 @@ def device_version():
@pytest.fixture(scope="function") @pytest.fixture(scope="function")
def client(): def client(request):
client = get_device() client = get_device()
wipe_device(client) wipe_device(client)
client.open() client.open()
yield client
client.close()
# fmt: off
setup_params = dict(
uninitialized=False,
mnemonic=" ".join(["all"] * 12),
pin=None,
passphrase=False,
)
# fmt: on
def setup_client(mnemonic=None, pin="", passphrase=False): marker = request.node.get_closest_marker("setup_client")
if mnemonic is None: if marker:
mnemonic = " ".join(["all"] * 12) setup_params.update(marker.kwargs)
if pin is True:
pin = "1234" if not setup_params["uninitialized"]:
if setup_params["pin"] is True:
setup_params["pin"] = "1234"
def client_decorator(function):
@functools.wraps(function)
def wrapper(client, *args, **kwargs):
debuglink.load_device_by_mnemonic( debuglink.load_device_by_mnemonic(
client, client,
mnemonic=mnemonic, mnemonic=setup_params["mnemonic"],
pin=pin, pin=setup_params["pin"],
passphrase_protection=passphrase, passphrase_protection=setup_params["passphrase"],
label="test", label="test",
language="english", language="english",
) )
if TREZOR_VERSION > 1 and passphrase: if setup_params["passphrase"] and client.features.model != "1":
apply_settings(client, passphrase_source=PASSPHRASE_ON_HOST) apply_settings(client, passphrase_source=PASSPHRASE_ON_HOST)
return function(*args, client=client, **kwargs)
return wrapper yield client
client.close()
return client_decorator
def pytest_configure(config): def pytest_configure(config):
@ -104,6 +106,10 @@ def pytest_configure(config):
# register known markers # register known markers
config.addinivalue_line("markers", "skip_t1: skip the test on Trezor One") config.addinivalue_line("markers", "skip_t1: skip the test on Trezor One")
config.addinivalue_line("markers", "skip_t2: skip the test on Trezor T") config.addinivalue_line("markers", "skip_t2: skip the test on Trezor T")
config.addinivalue_line(
"markers",
'setup_client(mnemonic="all all all...", pin=None, passphrase=False, uninitialized=False): configure the client instance',
)
with open(os.path.join(os.path.dirname(__file__), "REGISTERED_MARKERS")) as f: with open(os.path.join(os.path.dirname(__file__), "REGISTERED_MARKERS")) as f:
for line in f: for line in f:
config.addinivalue_line("markers", line.strip()) config.addinivalue_line("markers", line.strip())

4
tests/device_tests/test_cancel.py
View File

@ -19,10 +19,7 @@ import pytest
import trezorlib.messages as m import trezorlib.messages as m
from trezorlib.exceptions import Cancelled from trezorlib.exceptions import Cancelled
from .conftest import setup_client
@setup_client()
@pytest.mark.parametrize( @pytest.mark.parametrize(
"message", "message",
[ [
@ -46,7 +43,6 @@ def test_cancel_message_via_cancel(client, message):
client.call(message) client.call(message)
@setup_client()
@pytest.mark.parametrize( @pytest.mark.parametrize(
"message", "message",
[ [

4
tests/device_tests/test_msg_binance_get_address.py
View File

@ -3,8 +3,6 @@ import pytest
from trezorlib.binance import get_address from trezorlib.binance import get_address
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
BINANCE_ADDRESS_TEST_VECTORS = [ BINANCE_ADDRESS_TEST_VECTORS = [
("m/44'/714'/0'/0/0", "bnb1hgm0p7khfk85zpz5v0j8wnej3a90w709vhkdfu"), ("m/44'/714'/0'/0/0", "bnb1hgm0p7khfk85zpz5v0j8wnej3a90w709vhkdfu"),
("m/44'/714'/0'/0/1", "bnb1egswqkszzfc2uq78zjslc6u2uky4pw46x4rstd"), ("m/44'/714'/0'/0/1", "bnb1egswqkszzfc2uq78zjslc6u2uky4pw46x4rstd"),
@ -14,7 +12,7 @@ BINANCE_ADDRESS_TEST_VECTORS = [
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.binance @pytest.mark.binance
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client( @pytest.mark.setup_client(
mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin" mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin"
) )
@pytest.mark.parametrize("path, expected_address", BINANCE_ADDRESS_TEST_VECTORS) @pytest.mark.parametrize("path, expected_address", BINANCE_ADDRESS_TEST_VECTORS)

4
tests/device_tests/test_msg_binance_get_public_key.py
View File

@ -3,15 +3,13 @@ import pytest
from trezorlib import binance from trezorlib import binance
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
BINANCE_PATH = parse_path("m/44h/714h/0h/0/0") BINANCE_PATH = parse_path("m/44h/714h/0h/0/0")
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.binance @pytest.mark.binance
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client( @pytest.mark.setup_client(
mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin" mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin"
) )
def test_binance_get_public_key(client): def test_binance_get_public_key(client):

4
tests/device_tests/test_msg_binance_sign_tx.py
View File

@ -3,8 +3,6 @@ import pytest
from trezorlib import binance from trezorlib import binance
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
BINANCE_TEST_VECTORS = [ BINANCE_TEST_VECTORS = [
( # CANCEL ( # CANCEL
{ {
@ -89,7 +87,7 @@ BINANCE_TEST_VECTORS = [
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.binance @pytest.mark.binance
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client( @pytest.mark.setup_client(
mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin" mnemonic="offer caution gift cross surge pretty orange during eye soldier popular holiday mention east eight office fashion ill parrot vault rent devote earth cousin"
) )
@pytest.mark.parametrize("message, expected_response", BINANCE_TEST_VECTORS) @pytest.mark.parametrize("message, expected_response", BINANCE_TEST_VECTORS)

3
tests/device_tests/test_msg_cardano_get_address.py
View File

@ -20,7 +20,6 @@ from trezorlib.cardano import get_address
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .common import TrezorTest from .common import TrezorTest
from .conftest import setup_client
@pytest.mark.altcoin @pytest.mark.altcoin
@ -43,7 +42,7 @@ from .conftest import setup_client
), ),
], ],
) )
@setup_client(mnemonic=TrezorTest.mnemonic12) @pytest.mark.setup_client(mnemonic=TrezorTest.mnemonic12)
def test_cardano_get_address(client, path, expected_address): def test_cardano_get_address(client, path, expected_address):
# data from https://iancoleman.io/bip39/#english # data from https://iancoleman.io/bip39/#english

4
tests/device_tests/test_msg_cardano_get_address_shamir.py
View File

@ -19,8 +19,6 @@ import pytest
from trezorlib.cardano import get_address from trezorlib.cardano import get_address
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
SLIP39_MNEMONIC = [ SLIP39_MNEMONIC = [
"extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern", "extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern",
"extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse", "extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse",
@ -48,7 +46,7 @@ SLIP39_MNEMONIC = [
), ),
], ],
) )
@setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True) @pytest.mark.setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True)
def test_cardano_get_address(client, path, expected_address): def test_cardano_get_address(client, path, expected_address):
# enter passphrase # enter passphrase
assert client.features.passphrase_protection is True assert client.features.passphrase_protection is True

3
tests/device_tests/test_msg_cardano_get_public_key.py
View File

@ -19,13 +19,10 @@ import pytest
from trezorlib.cardano import get_public_key from trezorlib.cardano import get_public_key
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.cardano @pytest.mark.cardano
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client(mnemonic=" ".join(["all"] * 12))
@pytest.mark.parametrize( @pytest.mark.parametrize(
"path,public_key,chain_code", "path,public_key,chain_code",
[ [

4
tests/device_tests/test_msg_cardano_get_public_key_shamir.py
View File

@ -19,8 +19,6 @@ import pytest
from trezorlib.cardano import get_public_key from trezorlib.cardano import get_public_key
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
SLIP39_MNEMONIC = [ SLIP39_MNEMONIC = [
"extra extend academic bishop cricket bundle tofu goat apart victim " "extra extend academic bishop cricket bundle tofu goat apart victim "
"enlarge program behavior permit course armed jerky faint language modern", "enlarge program behavior permit course armed jerky faint language modern",
@ -34,7 +32,7 @@ SLIP39_MNEMONIC = [
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.cardano @pytest.mark.cardano
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True) @pytest.mark.setup_client(mnemonic=SLIP39_MNEMONIC, passphrase=True)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"path,public_key,chain_code", "path,public_key,chain_code",
[ [

4
tests/device_tests/test_msg_cardano_sign_transaction.py
View File

@ -19,8 +19,6 @@ import pytest
from trezorlib import cardano, messages from trezorlib import cardano, messages
from trezorlib.exceptions import TrezorFailure from trezorlib.exceptions import TrezorFailure
from .conftest import setup_client
PROTOCOL_MAGICS = {"mainnet": 764824073, "testnet": 1097911063} PROTOCOL_MAGICS = {"mainnet": 764824073, "testnet": 1097911063}
SAMPLE_INPUTS = [ SAMPLE_INPUTS = [
@ -173,7 +171,6 @@ INVALID_VECTORS = [
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.cardano @pytest.mark.cardano
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client()
@pytest.mark.parametrize( @pytest.mark.parametrize(
"protocol_magic,inputs,outputs,transactions,tx_hash,tx_body", VALID_VECTORS "protocol_magic,inputs,outputs,transactions,tx_hash,tx_body", VALID_VECTORS
) )
@ -213,7 +210,6 @@ def test_cardano_sign_tx(
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.cardano @pytest.mark.cardano
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client()
@pytest.mark.parametrize( @pytest.mark.parametrize(
"protocol_magic,inputs,outputs,transactions,expected_error_message", INVALID_VECTORS "protocol_magic,inputs,outputs,transactions,expected_error_message", INVALID_VECTORS
) )

4
tests/device_tests/test_msg_cardano_sign_tx_shamir.py
View File

@ -18,8 +18,6 @@ import pytest
from trezorlib import cardano, messages from trezorlib import cardano, messages
from .conftest import setup_client
SHARES_20_3of6 = [ SHARES_20_3of6 = [
"extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern", "extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern",
"extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse", "extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse",
@ -115,7 +113,7 @@ VALID_VECTORS = [
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.cardano @pytest.mark.cardano
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client(mnemonic=SHARES_20_3of6, passphrase=True) @pytest.mark.setup_client(mnemonic=SHARES_20_3of6, passphrase=True)
@pytest.mark.parametrize( @pytest.mark.parametrize(
"protocol_magic,inputs,outputs,transactions,tx_hash,tx_body", VALID_VECTORS "protocol_magic,inputs,outputs,transactions,tx_hash,tx_body", VALID_VECTORS
) )

5
tests/device_tests/test_msg_recoverydevice_shamir.py
View File

@ -92,6 +92,7 @@ def test_secret(client, shares, secret):
assert debug.read_mnemonic_secret().hex() == secret assert debug.read_mnemonic_secret().hex() == secret
@pytest.mark.setup_client(uninitialized=True)
def test_recover_with_pin_passphrase(client): def test_recover_with_pin_passphrase(client):
debug = client.debug debug = client.debug
@ -117,6 +118,7 @@ def test_recover_with_pin_passphrase(client):
assert client.features.passphrase_protection is True assert client.features.passphrase_protection is True
@pytest.mark.setup_client(uninitialized=True)
def test_abort(client): def test_abort(client):
debug = client.debug debug = client.debug
@ -136,6 +138,7 @@ def test_abort(client):
assert client.features.initialized is False assert client.features.initialized is False
@pytest.mark.setup_client(uninitialized=True)
def test_noabort(client): def test_noabort(client):
debug = client.debug debug = client.debug
@ -155,6 +158,7 @@ def test_noabort(client):
assert client.features.initialized is True assert client.features.initialized is True
@pytest.mark.setup_client(uninitialized=True)
@pytest.mark.parametrize("nth_word", range(3)) @pytest.mark.parametrize("nth_word", range(3))
def test_wrong_nth_word(client, nth_word): def test_wrong_nth_word(client, nth_word):
debug = client.debug debug = client.debug
@ -194,6 +198,7 @@ def test_wrong_nth_word(client, nth_word):
device.recover(client, pin_protection=False, label="label") device.recover(client, pin_protection=False, label="label")
@pytest.mark.setup_client(uninitialized=True)
def test_same_share(client): def test_same_share(client):
debug = client.debug debug = client.debug
first_share = SHARES_20_3of6[0].split(" ") first_share = SHARES_20_3of6[0].split(" ")

6
tests/device_tests/test_msg_recoverydevice_shamir_dryrun.py
View File

@ -3,8 +3,6 @@ import pytest
from trezorlib import device, messages from trezorlib import device, messages
from trezorlib.exceptions import TrezorFailure from trezorlib.exceptions import TrezorFailure
from .conftest import setup_client
pytestmark = pytest.mark.skip_t1 pytestmark = pytest.mark.skip_t1
SHARES_20_2of3 = [ SHARES_20_2of3 = [
@ -19,7 +17,7 @@ INVALID_SHARES_20_2of3 = [
] ]
@setup_client(mnemonic=SHARES_20_2of3[0:2], passphrase=True) @pytest.mark.setup_client(mnemonic=SHARES_20_2of3[0:2], passphrase=True)
def test_2of3_dryrun(client): def test_2of3_dryrun(client):
debug = client.debug debug = client.debug
@ -47,7 +45,7 @@ def test_2of3_dryrun(client):
) )
@setup_client(mnemonic=SHARES_20_2of3[0:2], passphrase=True) @pytest.mark.setup_client(mnemonic=SHARES_20_2of3[0:2], passphrase=True)
def test_2of3_invalid_seed_dryrun(client): def test_2of3_invalid_seed_dryrun(client):
debug = client.debug debug = client.debug

2
tests/device_tests/test_msg_signtx_capricoin.py
View File

@ -19,7 +19,6 @@ import pytest
from trezorlib import btc, messages from trezorlib import btc, messages
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from .conftest import setup_client
from .tx_cache import tx_cache from .tx_cache import tx_cache
TXHASH_3bf506 = bytes.fromhex( TXHASH_3bf506 = bytes.fromhex(
@ -34,7 +33,6 @@ TXHASH_f3a6e6 = bytes.fromhex(
@pytest.mark.altcoin @pytest.mark.altcoin
@pytest.mark.capricoin @pytest.mark.capricoin
@pytest.mark.skip_t1 # T1 support is not planned @pytest.mark.skip_t1 # T1 support is not planned
@setup_client()
def test_timestamp_included(client): def test_timestamp_included(client):
# tx: 3bf506c81ce84eda891679ddc797d162c17c60b15d6c0ac23be5e31369e7235f # tx: 3bf506c81ce84eda891679ddc797d162c17c60b15d6c0ac23be5e31369e7235f
# input 0: 0.01 CPC # input 0: 0.01 CPC

6
tests/device_tests/test_shamir_passphrase.py
View File

@ -19,10 +19,8 @@ import pytest
from trezorlib import btc from trezorlib import btc
from .conftest import setup_client
@pytest.mark.setup_client(
@setup_client(
mnemonic=( mnemonic=(
"extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern", "extra extend academic bishop cricket bundle tofu goat apart victim enlarge program behavior permit course armed jerky faint language modern",
"extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse", "extra extend academic acne away best indicate impact square oasis prospect painting voting guest either argue username racism enemy eclipse",
@ -43,7 +41,7 @@ def test_3of6_passphrase(client):
assert address == "18oZEMRWurCZW1FeK8sWYyXuWx2bFqEKyX" assert address == "18oZEMRWurCZW1FeK8sWYyXuWx2bFqEKyX"
@setup_client( @pytest.mark.setup_client(
mnemonic=( mnemonic=(
"hobo romp academic axis august founder knife legal recover alien expect emphasis loan kitchen involve teacher capture rebuild trial numb spider forward ladle lying voter typical security quantity hawk legs idle leaves gasoline", "hobo romp academic axis august founder knife legal recover alien expect emphasis loan kitchen involve teacher capture rebuild trial numb spider forward ladle lying voter typical security quantity hawk legs idle leaves gasoline",
"hobo romp academic agency ancestor industry argue sister scene midst graduate profile numb paid headset airport daisy flame express scene usual welcome quick silent downtown oral critical step remove says rhythm venture aunt", "hobo romp academic agency ancestor industry argue sister scene midst graduate profile numb paid headset airport daisy flame express scene usual welcome quick silent downtown oral critical step remove says rhythm venture aunt",

1
tests/device_tests/test_shamir_reset_recovery.py
View File

@ -6,6 +6,7 @@ from trezorlib.tools import parse_path
@pytest.mark.skip_t1 @pytest.mark.skip_t1
@pytest.mark.setup_client(uninitialized=True)
def test_reset_recovery(client): def test_reset_recovery(client):
mnemonics = reset(client) mnemonics = reset(client)
address_before = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0")) address_before = btc.get_address(client, "Bitcoin", parse_path("44'/0'/0'/0/0"))