arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-02-10 14:42:44 +00:00
Code Issues Releases Wiki Activity

tests: add wipe test for sd_protect

Browse Source
This commit is contained in:
matejcik 2019-10-21 12:12:03 +02:00
parent ccd1940ea8
commit 6a1a94ac6a
1 changed files with 26 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tests/device_tests/test_msg_sd_protect.py
View File

@ -16,10 +16,12 @@
import pytest import pytest
from trezorlib import device from trezorlib import debuglink, device
from trezorlib.exceptions import TrezorFailure from trezorlib.exceptions import TrezorFailure
from trezorlib.messages import SdProtectOperationType as Op from trezorlib.messages import SdProtectOperationType as Op
from ..common import MNEMONIC12
pytestmark = [pytest.mark.skip_t1, pytest.mark.sd_card] pytestmark = [pytest.mark.skip_t1, pytest.mark.sd_card]
@ -61,3 +63,26 @@ def test_refresh(client):
with pytest.raises(TrezorFailure): with pytest.raises(TrezorFailure):
device.sd_protect(client, Op.REFRESH) device.sd_protect(client, Op.REFRESH)
assert client.features.sd_protection is False assert client.features.sd_protection is False
def test_wipe(client):
# Enable SD protection
device.sd_protect(client, Op.ENABLE)
assert client.features.sd_protection is True
# Wipe device (this wipes internal storage)
device.wipe(client)
assert client.features.sd_protection is False
# Restore device to working status
debuglink.load_device_by_mnemonic(
client, mnemonic=MNEMONIC12, pin=None, passphrase_protection=False, label="test"
)
assert client.features.sd_protection is False
# Enable SD protection
device.sd_protect(client, Op.ENABLE)
assert client.features.sd_protection is True
# Refresh SD protection
device.sd_protect(client, Op.REFRESH)