mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-12-01 11:58:28 +00:00
67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import pytest
|
|
|
|
from python.src import consts
|
|
|
|
from . import common
|
|
|
|
|
|
def test_init_pin():
|
|
sc, sp = common.init(uid=b"\x00\x00\x00\x00\x00\x00")
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
sc, sp = common.init(uid=b"\x22\x00\xDD\x00\x00\xBE")
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
|
|
def test_change_pin():
|
|
sc, sp = common.init(unlock=True)
|
|
for s in (sc, sp):
|
|
assert s.change_pin(1, 2221)
|
|
# invalid PIN
|
|
assert not s.change_pin(99991, 1)
|
|
assert s.unlock(2221)
|
|
assert s.change_pin(2221, 999991)
|
|
assert s.change_pin(999991, 991)
|
|
assert s.unlock(991)
|
|
assert not s.unlock(99991)
|
|
|
|
assert common.memory_equals(sc, sp)
|
|
|
|
|
|
def test_has_pin():
|
|
sc, sp = common.init()
|
|
for s in (sc, sp):
|
|
assert not s.has_pin()
|
|
assert s.unlock(1)
|
|
assert not s.has_pin()
|
|
assert s.change_pin(1, 221)
|
|
assert s.has_pin()
|
|
assert s.change_pin(221, 1)
|
|
assert not s.has_pin()
|
|
|
|
|
|
def test_wipe_after_max_pin():
|
|
sc, sp = common.init(unlock=True)
|
|
for s in (sc, sp):
|
|
assert s.change_pin(1, 2221)
|
|
assert s.unlock(2221)
|
|
s.set(0x0202, b"Hello")
|
|
|
|
# try an invalid PIN MAX - 1 times
|
|
for i in range(consts.PIN_MAX_TRIES - 1):
|
|
assert not s.unlock(99991)
|
|
# this should pass
|
|
assert s.unlock(2221)
|
|
assert s.get(0x0202) == b"Hello"
|
|
|
|
# try an invalid PIN MAX times, the storage should get wiped
|
|
for i in range(consts.PIN_MAX_TRIES):
|
|
assert not s.unlock(99991)
|
|
assert i == consts.PIN_MAX_TRIES - 1
|
|
# this should return False and raise an exception, the storage is wiped
|
|
assert not s.unlock(2221)
|
|
with pytest.raises(RuntimeError):
|
|
assert s.get(0x0202) == b"Hello"
|
|
|
|
assert common.memory_equals(sc, sp)
|