Adding tests for the StreamWrapper update