def test_stream_to_handler(self):
fs = motor.MotorGridFS(self.db)
for content_length in (0, 1, 100, 100 * 1000):
_id = yield fs.put(b'a' * content_length)
gridout = yield fs.get(_id)
handler = MockRequestHandler()
yield gridout.stream_to_handler(handler)
self.assertEqual(content_length, handler.n_written)
yield fs.delete(_id)
async def test_stream_to_handler(self):
fs = motor.MotorGridFSBucket(self.db)
for content_length in (0, 1, 100, 100 * 1000):
_id = await fs.upload_from_stream('filename', b'a' * content_length)
gridout = await fs.open_download_stream(_id)
handler = MockRequestHandler()
await gridout.stream_to_handler(handler)
self.assertEqual(content_length, handler.n_written)
await fs.delete(_id)