def test_feed_data_err(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = protocol.DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.zlib = unittest.mock.Mock()
dbuf.zlib.decompress.side_effect = exc
self.assertRaises(errors.ContentEncodingError, dbuf.feed_data, b'data')
def test_feed_data(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = protocol.DeflateBuffer(buf, 'deflate')
dbuf.zlib = unittest.mock.Mock()
dbuf.zlib.decompress.return_value = b'line'
dbuf.feed_data(b'data')
self.assertEqual([b'line'], list(buf._buffer))
def test_feed_eof_err(self):
buf = aiohttp.DataQueue()
dbuf = protocol.DeflateBuffer(buf, 'deflate')
dbuf.zlib = unittest.mock.Mock()
dbuf.zlib.flush.return_value = b'line'
dbuf.zlib.eof = False
self.assertRaises(errors.IncompleteRead, dbuf.feed_eof)
def test_feed_data_err(self):
buf = aiohttp.DataQueue()
dbuf = protocol.DeflateBuffer(buf, 'deflate')
exc = ValueError()
dbuf.zlib = unittest.mock.Mock()
dbuf.zlib.decompress.side_effect = exc
self.assertRaises(errors.IncompleteRead, dbuf.feed_data, b'data')
def test_feed_eof_err(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = protocol.DeflateBuffer(buf, 'deflate')
dbuf.zlib = mock.Mock()
dbuf.zlib.flush.return_value = b'line'
dbuf.zlib.eof = False
self.assertRaises(errors.ContentEncodingError, dbuf.feed_eof)
def test_feed_eof(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = protocol.DeflateBuffer(buf, 'deflate')
dbuf.zlib = unittest.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_empty_body(self):
buf = aiohttp.FlowControlDataQueue(self.stream)
dbuf = protocol.DeflateBuffer(buf, 'deflate')
dbuf.feed_eof()
self.assertTrue(buf.at_eof())
