1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-08-03 04:18:17 +00:00
trezor-firmware/storage/tests/tests/common.py
2023-11-15 10:24:54 +01:00

26 lines
639 B
Python

from c.storage import Storage as StorageC
from python.src import prng
from python.src.storage import Storage as StoragePy
test_uid = b"\x67\xce\x6a\xe8\xf7\x9b\x73\x96\x83\x88\x21\x5e"
def init(
unlock: bool = False, reseed: int = 0, uid: int = test_uid, flash_byte_access=True
) -> (StorageC, StoragePy):
sc = StorageC(flash_byte_access)
sp = StoragePy(flash_byte_access)
sc.lib.random_reseed(reseed)
prng.random_reseed(reseed)
for s in (sc, sp):
s.init(uid)
if unlock:
assert s.unlock("")
return sc, sp
def memory_equals(sc, sp) -> bool:
return sc._dump() == sp._dump()