def test_unreader_read_when_size_is_none():
unreader = Unreader()
unreader.chunk = mock.MagicMock(side_effect=[b'qwerty', b'123456', b''])
assert unreader.read(size=None) == b'qwerty'
assert unreader.read(size=None) == b'123456'
assert unreader.read(size=None) == b''
def test_unreader_raises_excpetion_on_invalid_size():
unreader = Unreader()
with pytest.raises(TypeError):
unreader.read(size='foobar')
with pytest.raises(TypeError):
unreader.read(size=3.14)
with pytest.raises(TypeError):
unreader.read(size=[])
def test_unreader_read_with_nonzero_size():
unreader = Unreader()
unreader.chunk = mock.MagicMock(
side_effect=[b'qwerty', b'asdfgh', b'zxcvbn', b'123456', b'', b''])
assert unreader.read(size=5) == b'qwert'
assert unreader.read(size=5) == b'yasdf'
assert unreader.read(size=5) == b'ghzxc'
assert unreader.read(size=5) == b'vbn12'
assert unreader.read(size=5) == b'3456'
assert unreader.read(size=5) == b''
def test_unreader_read_zero_size():
unreader = Unreader()
unreader.chunk = mock.MagicMock(side_effect=[b'qwerty', b'asdfgh'])
assert unreader.read(size=0) == b''
def test_unreader_unread():
unreader = Unreader()
unreader.unread(b'hi there')
assert b'hi there' in unreader.read()