1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-16 11:28:14 +00:00
trezor-firmware/storage/tests/tests/common.py
2024-02-20 09:37:28 +01:00

26 lines
627 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(
norcow_class, unlock: bool = False, reseed: int = 0, uid: int = test_uid
) -> (StorageC, StoragePy):
sp = StoragePy(norcow_class)
sc = StorageC(sp.nc.get_lib_name())
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()