async def test_one_chunk_empty_lines():
async def iter_chunked(n: int):
yield b'\n\nhello\n\nworld\n\n'
content = asynctest.Mock(iter_chunked=iter_chunked)
lines = []
async for line in _iter_jsonlines(content):
lines.append(line)
assert lines == [b'hello', b'world']
async def test_one_chunk_one_line():
async def iter_chunked(n: int):
yield b'hello'
content = asynctest.Mock(iter_chunked=iter_chunked)
lines = []
async for line in _iter_jsonlines(content):
lines.append(line)
assert lines == [b'hello']
async def test_empty_chunk():
async def iter_chunked(n: int):
yield b''
content = asynctest.Mock(iter_chunked=iter_chunked)
lines = []
async for line in _iter_jsonlines(content):
lines.append(line)
assert lines == []
async def test_empty_content():
async def iter_chunked(n: int):
if False: # to make this function a generator
yield b''
content = asynctest.Mock(iter_chunked=iter_chunked)
lines = []
async for line in _iter_jsonlines(content):
lines.append(line)
assert lines == []
async def test_few_chunks_split():
async def iter_chunked(n: int):
yield b'\n\nhell'
yield b'o\n\nwor'
yield b'ld\n\n'
content = asynctest.Mock(iter_chunked=iter_chunked)
lines = []
async for line in _iter_jsonlines(content):
lines.append(line)
assert lines == [b'hello', b'world']