def test_tee_rejects_StringIO(): fileobject = io.StringIO() with pytest.raises(TypeError): # The generator needs to be iterated over before the exception will be # raised sum(len(c) for c in tee.tee(None, fileobject))
def test_tee(streamed_response): response = streamed_response expected_len = len('chunk') * 8 fileobject = io.BytesIO() assert expected_len == sum(len(c) for c in tee.tee(response, fileobject)) assert fileobject.getvalue() == b'chunkchunkchunkchunkchunkchunkchunkchunk'