Update the test federation client to handle streaming responses.