mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-23 16:08:17 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
|
from common import *
|
||
|
|
|
||
|
|
import apps.common.writers as writers
|
||
|
|
|
||
|
|
|
||
|
|
class TestSeed(unittest.TestCase):
|
||
|
|
def test_write_uint8(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint8(buf, 0x12)
|
||
|
|
self.assertEqual(buf, b"\x12")
|
||
|
|
|
||
|
|
def test_write_uint16_le(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint16_le(buf, 0x1234)
|
||
|
|
self.assertEqual(buf, b"\x34\x12")
|
||
|
|
|
||
|
|
def test_write_uint16_le_overflow(self):
|
||
|
|
buf = bytearray()
|
||
|
|
with self.assertRaises(AssertionError):
|
||
|
|
writers.write_uint16_le(buf, 0x12345678)
|
||
|
|
|
||
|
|
def test_write_uint32_le(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint32_le(buf, 0x12345678)
|
||
|
|
self.assertEqual(buf, b"\x78\x56\x34\x12")
|
||
|
|
|
||
|
|
def test_write_uint64_le(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint64_le(buf, 0x1234567890abcdef)
|
||
|
|
self.assertEqual(buf, b"\xef\xcd\xab\x90\x78\x56\x34\x12")
|
||
|
|
|
||
|
|
def test_write_uint32_be(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint32_be(buf, 0x12345678)
|
||
|
|
self.assertEqual(buf, b"\x12\x34\x56\x78")
|
||
|
|
|
||
|
|
def test_write_uint64_be(self):
|
||
|
|
buf = bytearray()
|
||
|
|
writers.write_uint64_be(buf, 0x1234567890abcdef)
|
||
|
|
self.assertEqual(buf, b"\x12\x34\x56\x78\x90\xab\xcd\xef")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|