1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-07-02 04:42:33 +00:00
trezor-firmware/core/tests/test_trezor.io.sdcard.py
obrusvit a429da5b0e feat(core/sdbackup): improve handling of mocked SD
By default, unit tests and Emulator starts without virtual SD card. A
card must be explicitly inserted in test setup.
2024-01-20 18:13:48 +01:00

53 lines
1.4 KiB
Python

from common import *
from trezor import io
class TestTrezorIoSdcard(unittest.TestCase):
def setUp(self):
io.sdcard_switcher.insert(1)
def tearDown(self):
io.sdcard_switcher.eject()
def test_start(self):
self.assertTrue(io.sdcard.is_present())
def test_power(self):
x = bytearray(8 * 512)
self.assertEqual(io.sdcard.capacity(), 0)
with self.assertRaises(OSError):
io.sdcard.read(0, x)
io.sdcard.power_on()
self.assertTrue(io.sdcard.capacity() > 0)
io.sdcard.read(0, x)
io.sdcard.power_off()
self.assertEqual(io.sdcard.capacity(), 0)
with self.assertRaises(OSError):
io.sdcard.read(0, x)
def test_read(self):
x = bytearray(8 * 512)
io.sdcard.power_on()
io.sdcard.read(0, x)
io.sdcard.power_off()
with self.assertRaises(OSError):
io.sdcard.read(0, x)
def test_read_write(self):
r = bytearray(8 * 512)
w0 = bytearray(b"0" * (8 * 512))
w1 = bytearray(b"1" * (8 * 512))
io.sdcard.power_on()
io.sdcard.write(0, w0)
io.sdcard.read(0, r)
self.assertEqual(r, w0)
io.sdcard.write(0, w1)
io.sdcard.read(0, r)
self.assertEqual(r, w1)
io.sdcard.power_off()
if __name__ == "__main__":
unittest.main()