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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# Copyright 2015 Vladimir Rutsky <vladimir@rutsky.org>
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""aiohttp_cors.urldispatcher_router_adapter unit tests."""
from unittest import mock
import pytest
from aiohttp import web
from aiohttp_cors import ResourceOptions
from aiohttp_cors.urldispatcher_router_adapter import (
ResourcesUrlDispatcherRouterAdapter,
)
async def _handler(request):
return web.Response(text="Done")
@pytest.fixture
def app():
return web.Application()
@pytest.fixture
def adapter(app):
return ResourcesUrlDispatcherRouterAdapter(
app.router, defaults={"*": ResourceOptions()}
)
@pytest.fixture
def get_route(app):
return app.router.add_route("GET", "/get_path", _handler)
@pytest.fixture
def options_route(app):
return app.router.add_route("OPTIONS", "/options_path", _handler)
def test_add_get_route(adapter, get_route):
"""Test configuring GET route"""
result = adapter.add_preflight_handler(get_route.resource, _handler)
assert result is None
assert len(adapter._resource_config) == 0
assert len(adapter._resources_with_preflight_handlers) == 1
assert len(adapter._preflight_routes) == 1
def test_add_options_route(adapter, options_route):
"""Test configuring OPTIONS route"""
adapter.add_preflight_handler(options_route, _handler)
assert not adapter._resources_with_preflight_handlers
assert not adapter._preflight_routes
def test_get_non_preflight_request_config(adapter, get_route):
adapter.add_preflight_handler(get_route.resource, _handler)
adapter.set_config_for_routing_entity(
get_route.resource,
{
"http://example.org": ResourceOptions(),
},
)
adapter.add_preflight_handler(get_route, _handler)
adapter.set_config_for_routing_entity(
get_route,
{
"http://test.example.org": ResourceOptions(),
},
)
request = mock.Mock()
with mock.patch(
"aiohttp_cors.urldispatcher_router_adapter."
"ResourcesUrlDispatcherRouterAdapter."
"is_cors_enabled_on_request"
) as is_cors_enabled_on_request, mock.patch(
"aiohttp_cors.urldispatcher_router_adapter."
"ResourcesUrlDispatcherRouterAdapter."
"_request_resource"
) as _request_resource:
is_cors_enabled_on_request.return_value = True
_request_resource.return_value = get_route.resource
assert adapter.get_non_preflight_request_config(request) == {
"*": ResourceOptions(),
"http://example.org": ResourceOptions(),
}
request.method = "GET"
assert adapter.get_non_preflight_request_config(request) == {
"*": ResourceOptions(),
"http://example.org": ResourceOptions(),
"http://test.example.org": ResourceOptions(),
}
|
