def test_iterator_buffering(tmppath): "Test file iteration." f = "data/multiline.txt" xfile = XRootDPyFile(mkurl(join(tmppath, f)), 'r') assert len(list(iter(xfile))) == len(open(join(tmppath, f)).readlines()) xfile = XRootDPyFile(mkurl(join(tmppath, f)), 'r', buffering=10) assert len(list(iter(xfile))) == int(math.ceil(xfile.size / 10.0))
def test_writelines(tmppath): """Test writelines().""" xfile = XRootDPyFile(mkurl(join(tmppath, "data/multiline.txt")), 'r') yfile = XRootDPyFile(mkurl(join(tmppath, "data/newfile.txt")), 'w+') yfile.writelines(xfile.xreadlines()) xfile.seek(0), yfile.seek(0) assert xfile.readlines() == yfile.readlines()
def test__is_open(tmppath): """Test _is_open()""" fd = get_tsta_file(tmppath) full_path = fd['full_path'] xfile = XRootDPyFile(mkurl(full_path)) assert not xfile.closed xfile.close() assert xfile.closed
def test_open_close(tmppath): """Test close() on an open file.""" fd = get_tsta_file(tmppath) full_path = fd['full_path'] xfile = XRootDPyFile(mkurl(full_path)) assert xfile assert not xfile.closed xfile.close() assert xfile.closed
def test_open_close(tmppath): """Test close() on an open file.""" fd = get_tsta_file(tmppath) full_path = fd["full_path"] xfile = XRootDPyFile(mkurl(full_path)) assert xfile assert not xfile.closed xfile.close() assert xfile.closed
def test_truncate4(tmppath): """Verifies that truncate() raises errors on non-truncatable files.""" fd = get_mltl_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path), 'r') pytest.raises(IOError, xfile.truncate, 0) xfile.close() xfile = XRootDPyFile(mkurl(full_path), 'w-') pytest.raises(IOError, xfile.truncate, 0)
def test_read_binary(tmppath): """Tests reading binary data from an existing file.""" fd = get_bin_testfile(tmppath) fb = get_copy_file(fd, binary=True) fp, fc = fd['full_path'], fd['contents'] fp2 = fb['full_path'] pfile = open(fp2, 'rb') xfile = XRootDPyFile(mkurl(fp), 'rb') assert xfile.read() == pfile.read() == fc
def test_open_close(tmppath): """Test close() on an open file.""" fd = get_tsta_file(tmppath) full_path = fd['full_path'] xfile = XRootDPyFile(mkurl(full_path)) assert xfile assert not xfile.closed xfile.close() assert xfile.closed # Multiple calls to closed do nothing. xfile.close()
def test_xreadlines(tmppath): """Tests xreadlines()""" fp = get_mltl_file(tmppath)['full_path'] xfile = XRootDPyFile(mkurl(fp), 'r') rl = xfile.readlines() xfile.seek(0) xl = xfile.xreadlines() assert xl != rl assert list(xl) == rl
def test_reading_whole_big_file(tmppath): """Tests reading end of big file.""" f = "data/big_file.txt" create_big_file(tmppath, f) xfile = XRootDPyFile(mkurl(join(tmppath, f)), 'r') pytest.raises(IOError, xfile.read) xfile.close() remove_file(tmppath, f)
def test_seekable(tmppath): """Test seekable.""" assert XRootDPyFile( mkurl(join(tmppath, "data/testa.txt")), 'r-').seekable() is False assert XRootDPyFile( mkurl(join(tmppath, "data/testa.txt")), 'r').seekable() is True assert XRootDPyFile( mkurl(join(tmppath, "data/testa.txt")), 'w').seekable() is True assert XRootDPyFile( mkurl(join(tmppath, "data/testa.txt")), 'w-').seekable() is False assert XRootDPyFile( mkurl(join(tmppath, "data/testa.txt")), 'r+').seekable() is True
def test_reading_begining_of_big_file(tmppath): """Tests reading end of big file.""" f = "data/big_file.txt" create_big_file(tmppath, f, frontfile_content="test") xfile = XRootDPyFile(mkurl(join(tmppath, f)), 'r') data = xfile.read(4) assert data == b"test" xfile.close() remove_file(tmppath, f)
def test_init_readmode_basic(tmppath): # Non-existing file causes what? # Resource not found error. fn, fp, fc = "nope", "data/", "" full_path = join(tmppath, fp, fn) pytest.raises(ResourceNotFoundError, XRootDPyFile, mkurl(full_path), mode="r") # Existing file can be read? fd = get_tsta_file(tmppath) full_path, fc = fd["full_path"], fd["contents"] xfile = XRootDPyFile(mkurl(full_path), mode="r") assert xfile assert xfile.read() == fc
def test_init_readmode_basic(tmppath): # Non-existing file causes what? # Resource not found error. fn, fp, fc = 'nope', 'data/', '' full_path = join(tmppath, fp, fn) pytest.raises(ResourceNotFoundError, XRootDPyFile, mkurl(full_path), mode='r') # Existing file can be read? fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path), mode='r') assert xfile assert xfile.read() == fc
def test_read_existing(tmppath): """Test read() on an existing non-empty file.""" fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path)) res = xfile.read() assert res == fc # After having read the entire file, the file pointer is at the # end of the file and consecutive reads return the empty string. assert xfile.read() == '' # reset ipp to start xfile.seek(0) assert xfile.read(1) == fc[0] assert xfile.read(2) == fc[1:3] overflow_read = xfile.read(len(fc)) assert overflow_read == fc[3:] # Mock an error, yayy! fake_status = { "status": 3, "code": 0, "ok": False, "errno": errno.EREMOTE, "error": True, "message": '[FATAL] Remote I/O Error', "fatal": True, "shellcode": 51 } xfile._file.read = Mock(return_value=(XRootDStatus(fake_status), None)) pytest.raises(IOError, xfile.read)
def test_readwrite_diffrent_encodings_fails(tmppath): """Test read/write a unicode str in non unicode files.""" fd = get_tsta_file(tmppath) fb = get_copy_file(fd) fp, dummy = fd['full_path'], fd['contents'] fp2 = fb['full_path'] unicodestr = u"æøå" pfile = open(fp2, 'w') # default encoding is ANSI pytest.raises(UnicodeEncodeError, pfile.write, unicodestr) xfile = XRootDPyFile(mkurl(fp), 'w', encoding='ascii') pytest.raises(UnicodeEncodeError, xfile.write, unicodestr) xfile.close()
def test_init_writemode_basic(tmppath): # Non-existing file is created. fn, fp, fc = "nope", "data/", "" full_path = join(tmppath, fp, fn) xfile = XRootDPyFile(mkurl(full_path), mode="w+") assert xfile assert xfile.read() == fc # Existing file is truncated fd = get_tsta_file(tmppath) full_path = fd["full_path"] xfile = XRootDPyFile(mkurl(full_path), mode="w+") assert xfile assert xfile.read() == "" assert xfile.size == 0 assert xfile.tell() == 0
def test_init_writemode_basic(tmppath): # Non-existing file is created. fn, fp, fc = 'nope', 'data/', '' full_path = join(tmppath, fp, fn) xfile = XRootDPyFile(mkurl(full_path), mode='w+') assert xfile assert xfile.read() == fc # Existing file is truncated fd = get_tsta_file(tmppath) full_path = fd['full_path'] xfile = XRootDPyFile(mkurl(full_path), mode='w+') assert xfile assert xfile.read() == '' assert xfile.size == 0 assert xfile.tell() == 0
def test_truncate3(tmppath): """Test truncate(0 < size < self._size).""" fd = get_mltl_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path), 'r+') initcp = xfile.tell() newsiz = len(fc)//2 xfile.truncate(newsiz) assert xfile.tell() == initcp xfile.seek(0) # reset the internal pointer before reading assert xfile.read() == fc[:newsiz]
def test_init_writemode(tmppath): """Tests for opening files in 'w(+)'""" fd = get_tsta_file(tmppath) fp, fc = fd["full_path"], fd["contents"] xfile = XRootDPyFile(mkurl(fp), "w") pytest.raises(IOError, xfile.read) xfile.seek(1) conts = "what" xfile.write(conts) assert xfile.tell() == 1 + len(conts) assert xfile.size == 1 + len(conts) xfile.close() xfile = XRootDPyFile(mkurl(fp), "r") fc = xfile.read() assert fc == "\x00" + conts assert not fc == conts
def test_init_writemode(tmppath): """Tests for opening files in 'w(+)'""" fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'w') pytest.raises(IOError, xfile.read) xfile.seek(1) conts = 'what' xfile.write(conts) assert xfile.tell() == 1 + len(conts) assert xfile.size == 1 + len(conts) xfile.close() xfile = XRootDPyFile(mkurl(fp), 'r') fc = xfile.read() assert fc == '\x00'+conts assert not fc == conts
def create_big_file(tmppath, filename, size=2 * 1024 * 1024 * 1024, frontfile_content=None, endfile_content=None): xfile = XRootDPyFile(mkurl(join(tmppath, filename)), 'w') if endfile_content: endfile_length = len(endfile_content) else: endfile_content = '\0' endfile_length = 1 # Prepare big file for testing if frontfile_content: xfile.write(frontfile_content) xfile.seek(size - endfile_length) xfile.write(endfile_content) xfile.close()
def test_size_len(tmppath): """Tests for the size and len property.""" fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path)) assert xfile.size == len(fc) assert len(xfile) == len(fc) # Length of empty file xfile = XRootDPyFile(mkurl(join(tmppath, fd['dir'], 'whut')), 'w+') assert xfile.size == len('') assert len(xfile) == len('') # Length of multiline file fd = get_mltl_file(tmppath) fpp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fpp)) assert xfile.size == len(fc) assert len(xfile) == len(fc) # Mock the error fake_status = { "status": 3, "code": 0, "ok": False, "errno": errno.EREMOTE, "error": True, "message": '[FATAL] Remote I/O Error', "fatal": True, "shellcode": 51 } xfile.close() xfile = XRootDPyFile(mkurl(full_path)) xfile._file.stat = Mock(return_value=(XRootDStatus(fake_status), None)) try: xfile.size assert False except IOError: assert True try: len(xfile) assert False except IOError: assert True
def test_truncate2(tmppath): """Test truncate(self._size).""" fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path), 'r+') conts = xfile.read() assert conts == fc newsize = xfile.size xfile.truncate(newsize) assert xfile.tell() == newsize assert xfile.size == len(fc) xfile.seek(0) assert xfile.read() == conts
def test_init_newline(tmppath): """Tests fs.open() with specified newline parameter.""" fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp)) assert xfile._newline == '\n' xfile.close() xfile = XRootDPyFile(mkurl(fp), newline='\n') assert xfile._newline == '\n' xfile.close() pytest.raises(UnsupportedError, XRootDPyFile, mkurl(fp), mode='r', newline='what')
def test_init_basic(tmppath): """Test basic initialization of existing file.""" fname = 'testa.txt' fpath = 'data/' fcontents = 'testa.txt\n' full_fpath = join(tmppath, fpath, fname) xfile = XRootDPyFile(mkurl(full_fpath)) assert xfile assert type(xfile == XRootDPyFile) assert xfile._file assert xfile.mode == 'r' # Verify that underlying/wrapped file can be read. statmsg, res = xfile._file.read() assert res == fcontents
def test_reading_end_of_big_file(tmppath): """Tests reading end of big file.""" f = "data/big_file.txt" create_big_file(tmppath, f, endfile_content="test\0") xfile = XRootDPyFile(mkurl(join(tmppath, f)), 'r') xfile.seek(xfile.size - 10) data = xfile.read() assert xfile.size == 2 * 1024 * 1024 * 1024 assert len(data) == 10 assert data[-5:-1] == b"test" xfile.close() remove_file(tmppath, f)
def test_close_error(tmppath): """Test error on close().""" xfile = XRootDPyFile(mkurl(get_tsta_file(tmppath)['full_path'])) # Mock error response on close. fake_status = { "status": 1, "code": 3, "ok": False, "errno": 0, "error": True, "message": '[ERROR] Invalid operation', "fatal": False, "shellcode": 50 } xfile._file.close = Mock(return_value=(XRootDStatus(fake_status), None)) # Ensure error is raised. pytest.raises(IOError, xfile.close)
def test_init_newline(tmppath): """Tests fs.open() with specified newline parameter.""" fd = get_tsta_file(tmppath) fp, fc = fd["full_path"], fd["contents"] xfile = XRootDPyFile(mkurl(fp)) assert xfile._newline == "\n" xfile.close() xfile = XRootDPyFile(mkurl(fp), newline="\n") assert xfile._newline == "\n" xfile.close() pytest.raises(UnsupportedError, XRootDPyFile, mkurl(fp), mode="r", newline="what")
def test_size(tmppath): """Tests for the size property size.""" fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path)) assert xfile.size == len(fc) # Length of empty file xfile = XRootDPyFile(mkurl(join(tmppath, fd['dir'], 'whut')), 'w+') assert xfile.size == len('') # Length of multiline file fd = get_mltl_file(tmppath) fpp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fpp)) assert xfile.size == len(fc) # Mock the error fake_status = { "status": 3, "code": 0, "ok": False, "errno": errno.EREMOTE, "error": True, "message": '[FATAL] Remote I/O Error', "fatal": True, "shellcode": 51 } xfile.close() xfile = XRootDPyFile(mkurl(full_path)) xfile._file.stat = Mock(return_value=(XRootDStatus(fake_status), None)) try: xfile.size assert False except IOError: assert True
def test_init_streammodes(tmppath): fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'r-') pytest.raises(IOError, xfile.seek, 3) assert xfile.size == len(fc) assert xfile.tell() == 0 assert xfile.read() == fc assert xfile.tell() == len(fc) xfile.close() xfile = XRootDPyFile(mkurl(fp), 'w-') pytest.raises(IOError, xfile.read) pytest.raises(IOError, xfile.seek, 3) assert xfile.tell() == 0 assert xfile.size == 0 conts = 'hugs are delightful' xfile.write(conts) assert xfile.tell() == len(conts) xfile.close() xfile = XRootDPyFile(mkurl(fp), 'r') assert xfile.read() == conts
def test_write(tmppath): """Test write().""" # With a new file. xfile = XRootDPyFile(mkurl(join(tmppath, 'data/nuts')), 'w+') assert xfile.size == 0 conts = xfile.read() assert not conts nconts = 'Write.' xfile.write(nconts) assert xfile.tell() == len(nconts) assert not xfile.closed xfile.seek(0) assert xfile.size == len(nconts) assert xfile.read() == nconts xfile.close() # Verify persistence after closing. xfile = XRootDPyFile(mkurl(join(tmppath, 'data/nuts')), 'r+') assert xfile.size == len(nconts) assert xfile.read() == nconts # Seek(x>0) followed by a write nc2 = 'hello' cntr = len(nconts)//2 xfile.seek(cntr) xfile.write(nc2) assert xfile.tell() == len(nc2) + cntr xfile.seek(0) assert xfile.read() == nconts[:cntr] + nc2 xfile.close() # Seek(x>0) followed by a write of len < size-x fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'r+') assert xfile.read() == fc xfile.seek(2) nc = 'yo' xfile.write(nc) assert xfile.tell() == len(nc) + 2 assert xfile.read() == fc[2+len(nc):] # run w/ flushing == true xfile.write('', True) # Mock an error, yayy! fake_status = { "status": 3, "code": 0, "ok": False, "errno": errno.EREMOTE, "error": True, "message": '[FATAL] Remote I/O Error', "fatal": True, "shellcode": 51 } xfile._file.write = Mock(return_value=(XRootDStatus(fake_status), None)) pytest.raises(IOError, xfile.write, '')
def test_readwrite_unicode(tmppath): """Test read/write unicode.""" if sys.getdefaultencoding() != 'ascii': # Python 2 only problem raise AssertionError( "Default system encoding is not ascii. This is likely due to some" " imported module changing it using sys.setdefaultencoding." ) fd = get_tsta_file(tmppath) fb = get_copy_file(fd) fp, dummy = fd['full_path'], fd['contents'] fp2 = fb['full_path'] unicodestr = u"æøå" pfile = open(fp2, 'w') xfile = XRootDPyFile(mkurl(fp), 'w') pytest.raises(UnicodeEncodeError, pfile.write, unicodestr) pytest.raises(UnicodeEncodeError, xfile.write, unicodestr) xfile.close() xfile = XRootDPyFile(mkurl(fp), 'w+', encoding='utf-8') xfile.write(unicodestr) xfile.flush() xfile.seek(0) assert unicodestr.encode('utf8') == xfile.read() xfile.close() xfile = XRootDPyFile(mkurl(fp), 'w+', errors='ignore') xfile.write(unicodestr) xfile.flush() xfile.seek(0) assert unicodestr.encode('ascii', 'ignore') == xfile.read() xfile.close()
def test_init_append(tmppath): """Test for files opened 'a'""" fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'a') assert xfile.mode == 'a' pytest.raises(IOError, xfile.read) assert xfile.tell() == len(fc) # Seeking is allowed, but writes still go on the end. xfile.seek(0) assert xfile.tell() == 0 newcont = u'butterflies' xfile.write(newcont) assert xfile.tell() == len(fc) + len(newcont) # Can't read in this mode. xfile.close() xfile = XRootDPyFile(mkurl(fp), 'r') assert xfile.read() == fc + newcont xfile.close() xfile = XRootDPyFile(mkurl(fp), 'a') xfile.write(fc) xfile.seek(0) pytest.raises(IOError, xfile.read)
def test_init_appendread(tmppath): """Test for files opened in mode 'a+'.""" fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'a+') assert xfile.mode == 'a+' assert xfile.tell() == len(fc) assert xfile.read() == u'' # Seeking is allowed, but writes still go on the end. xfile.seek(0) assert xfile.tell() == 0 newcont = u'butterflies' xfile.write(newcont) assert xfile.tell() == len(fc) + len(newcont) xfile.seek(0) assert xfile.read() == fc + newcont xfile.write(fc) xfile.seek(0) xfile.read() == fc + newcont + fc
def test_readline(tmppath): """Tests for readline().""" fd = get_mltl_file(tmppath) fb = get_copy_file(fd) fp, fc = fd['full_path'], fd['contents'] fp2 = fb['full_path'] xfile, pfile = XRootDPyFile(mkurl(fp), 'r'), opener.open(fp2, 'r') assert xfile.readline() == pfile.readline() assert xfile.readline() == pfile.readline() assert xfile.readline() == pfile.readline() xfile.close(), pfile.close() xfile, pfile = XRootDPyFile(mkurl(fp), 'r'), opener.open(fp2, 'r') assert xfile.readline() == pfile.readline() xfile.seek(0), pfile.seek(0) assert xfile.readline() == pfile.readline() assert xfile.tell(), pfile.tell() xfile.close(), pfile.close() xfile = XRootDPyFile(mkurl(fp), 'w+') str1 = 'hello\n' str2 = 'bye\n' xfile.write(str1+str2) xfile.seek(0) assert xfile.readline() == str1 assert xfile.readline() == str2 assert xfile.readline() == '' assert xfile.readline() == '' xfile.seek(100) assert xfile.readline() == '' xfile.close() xfile = XRootDPyFile(mkurl(fp), 'w+') xfile.write(str2) xfile.seek(len(str2)+1) xfile.write(str2) xfile.seek(0) assert xfile.readline() == str2 assert xfile.readline() == u'\x00'+str2
def test_read_and_write(tmppath): """Tests that the XRDFile behaves like a regular python file.""" fd = get_tsta_file(tmppath) fb = get_copy_file(fd) fp, fc = fd['full_path'], fd['contents'] fp2 = fb['full_path'] seekpoint = len(fc)//2 writestr = "Come what may in May this day says Ray all gay like Jay" pfile = open(fp2, 'r+') xfile = XRootDPyFile(mkurl(fp), 'r+') assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() assert xfile.tell() == pfile.tell() xfile.seek(seekpoint), pfile.seek(seekpoint) assert xfile.tell() == pfile.tell() xfile.write(writestr), pfile.write(writestr) assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() xfile.seek(0), pfile.seek(0) assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read()
def test_read_errors(tmppath): fd = get_tsta_file(tmppath) fp, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(fp), 'r') xfile.close() pytest.raises(ValueError, xfile.read)
def test_write_and_read(tmppath): """Tests that the XRootDPyFile behaves like a regular python file in w+.""" fd = get_tsta_file(tmppath) fb = get_copy_file(fd) fp, fc = fd['full_path'], fd['contents'] fp2 = fb['full_path'] writestr = "Hello fair mare what fine stairs." seekpoint = len(writestr)//2 # In 'w' (and variant modes) the file's contents are deleted upon opening. pfile = open(fp2, 'w+') xfile = XRootDPyFile(mkurl(fp), 'w+') assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() assert xfile.tell() == pfile.tell() xfile.write(writestr), pfile.write(writestr) assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() xfile.seek(0), pfile.seek(0) assert xfile.read() == pfile.read() assert xfile.tell() == pfile.tell() xfile.seek(seekpoint), pfile.seek(seekpoint) assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() assert xfile.tell() == pfile.tell()
def test_seek_past_eof_wr(tmppath): """Tests read/write/truncate behaviour after seeking past the EOF, 'w+'""" fd = get_tsta_file(tmppath) fb = get_copy_file(fd) fp, fc = fd['full_path'], u'' fp2 = fb['full_path'] wstr = "www" eof = len(fc) skpnt = len(fc)+4 pfile = open(fp2, 'w+') xfile = XRootDPyFile(mkurl(fp), 'w+') xfile.seek(skpnt), pfile.seek(skpnt) assert xfile.tell() == pfile.tell() assert xfile.read() == pfile.read() assert xfile.tell() == pfile.tell() assert xfile.tell() == skpnt xfile.write(wstr), pfile.write(wstr) assert xfile.tell() == pfile.tell() xfile.seek(eof), pfile.seek(eof) assert xfile.read() == pfile.read() == '\x00'*(skpnt-eof) + wstr assert xfile.tell() == pfile.tell() xfile.seek(0), pfile.seek(0) assert xfile.read() == pfile.read() xfile.truncate(skpnt), pfile.truncate(skpnt) assert xfile.tell() == pfile.tell() == skpnt + len(wstr) xfile.write(wstr), pfile.write(wstr) expected = fc + '\x00'*(skpnt-eof+len(wstr)) + wstr xfile.seek(0), pfile.seek(0) assert xfile.read() == pfile.read() == expected
def test_write_binary(tmppath): """Tests for writing binary data to file.""" fd = get_bin_testfile(tmppath) fp, fc = fd['full_path'], fd['contents'] # Test w/ confirmed binary data read from a binary file xf_new = XRootDPyFile(mkurl(join(tmppath, 'data/tmp_bin')), 'wb+') xf_new.write(fc), xf_new.seek(0) assert xf_new.read() == fc xf_new.close() # Verify persistence. xf_new = XRootDPyFile(mkurl(join(tmppath, 'data/tmp_bin')), 'r+') assert xf_new.read() == fc # Test truncate xf_new.truncate() xf_new.seek(0) assert xf_new.read() == fc xf_new.close() # Test with bytearray xf_new = XRootDPyFile(mkurl(join(tmppath, 'data/tmp_bin')), 'wb+') barr = bytearray(range(0, 5)) xf_new.write(barr), xf_new.seek(0) assert xf_new.read() == barr xf_new.close() # Verify persistence. xf_new = XRootDPyFile(mkurl(join(tmppath, 'data/tmp_bin')), 'r') assert xf_new.read() == barr xf_new.close()
def test_flush(tmppath): """Tests for flush()""" # Mostly it just ensures calling it doesn't crash the program. fd = get_tsta_file(tmppath) full_path, fc = fd['full_path'], fd['contents'] xfile = XRootDPyFile(mkurl(full_path), 'w') writestr = 'whut' xfile.flush() xfile.seek(0, SEEK_END) xfile.write(writestr) xfile.flush() xfile.close() xfile = XRootDPyFile(mkurl(full_path), 'r') assert xfile.read() == writestr # Fake/mock an error response fake_status = { "status": 3, "code": 0, "ok": False, "errno": errno.EREMOTE, "error": True, "message": '[FATAL] Remote I/O Error', "fatal": True, "shellcode": 51 } # Assign mock return value to the file's sync() function # (which is called by flush()) xfile._file.sync = Mock(return_value=(XRootDStatus(fake_status), None)) pytest.raises(IOError, xfile.flush)