Add test infrastructure for streaming API