Merge "Add test for partial responses" into main