mirror of
https://github.com/trezor/trezor-firmware.git
synced 2024-11-30 11:28:21 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
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
|