Add test for infinite stream