Add test for generating screenshots using streams