Integration tests refactor

Move streaming tests to its own test class
Move util stuff to package
Comments for tests