arno
/
trezor-firmware
mirror of https://github.com/trezor/trezor-firmware.git
1
0
Fork 0
Code Issues Releases Wiki Activity

test: Use long PIN values in device tests.

Browse Source
pull/1546/head
Andrew Kozlik 4 years ago committed by Andrew Kozlik
parent 885b6f7214
commit 602761694f
4 changed files with 28 additions and 21 deletions
Show Stats Download Patch File Download Diff File

9
tests/device_tests/test_msg_change_wipe_code_t1.py
Unescape View File

@ -17,12 +17,15 @@
import pytest
from trezorlib import device, exceptions, messages
from trezorlib.client import MAX_PIN_LENGTH
PinType = messages.PinMatrixRequestType
PIN4 = "1234"
WIPE_CODE4 = "4321"
WIPE_CODE6 = "456789"
WIPE_CODE_MAX = "".join(chr((i % 9) + ord("1")) for i in range(MAX_PIN_LENGTH))
WIPE_CODE_TOO_LONG = WIPE_CODE_MAX + "1"
pytestmark = pytest.mark.skip_t2
@ -75,14 +78,14 @@ def test_set_remove_wipe_code(client):
assert client.features.wipe_code_protection is None
# Test set wipe code.
_set_wipe_code(client, PIN4, WIPE_CODE4)
_set_wipe_code(client, PIN4, WIPE_CODE_MAX)
# Check that there's wipe code protection now.
client.init_device()
assert client.features.wipe_code_protection is True
# Check that the wipe code is correct.
_check_wipe_code(client, PIN4, WIPE_CODE4)
_check_wipe_code(client, PIN4, WIPE_CODE_MAX)
# Test change wipe code.
_set_wipe_code(client, PIN4, WIPE_CODE6)
@ -177,7 +180,7 @@ def test_set_pin_to_wipe_code(client):
assert isinstance(resp, messages.Address)
@pytest.mark.parametrize("invalid_wipe_code", ("1204", "", "1234567891"))
@pytest.mark.parametrize("invalid_wipe_code", ("1204", "", WIPE_CODE_TOO_LONG))
def test_set_wipe_code_invalid(client, invalid_wipe_code):
# Let's set the wipe code
ret = client.call_raw(messages.ChangeWipeCode())

7
tests/device_tests/test_msg_change_wipe_code_t2.py
Unescape View File

@ -17,12 +17,13 @@
import pytest
from trezorlib import btc, device, messages
from trezorlib.client import PASSPHRASE_TEST_PATH
from trezorlib.client import MAX_PIN_LENGTH, PASSPHRASE_TEST_PATH
from trezorlib.exceptions import Cancelled, TrezorFailure
PIN4 = "1234"
WIPE_CODE4 = "4321"
WIPE_CODE6 = "456789"
WIPE_CODE_MAX = "".join(chr((i % 10) + ord("0")) for i in range(MAX_PIN_LENGTH))
pytestmark = pytest.mark.skip_t1
@ -61,12 +62,12 @@ def test_set_remove_wipe_code(client):
client.set_expected_responses(
[messages.ButtonRequest()] * 5 + [messages.Success, messages.Features]
)
client.use_pin_sequence([PIN4, WIPE_CODE4, WIPE_CODE4])
client.use_pin_sequence([PIN4, WIPE_CODE_MAX, WIPE_CODE_MAX])
device.change_wipe_code(client)
client.init_device()
assert client.features.wipe_code_protection is True
_check_wipe_code(client, PIN4, WIPE_CODE4)
_check_wipe_code(client, PIN4, WIPE_CODE_MAX)
# Test change wipe code.
with client:

18
tests/device_tests/test_msg_changepin.py
Unescape View File

