add tests for the streaming