Add unit test for streams