@ -17,13 +17,15 @@
import pytest
from trezorlib import device, messages
from trezorlib.client import MAX_PIN_LENGTH
from trezorlib.exceptions import TrezorFailure
from ..common import get_test_address
PIN4 = "1234"
PIN6 = "789456"
PIN_MAX = "".join(chr((i % 9) + ord("1")) for i in range(MAX_PIN_LENGTH))
PIN_TOO_LONG = PIN_MAX + "1"
pytestmark = pytest.mark.skip_t2
@ -51,7 +53,7 @@ def test_set_pin(client):
# Let's set new PIN
with client:
client.use_pin_sequence([PIN6, PIN6])
client.use_pin_sequence([PIN_MAX, PIN_MAX])
client.set_expected_responses(
[
messages.ButtonRequest(code=messages.ButtonRequestType.ProtectCall),
@ -66,7 +68,7 @@ def test_set_pin(client):
# Check that there's PIN protection now
assert client.features.pin_protection is True
# Check that the PIN is correct
_check_pin(client, PIN6)
_check_pin(client, PIN_MAX)
@pytest.mark.setup_client(pin=PIN4)
@ -77,7 +79,7 @@ def test_change_pin(client):
# Let's change PIN
with client:
client.use_pin_sequence([PIN4, PIN6, PIN6])
client.use_pin_sequence([PIN4, PIN_MAX, PIN_MAX])
client.set_expected_responses(
[
messages.ButtonRequest(code=messages.ButtonRequestType.ProtectCall),
@ -93,7 +95,7 @@ def test_change_pin(client):
# Check that there's still PIN protection now
assert client.features.pin_protection is True
# Check that the PIN is correct
_check_pin(client, PIN6)
_check_pin(client, PIN_MAX)
@pytest.mark.setup_client(pin=PIN4)
@ -128,7 +130,7 @@ def test_set_mismatch(client):
# Let's set new PIN
with pytest.raises(TrezorFailure, match="PIN mismatch"), client:
# use different PINs for first and second attempt. This will fail.
client.use_pin_sequence([PIN4, PIN6])
client.use_pin_sequence([PIN4, PIN_MAX])
client.set_expected_responses(
[
messages.ButtonRequest(code=messages.ButtonRequestType.ProtectCall),
@ -169,7 +171,7 @@ def test_change_mismatch(client):
_check_pin(client, PIN4)
@pytest.mark.parametrize("invalid_pin", ("1204", "", "1234567891"))
@pytest.mark.parametrize("invalid_pin", ("1204", "", PIN_TOO_LONG))
def test_set_invalid(client, invalid_pin):
assert client.features.pin_protection is False
@ -194,7 +196,7 @@ def test_set_invalid(client, invalid_pin):
_check_no_pin(client)
@pytest.mark.parametrize("invalid_pin", ("1204", "", "1234567891"))
@pytest.mark.parametrize("invalid_pin", ("1204", "", PIN_TOO_LONG))
@pytest.mark.setup_client(pin=PIN4)
def test_enter_invalid(client, invalid_pin):
assert client.features.pin_protection is True

15
tests/device_tests/test_msg_changepin_t2.py
Unescape View File

@ -17,11 +17,12 @@
import pytest
from trezorlib import btc, device, messages
from trezorlib.client import PASSPHRASE_TEST_PATH
from trezorlib.client import MAX_PIN_LENGTH, PASSPHRASE_TEST_PATH
from trezorlib.exceptions import Cancelled
PIN4 = "1234"
PIN6 = "789456"
PIN60 = "789456" * 10
PIN_MAX = "".join(chr((i % 10) + ord("0")) for i in range(MAX_PIN_LENGTH))
pytestmark = pytest.mark.skip_t1
@ -54,7 +55,7 @@ def test_set_pin(client):
# Let's set new PIN
with client:
client.use_pin_sequence([PIN6, PIN6])
client.use_pin_sequence([PIN_MAX, PIN_MAX])
client.set_expected_responses(
[messages.ButtonRequest] * 4 + [messages.Success, messages.Features]
)
@ -62,7 +63,7 @@ def test_set_pin(client):
client.init_device()
assert client.features.pin_protection is True
_check_pin(client, PIN6)
_check_pin(client, PIN_MAX)
@pytest.mark.setup_client(pin=PIN4)
@ -74,7 +75,7 @@ def test_change_pin(client):
# Let's change PIN
with client:
client.use_pin_sequence([PIN4, PIN6, PIN6])
client.use_pin_sequence([PIN4, PIN_MAX, PIN_MAX])
client.set_expected_responses(
[messages.ButtonRequest] * 5 + [messages.Success, messages.Features]
)
@ -84,7 +85,7 @@ def test_change_pin(client):
client.init_device()
assert client.features.pin_protection is True
# Check that the PIN is correct
_check_pin(client, PIN6)
_check_pin(client, PIN_MAX)
@pytest.mark.setup_client(pin=PIN4)
@ -121,7 +122,7 @@ def test_set_failed(client):
yield # enter new pin
client.debug.input(PIN4)
yield # enter new pin again (but different)
client.debug.input(PIN6)
client.debug.input(PIN60)
# failed retry
yield # enter new pin

Write Preview
Loading…
Cancel
Save