示例#1
0
def test_file_upload_finalize_failed(api, given):
    upload_id = generator.uuid4()
    file_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()
    given.uploads.finalized_upload(file_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)

    with pytest.raises(SbgError):
        upload.run()

    os.remove(temp_file.name)
示例#2
0
def test_file_upload_init_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(str(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)

    with pytest.raises(SbgError):
        upload.run()