1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/python/tests/test_rlp.py

34 lines
1.1 KiB
Python
Raw Normal View History

import pytest
from trezorlib import _rlp
VECTORS = ( # data, expected
(b"\x10", b"\x10"),
(b"dog", b"\x83dog"),
(b"A" * 55, b"\xb7" + b"A" * 55),
(b"A" * 56, b"\xb8\x38" + b"A" * 56),
(b"A" * 1024, b"\xb9\x04\x00" + b"A" * 1024),
([b"dog", b"cat", [b"spy"]], b"\xcd\x83dog\x83cat\xc4\x83spy"),
([b"A" * 1024], b"\xf9\x04\x03\xb9\x04\x00" + b"A" * 1024),
([], b"\xc0"),
([b"A"] * 55, b"\xf7" + b"A" * 55),
([b"A"] * 56, b"\xf8\x38" + b"A" * 56),
([b"A"] * 1024, b"\xf9\x04\x00" + b"A" * 1024),
([b"dog"] * 1024, b"\xf9\x10\x00" + b"\x83dog" * 1024),
(b"", b"\x80"),
(1, b"\x01"),
(0x7F, b"\x7f"),
(0x80, b"\x81\x80"),
(0x1_0000_0001, b"\x85\x01\x00\x00\x00\x01"),
(2 ** (54 * 8), b"\xb7\x01" + b"\x00" * 54),
(2 ** (55 * 8), b"\xb8\x38\x01" + b"\x00" * 55),
([0x1234, 0x5678], b"\xc6\x82\x12\x34\x82\x56\x78"),
)
@pytest.mark.parametrize("data, expected", VECTORS)
def test_encode(data: "_rlp.RLPItem", expected: bytes):
actual = _rlp.encode(data)
assert len(actual) == len(expected)
assert actual == expected