示例#1
0
def test_multiple_files(empty_swagger_spec):
    request = {}
    file1_contents = "I am the contents of a file1"
    file2_contents = "I am the contents of a file2"
    op = Mock(spec=Operation, consumes=['multipart/form-data'])
    param1_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'photo',
    }
    param2_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'headshot',
    }

    param1 = Param(empty_swagger_spec, op, param1_spec)
    param2 = Param(empty_swagger_spec, op, param2_spec)
    add_file(param1, file1_contents, request)
    add_file(param2, file2_contents, request)
    expected_request = {
        'files': [
            ('photo', ('photo', 'I am the contents of a file1')),
            ('headshot', ('headshot', 'I am the contents of a file2')),
        ],
    }
    assert expected_request == request
示例#2
0
def test_multiple_files(empty_swagger_spec):
    request = {}
    file1_contents = "I am the contents of a file1"
    file2_contents = "I am the contents of a file2"
    op = Mock(spec=Operation, consumes=['multipart/form-data'])
    param1_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'photo'
    }
    param2_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'headshot'
    }

    param1 = Param(empty_swagger_spec, op, param1_spec)
    param2 = Param(empty_swagger_spec, op, param2_spec)
    add_file(param1, file1_contents, request)
    add_file(param2, file2_contents, request)
    expected_request = {
        'files': [
            ('photo', ('photo', 'I am the contents of a file1')),
            ('headshot', ('headshot', 'I am the contents of a file2')),
        ]
    }
    assert expected_request == request
示例#3
0
def test_mime_type_not_found_in_consumes(empty_swagger_spec):
    request = {}
    file_contents = "I am the contents of a file"
    op = Mock(spec=Operation, operation_id='upload_photos', consumes=[])
    param_spec = {'type': 'file', 'in': 'formData', 'name': 'photo'}
    param = Param(empty_swagger_spec, op, param_spec)
    with pytest.raises(SwaggerMappingError) as excinfo:
        add_file(param, file_contents, request)
    assert "not found in list of supported mime-types" in str(excinfo.value)
示例#4
0
def test_single_file(empty_swagger_spec):
    request = {}
    file_contents = "I am the contents of a file"
    op = Mock(spec=Operation, consumes=['multipart/form-data'])
    param_spec = {'type': 'file', 'in': 'formData', 'name': 'photo'}
    param = Param(empty_swagger_spec, op, param_spec)
    add_file(param, file_contents, request)
    expected_request = {
        'files': [('photo', ('photo', 'I am the contents of a file'))]
    }
    assert expected_request == request
示例#5
0
def test_mime_type_not_found_in_consumes(empty_swagger_spec):
    request = {}
    file_contents = "I am the contents of a file"
    op = Mock(spec=Operation, operation_id='upload_photos', consumes=[])
    param_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'photo'
    }
    param = Param(empty_swagger_spec, op, param_spec)
    with pytest.raises(SwaggerMappingError) as excinfo:
        add_file(param, file_contents, request)
    assert "not found in list of supported mime-types" in str(excinfo.value)
示例#6
0
def test_single_file(empty_swagger_spec):
    request = {}
    file_contents = "I am the contents of a file"
    op = Mock(spec=Operation, consumes=['multipart/form-data'])
    param_spec = {
        'type': 'file',
        'in': 'formData',
        'name': 'photo'
    }
    param = Param(empty_swagger_spec, op, param_spec)
    add_file(param, file_contents, request)
    expected_request = {
        'files': [('photo', ('photo', 'I am the contents of a file'))]
    }
    assert expected_request == request