1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
import pytest
from werkzeug.datastructures import Headers
from werkzeug.datastructures import ImmutableMultiDict
from openapi_core.contrib.flask import FlaskOpenAPIRequest
from openapi_core.datatypes import RequestParameters
class TestFlaskOpenAPIRequest:
def test_type_invalid(self):
with pytest.raises(TypeError):
FlaskOpenAPIRequest(None)
def test_simple(self, request_factory, request):
request = request_factory("GET", "/", subdomain="www")
openapi_request = FlaskOpenAPIRequest(request)
path = {}
query = ImmutableMultiDict([])
headers = Headers(request.headers)
cookies = {}
assert openapi_request.parameters == RequestParameters(
path=path,
query=query,
header=headers,
cookie=cookies,
)
assert openapi_request.method == "get"
assert openapi_request.host_url == request.host_url
assert openapi_request.path == request.path
assert openapi_request.body == b""
assert openapi_request.content_type == "application/octet-stream"
def test_multiple_values(self, request_factory, request):
request = request_factory(
"GET", "/", subdomain="www", query_string="a=b&a=c"
)
openapi_request = FlaskOpenAPIRequest(request)
path = {}
query = ImmutableMultiDict(
[
("a", "b"),
("a", "c"),
]
)
headers = Headers(request.headers)
cookies = {}
assert openapi_request.parameters == RequestParameters(
path=path,
query=query,
header=headers,
cookie=cookies,
)
assert openapi_request.method == "get"
assert openapi_request.host_url == request.host_url
assert openapi_request.path == request.path
assert openapi_request.body == b""
assert openapi_request.content_type == "application/octet-stream"
def test_url_rule(self, request_factory, request):
request = request_factory("GET", "/browse/12/", subdomain="kb")
openapi_request = FlaskOpenAPIRequest(request)
path = {"id": 12}
query = ImmutableMultiDict([])
headers = Headers(request.headers)
cookies = {}
assert openapi_request.parameters == RequestParameters(
path=path,
query=query,
header=headers,
cookie=cookies,
)
assert openapi_request.method == "get"
assert openapi_request.host_url == request.host_url
assert openapi_request.path == request.path
assert openapi_request.path_pattern == "/browse/{id}/"
assert openapi_request.body == b""
assert openapi_request.content_type == "application/octet-stream"
|