async def test_empty_body(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.feed_eof()
assert buf.at_eof()
async def test_empty_body(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.feed_eof()
assert buf.at_eof()
def function751(self):
var711 = aiohttp.FlowControlDataQueue(self.attribute1675)
var3846 = DeflateBuffer(var711, 'deflate')
var3846.zlib = mock.Mock()
var3846.zlib.flush.return_value = b'line'
var3846.zlib.eof = False
self.assertRaises(http_exceptions.ContentEncodingError,
var3846.feed_eof)
def function660(self):
var3553 = aiohttp.FlowControlDataQueue(self.attribute1675)
var119 = DeflateBuffer(var3553, 'deflate')
var119.zlib = mock.Mock()
var119.zlib.decompress.return_value = b'line'
var119.feed_data(b'data', 4)
self.assertEqual([b'line'],
list((d for (var719, var2543) in var3553._buffer)))
def function641(self):
var1278 = aiohttp.FlowControlDataQueue(self.attribute1675)
var1584 = DeflateBuffer(var1278, 'deflate')
var2970 = ValueError()
var1584.zlib = mock.Mock()
var1584.zlib.decompress.side_effect = var2970
self.assertRaises(http_exceptions.ContentEncodingError,
var1584.feed_data, b'data', 4)
def test_feed_eof_err(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.zlib = mock.Mock()
dbuf.zlib.flush.return_value = b'line'
dbuf.zlib.eof = False
self.assertRaises(http_exceptions.ContentEncodingError, dbuf.feed_eof)
def test_feed_data(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.return_value = b'line'
dbuf.feed_data(b'data', 4)
assert [b'line'] == list(d for d, _ in buf._buffer)
def test_feed_data(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.return_value = b'line'
dbuf.feed_data(b'data', 4)
assert [b'line'] == list(d for d, _ in buf._buffer)
def function946(self):
var3443 = aiohttp.FlowControlDataQueue(self.attribute1675)
var4336 = DeflateBuffer(var3443, 'deflate')
var4336.zlib = mock.Mock()
var4336.zlib.flush.return_value = b'line'
var4336.feed_eof()
self.assertEqual([b'line'],
list((d for (var3003, var4630) in var3443._buffer)))
self.assertTrue(var3443._eof)
def test_feed_data(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.zlib = mock.Mock()
dbuf.zlib.decompress.return_value = b'line'
dbuf.feed_data(b'data', 4)
self.assertEqual([b'line'], list(d for d, _ in buf._buffer))
def test_feed_eof_no_err_brotli(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'br')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
dbuf.feed_eof()
assert [b'line'] == list(d for d, _ in buf._buffer)
def test_feed_eof_err_deflate(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
with pytest.raises(http_exceptions.ContentEncodingError):
dbuf.feed_eof()
def test_feed_data_err(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.side_effect = exc
with pytest.raises(http_exceptions.ContentEncodingError):
dbuf.feed_data(b'data', 4)
def test_feed_data_err(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.zlib = mock.Mock()
dbuf.zlib.decompress.side_effect = exc
self.assertRaises(
http_exceptions.ContentEncodingError, dbuf.feed_data, b'data', 4)
def test_feed_data_err(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.side_effect = exc
with pytest.raises(http_exceptions.ContentEncodingError):
dbuf.feed_data(b'data', 4)
def test_feed_eof_err(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
with pytest.raises(http_exceptions.ContentEncodingError):
dbuf.feed_eof()
def test_feed_eof(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.zlib = mock.Mock()
dbuf.zlib.flush.return_value = b'line'
dbuf.feed_eof()
self.assertEqual([b'line'], list(d for d, _ in buf._buffer))
self.assertTrue(buf._eof)
def test_feed_eof_no_err_brotli(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'br')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
dbuf.feed_eof()
assert [b'line'] == list(d for d, _ in buf._buffer)
async def test_feed_data(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.return_value = b'line'
# First byte should be b'x' in order code not to change the decoder.
dbuf.feed_data(b'xxxx', 4)
assert [b'line'] == list(d for d, _ in buf._buffer)
async def test_feed_eof_no_err_gzip(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'gzip')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
dbuf.feed_eof()
assert [b'line'] == list(d for d, _ in buf._buffer)
async def test_feed_eof_no_err_gzip(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'gzip')
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b'line'
dbuf.decompressor.eof = False
dbuf.feed_eof()
assert [b'line'] == list(d for d, _ in buf._buffer)
async def test_feed_eof_err_deflate(self, stream: Any) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
2**16,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, "deflate")
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b"line"
dbuf.decompressor.eof = False
with pytest.raises(http_exceptions.ContentEncodingError):
dbuf.feed_eof()
async def test_feed_eof_no_err_brotli(self, stream: Any) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
2**16,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, "br")
dbuf.decompressor = mock.Mock()
dbuf.decompressor.flush.return_value = b"line"
dbuf.decompressor.eof = False
dbuf.feed_eof()
assert [b"line"] == list(d for d, _ in buf._buffer)
async def test_feed_data_err(self, stream) -> None:
buf = aiohttp.FlowControlDataQueue(stream,
loop=asyncio.get_event_loop())
dbuf = DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.decompressor = mock.Mock()
dbuf.decompressor.decompress.side_effect = exc
with pytest.raises(http_exceptions.ContentEncodingError):
# Should be more than 4 bytes to trigger deflate FSM error.
# Should start with b'x', otherwise code switch mocked decoder.
dbuf.feed_data(b'xsomedata', 9)
def test_empty_body(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.feed_eof()
assert buf.at_eof()
def test_empty_body(self, stream):
buf = aiohttp.FlowControlDataQueue(stream)
dbuf = DeflateBuffer(buf, 'deflate')
dbuf.feed_eof()
assert buf.at_eof()
def function279(self):
var2756 = aiohttp.FlowControlDataQueue(self.attribute1675)
var4405 = DeflateBuffer(var2756, 'deflate')
var4405.feed_eof()
self.assertTrue(var2756.at_eof())
