Telethon/generator/tests/tl_iterator_test.py
2024-10-06 20:05:11 +02:00

30 lines
780 B
Python

from pytest import raises
from telethon_generator.tl_parser import FunctionDef, TypeDef, parse_tl_file
def test_parse_bad_separator() -> None:
with raises(ValueError) as e:
for _ in parse_tl_file("---foo---"):
pass
e.match("bad separator")
def test_parse_file() -> None:
items = list(
parse_tl_file(
"""
// leading; comment
first#1 = t; // inline comment
---functions---
second and bad;
third#3 = t;
// trailing comment
"""
)
)
assert len(items) == 3
assert isinstance(items[0], TypeDef) and items[0].id == 1
assert isinstance(items[1], ValueError)
assert isinstance(items[2], FunctionDef) and items[2].id == 3