2023-07-05 23:35:06 +03:00
|
|
|
from pytest import mark
|
2024-10-06 21:05:11 +03:00
|
|
|
|
2023-07-05 23:35:06 +03:00
|
|
|
from telethon._impl.tl.core import serialize_bytes_to
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@mark.parametrize(
|
|
|
|
|
("string", "prefix", "suffix"),
|
|
|
|
|
[
|
|
|
|
|
("", b"\00", b"\00\x00\x00"),
|
|
|
|
|
("Hi", b"\02", b"\00"),
|
|
|
|
|
("Hi!", b"\03", b""),
|
|
|
|
|
("Hello", b"\05", b"\00\x00"),
|
|
|
|
|
("Hello, world!", b"\x0d", b"\00\x00"),
|
|
|
|
|
(
|
|
|
|
|
"This is a very long string, and it has to be longer than 253 \
|
|
|
|
|
characters, which are quite a few but we can make it! Although, \
|
|
|
|
|
it is quite challenging. The quick brown fox jumps over the lazy \
|
|
|
|
|
fox. There is still some more text we need to type. Oh, this \
|
|
|
|
|
sentence made it past!",
|
|
|
|
|
b"\xfe\x11\x01\x00",
|
|
|
|
|
b"\x00\x00\x00",
|
|
|
|
|
),
|
|
|
|
|
],
|
|
|
|
|
)
|
|
|
|
|
def test_string(string: str, prefix: bytes, suffix: bytes) -> None:
|
|
|
|
|
expected = prefix + string.encode("ascii") + suffix
|
|
|
|
|
buffer = bytearray()
|
|
|
|
|
serialize_bytes_to(buffer, string.encode("ascii"))
|
|
|
|
|
assert bytes(buffer) == expected
|
