File size: 652 Bytes
95f8c4b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 | from aistudio_api.infrastructure.gateway.stream_parser import IncrementalJSONStreamParser
def test_stream_parser_extracts_chunk_from_fragmented_input():
parser = IncrementalJSONStreamParser()
pieces = ['[[[null,"he', 'llo"]]]']
chunks = []
for piece in pieces:
chunks.extend(list(parser.feed(piece)))
assert chunks == [[None, "hello"]]
def test_stream_parser_handles_fragmented_xssi_preamble():
parser = IncrementalJSONStreamParser()
pieces = [")]", "}'\n", '[[[null,"hello"]]]']
chunks = []
for piece in pieces:
chunks.extend(list(parser.feed(piece)))
assert chunks == [[None, "hello"]]
|