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())