def test_streamreader_z_string():
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.get_z_string() == ''
# Empty string
stream = StringStream(b'\x00')
reader = StreamReader(stream, False)
assert reader.get_z_string() == ''
# String of length 1
stream = StringStream(b'A\x00')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'A'
# String with excess data
stream = StringStream(b'ABC\x00AB')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'ABC'
# EOF before end of string
stream = StringStream(b'ABC')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'ABC'
def test_streamreader_fixed_string():
# Zero-length string
stream = StringStream(b'ABC')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(0) == ''
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == ''
# Empty string
stream = StringStream(b'\x00')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == ''
# String of length 1
stream = StringStream(b'A')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == 'A'
# String of length 1, excess data
stream = StringStream(b'ABC\x00')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == 'A'
# EOF before end of string
stream = StringStream(b'AB')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(4) == 'AB'
def test_streamreader_z_string():
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.get_z_string() == ''
# Empty string
stream = StringStream(b'\x00')
reader = StreamReader(stream, False)
assert reader.get_z_string() == ''
# String of length 1
stream = StringStream(b'A\x00')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'A'
# String with excess data
stream = StringStream(b'ABC\x00AB')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'ABC'
# EOF before end of string
stream = StringStream(b'ABC')
reader = StreamReader(stream, False)
assert reader.get_z_string() == 'ABC'
def test_streamreader_string():
# Empty string
stream = StringStream(b'\x00\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == ''
# String size but no string contents
stream = StringStream(b'\x01\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == ''
# String of length 1
stream = StringStream(b'\x01\x00A')
reader = StreamReader(stream, False)
assert reader.get_string() == 'A'
# String with excess data
stream = StringStream(b'\x01\x00AB')
reader = StreamReader(stream, False)
assert reader.get_string() == 'A'
# EOF before end of string
stream = StringStream(b'\x03\x00AB')
reader = StreamReader(stream, False)
assert reader.get_string() == 'AB'
# Preserves null bytes
stream = StringStream(b'\x02\x00\x00\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == '\x00\x00'
def test_streamreader_extract_bytes():
# Empty bytes
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.extract_bytes(10) == b''
# Small bytes object, small reads
stream = StringStream(b'abcd')
reader = StreamReader(stream, False)
assert reader.extract_bytes(2) == b'ab'
assert reader.extract_bytes(2) == b'cd'
assert reader.extract_bytes(2) == b''
# Embedded null bytes
stream = StringStream(b'a\x00b\x00c')
reader = StreamReader(stream, False)
assert reader.extract_bytes(5) == b'a\x00b\x00c'
# Not enough data in stream to fill buffer
stream = StringStream(b'abcdefghijklmnop')
reader = StreamReader(stream, False)
assert reader.extract_bytes(10) == b'abcdefghij'
assert reader.extract_bytes(10) == b'klmnop'
assert reader.extract_bytes(10) == b''
# Read of 0 bytes
stream = StringStream(b'abcd')
reader = StreamReader(stream, False)
assert reader.extract_bytes(0) == b''
assert reader.extract_bytes(0) == b''
# Very large read (8 MiB buffer allocation)
assert reader.extract_bytes(8 * 1024 * 1024) == b'abcd'
def test_streamreader_readline():
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.readline() == b''
assert reader.readline() == b''
# Single line without newline
stream = StringStream(b'A')
reader = StreamReader(stream, False)
assert reader.readline() == b'A'
assert reader.readline() == b''
# Single newline
stream = StringStream(b'\n')
reader = StreamReader(stream, False)
assert reader.readline() == b'\n'
assert reader.readline() == b''
# Line with text followed by empty line
stream = StringStream(b'A\n\n')
reader = StreamReader(stream, False)
assert reader.readline() == b'A\n'
assert reader.readline() == b'\n'
assert reader.readline() == b''
# Preserve null byte
stream = StringStream(b'\x00\x00')
reader = StreamReader(stream, False)
assert reader.readline() == b'\x00\x00'
def test_streamreader_fixed_string():
# Zero-length string
stream = StringStream(b'ABC')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(0) == ''
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == ''
# Empty string
stream = StringStream(b'\x00')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == ''
# String of length 1
stream = StringStream(b'A')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == 'A'
# String of length 1, excess data
stream = StringStream(b'ABC\x00')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(1) == 'A'
# EOF before end of string
stream = StringStream(b'AB')
reader = StreamReader(stream, False)
assert reader.get_fixed_string(4) == 'AB'
def test_streamreader_string():
# Empty string
stream = StringStream(b'\x00\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == ''
# String size but no string contents
stream = StringStream(b'\x01\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == ''
# String of length 1
stream = StringStream(b'\x01\x00A')
reader = StreamReader(stream, False)
assert reader.get_string() == 'A'
# String with excess data
stream = StringStream(b'\x01\x00AB')
reader = StreamReader(stream, False)
assert reader.get_string() == 'A'
# EOF before end of string
stream = StringStream(b'\x03\x00AB')
reader = StreamReader(stream, False)
assert reader.get_string() == 'AB'
# Preserves null bytes
stream = StringStream(b'\x02\x00\x00\x00')
reader = StreamReader(stream, False)
assert reader.get_string() == '\x00\x00'
def test_streamreader_readline():
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.readline() == b''
assert reader.readline() == b''
# Single line without newline
stream = StringStream(b'A')
reader = StreamReader(stream, False)
assert reader.readline() == b'A'
assert reader.readline() == b''
# Single newline
stream = StringStream(b'\n')
reader = StreamReader(stream, False)
assert reader.readline() == b'\n'
assert reader.readline() == b''
# Line with text followed by empty line
stream = StringStream(b'A\n\n')
reader = StreamReader(stream, False)
assert reader.readline() == b'A\n'
assert reader.readline() == b'\n'
assert reader.readline() == b''
# Preserve null byte
stream = StringStream(b'\x00\x00')
reader = StreamReader(stream, False)
assert reader.readline() == b'\x00\x00'
def test_streamreader_readlines():
# Empty stream
stream = StringStream(b'')
reader = StreamReader(stream, False)
assert reader.readlines() == []
# Single line without newline
stream = StringStream(b'A')
reader = StreamReader(stream, False)
assert reader.readlines() == [b'A']
assert reader.readlines() == []
# Single newline
stream = StringStream(b'\n')
reader = StreamReader(stream, False)
assert reader.readlines() == [b'\n']
assert reader.readlines() == []
# Line with text followed by empty line
stream = StringStream(b'A\n\n')
reader = StreamReader(stream, False)
assert reader.readlines() == [b'A\n', b'\n']
assert reader.readlines() == []
# Multiple lines
stream = StringStream(b'A\nB\nC')
reader = StreamReader(stream, False)
assert reader.readlines() == [b'A\n', b'B\n', b'C']
assert reader.readlines() == []
# Preserve null byte
stream = StringStream(b'\x00\x00')
reader = StreamReader(stream, False)
assert reader.readlines() == [b'\x00\x00']