trezor-firmware/core/tests/test_trezor.io.sdcard.py

from common import *

from trezor import io


class TestTrezorIoSdcard(unittest.TestCase):

    def test_start(self):
        sd = io.SDCard()
        self.assertTrue(sd.present())

    def test_power(self):
        sd = io.SDCard()
        x = bytearray(8 * 512)
        self.assertEqual(sd.capacity(), 0)
        with self.assertRaises(OSError):
            sd.read(0, x)
        sd.power(True)
        self.assertTrue(sd.capacity() > 0)
        sd.read(0, x)
        sd.power(False)
        self.assertEqual(sd.capacity(), 0)
        with self.assertRaises(OSError):
            sd.read(0, x)

    def test_read(self):
        sd = io.SDCard()
        x = bytearray(8 * 512)
        sd.power(True)
        sd.read(0, x)
        sd.power(False)
        with self.assertRaises(OSError):
            sd.read(0, x)

    def test_read_write(self):
        sd = io.SDCard()
        r = bytearray(8 * 512)
        w0 = bytearray(b'0' * (8 * 512))
        w1 = bytearray(b'1' * (8 * 512))
        sd.power(True)
        sd.write(0, w0)
        sd.read(0, r)
        self.assertEqual(r, w0)
        sd.write(0, w1)
        sd.read(0, r)
        self.assertEqual(r, w1)
        sd.power(False)


if __name__ == '__main__':
    unittest.main()