Esempio n. 1
0
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'
Esempio n. 2
0
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()