Add test for simple stream.drain()