def test_file_upload_resume(api, given): upload_id = generator.uuid4() project_id = generator.uuid4() file_part_url = generator.url() file_name = generator.uuid4() given.uploads.initialized_upload(part_size=1, upload_id=upload_id) given.uploads.got_file_part(file_part_url) given.uploads.got_etag(file_part_url) given.uploads.reported_part() temp_file = tempfile.NamedTemporaryFile('w', delete=False, dir='.') temp_file.write('dummy content') temp_file.close() upload = Upload(temp_file.name, project_id, api=api, part_size=1, file_name=file_name) upload.start() upload._status = 'PAUSED' upload.resume() os.remove(temp_file.name) assert upload.status == 'RUNNING'
def test_file_upload_resume(api, given, tmpdir): upload_id = generator.uuid4() project_id = generator.uuid4() file_part_url = generator.url() file_name = generator.uuid4() given.uploads.initialized_upload( part_size=PartSize.UPLOAD_RECOMMENDED_SIZE, upload_id=upload_id) given.uploads.got_file_part(file_part_url) given.uploads.got_etag(file_part_url) given.uploads.reported_part() given.uploads.deleted() with open(tmpdir / file_name, 'w') as temp_file: temp_file.write('dummy content') upload = Upload(temp_file.name, project_id, api=api, part_size=PartSize.UPLOAD_RECOMMENDED_SIZE, file_name=file_name) upload.start() upload._status = TransferState.RUNNING upload.pause() upload._status = TransferState.PAUSED upload.resume() assert upload.status == TransferState.RUNNING upload.stop()