def test_read_file_bigger_than_buffer(): old_max_buf = InputFile.max_buffer_size InputFile.max_buffer_size = 10 try: input_file = InputFile(rfile, len(bstr)) assert input_file.read() == test_file.read() finally: InputFile.max_buffer_size = old_max_buf
def test_read_file_bigger_than_buffer(): old_max_buf = InputFile.max_buffer_size InputFile.max_buffer_size = 10 try: input_file = InputFile(rfile, len(bstr)) assert input_file.read() == test_file.read() finally: InputFile.max_buffer_size = old_max_buf
def test_iter_file_bigger_than_buffer(): old_max_buf = InputFile.max_buffer_size InputFile.max_buffer_size = 10 try: input_file = InputFile(rfile, len(bstr)) for a, b in zip(input_file, test_file): assert a == b finally: InputFile.max_buffer_size = old_max_buf
def test_seek_file_bigger_than_buffer(): old_max_buf = InputFile.max_buffer_size InputFile.max_buffer_size = 10 try: input_file = InputFile(rfile, len(bstr)) input_file.seek(2) test_file.seek(2) assert input_file.read(1) == test_file.read(1) input_file.seek(4) test_file.seek(4) assert input_file.read(1) == test_file.read(1) finally: InputFile.max_buffer_size = old_max_buf
def test_seek_file_bigger_than_buffer(): old_max_buf = InputFile.max_buffer_size InputFile.max_buffer_size = 10 try: input_file = InputFile(rfile, len(bstr)) input_file.seek(2) test_file.seek(2) assert input_file.read(1) == test_file.read(1) input_file.seek(4) test_file.seek(4) assert input_file.read(1) == test_file.read(1) finally: InputFile.max_buffer_size = old_max_buf
def setup_function(function): global rfile, input_file, test_file rfile = BytesIO(bstr) test_file = BytesIO(bstr) input_file = InputFile(rfile, len(bstr))
import sys from io import BytesIO import pytest from wptserve.request import InputFile bstr = b'This is a test document\nWith new lines\nSeveral in fact...' rfile = '' test_file = '' # This will be used to test the InputFile functions against input_file = InputFile(None, 0) def setup_function(function): global rfile, input_file, test_file rfile = BytesIO(bstr) test_file = BytesIO(bstr) input_file = InputFile(rfile, len(bstr)) def teardown_function(function): rfile.close() test_file.close() def test_seek(): input_file.seek(2) test_file.seek(2) assert input_file.read(1) == test_file.read(1) input_file.seek(4)