Telethon/generator/tests/tl_iterator_test.py
2023-07-03 19:19:20 +02:00

29 lines
779 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