Refactor integration test
Add integration test for streaming responses