def test_gfal_download_to_file(): gfal_file = StringIO() local_file = StringIO() gfal_file.write('content') gfal_file.seek(0) with gfal2.mocked_gfal2(dumper, files={'srm://example.com/file': gfal_file}): dumper.gfal_download_to_file('srm://example.com/file', local_file) local_file.seek(0) assert local_file.read() == 'content'
def test_gfal_download_creates_file_with_content(): gfal_file = StringIO() local_file = StringIO() gfal_file.write('content') gfal_file.seek(0) with gfal2.mocked_gfal2(dumper, files={'srm://example.com/file': gfal_file}): with mock_open(dumper, local_file): dumper.gfal_download('srm://example.com/file', 'filename') local_file.seek(0) assert local_file.read() == 'content'