mirror of
https://github.com/LukeHagar/connexion.git
synced 2025-12-09 12:27:46 +00:00
72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
import flask
|
|
import json
|
|
import pytest
|
|
# we are using "mock" module here for Py 2.7 support
|
|
from mock import MagicMock
|
|
|
|
from connexion.problem import problem
|
|
from connexion.decorators.validation import validate_pattern, validate_minimum, validate_maximum, ParameterValidator
|
|
|
|
def test_validate_pattern():
|
|
assert validate_pattern({}, '') is None
|
|
assert validate_pattern({'pattern': 'a'}, 'a') is None
|
|
assert validate_pattern({'pattern': 'a'}, 'b') == 'Invalid value, pattern "a" does not match'
|
|
|
|
|
|
def test_validate_minimum():
|
|
assert validate_minimum({}, 1) is None
|
|
assert validate_minimum({'minimum': 1}, 1) is None
|
|
assert validate_minimum({'minimum': 1.1}, 1) == 'Invalid value, must be at least 1.1'
|
|
|
|
|
|
def test_validate_maximum():
|
|
assert validate_maximum({}, 1) is None
|
|
assert validate_maximum({'maximum': 1}, 1) is None
|
|
assert validate_maximum({'maximum': 0}, 1) == 'Invalid value, must be at most 0'
|
|
|
|
|
|
def test_parameter_validator(monkeypatch):
|
|
request = MagicMock(name='request')
|
|
request.args = {}
|
|
request.headers = {}
|
|
request.params = {}
|
|
app = MagicMock(name='app')
|
|
app.response_class = lambda a, mimetype, status: json.loads(a)['detail']
|
|
monkeypatch.setattr('flask.request', request)
|
|
monkeypatch.setattr('flask.current_app', app)
|
|
|
|
def orig_handler(*args, **kwargs):
|
|
return 'OK'
|
|
|
|
params = [{'name': 'p1', 'in': 'path', 'type': 'integer', 'required': True},
|
|
{'name': 'h1', 'in': 'header', 'type': 'string', 'enum': ['a', 'b']},
|
|
{'name': 'q1', 'in': 'query', 'type': 'integer', 'maximum': 3},
|
|
{'name': 'a1', 'in': 'query', 'type': 'array', 'items': {'type': 'integer', 'minimum': 0}}]
|
|
validator = ParameterValidator(params)
|
|
handler = validator(orig_handler)
|
|
|
|
assert handler() == "Missing path parameter 'p1'"
|
|
assert handler(p1='123') == 'OK'
|
|
assert handler(p1='') == "Wrong type, expected 'integer' for path parameter 'p1'"
|
|
assert handler(p1='foo') == "Wrong type, expected 'integer' for path parameter 'p1'"
|
|
assert handler(p1='1.2') == "Wrong type, expected 'integer' for path parameter 'p1'"
|
|
|
|
request.args = {'q1': '4'}
|
|
assert handler(p1=1) == 'Invalid value, must be at most 3'
|
|
request.args = {'q1': '3'}
|
|
assert handler(p1=1) == 'OK'
|
|
|
|
request.args = {'a1': '1,2'}
|
|
assert handler(p1=1) == "OK"
|
|
request.args = {'a1': '1,a'}
|
|
assert handler(p1=1) == "Wrong type, expected 'integer' for query parameter 'a1'"
|
|
request.args = {'a1': '1,-1'}
|
|
assert handler(p1=1) == "Invalid value, must be at least 0"
|
|
del request.args['a1']
|
|
|
|
request.headers = {'h1': 'a'}
|
|
assert handler(p1='123') == 'OK'
|
|
|
|
request.headers = {'h1': 'x'}
|
|
assert handler(p1='123') == "Enum value must be one of ['a', 'b']"
|