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()
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_stop_failed(
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 = 'STOPPED'
with pytest.raises(SbgError):
upload.stop()
os.remove(temp_file.name)
def test_file_upload_start_failed(api, given, tmpdir):
upload_id = generator.uuid4()
project_id = generator.uuid4()
file_name = generator.uuid4()
given.uploads.initialized_upload(
part_size=PartSize.UPLOAD_RECOMMENDED_SIZE,
upload_id=upload_id,
failed=True)
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._status = TransferState.RUNNING
with pytest.raises(SbgError):
upload.start()
def test_file_upload_start_failed(api, given):
upload_id = generator.uuid4()
project_id = generator.uuid4()
file_name = generator.uuid4()
given.uploads.initialized_upload(part_size=1,
upload_id=upload_id,
failed=True)
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._status = 'RUNNING'
with pytest.raises(SbgError):
upload.start()
os.remove(temp_file.name)
def test_file_upload_pause_failed(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._status = TransferState.STOPPED
with pytest.raises(SbgError):
upload.pause()