Repository navigation
Expand file tree
/
Copy pathblocking_httpserver.py
More file actions
180 lines (145 loc) · 7.49 KB
/
Copy pathblocking_httpserver.py
File metadata and controls
180 lines (145 loc) · 7.49 KB
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
from __future__ import annotations
from queue import Empty
from queue import Queue
from typing import TYPE_CHECKING
from typing import Any
from pytest_httpserver.httpserver import HVMATCHER_T
from pytest_httpserver.httpserver import METHOD_ALL
from pytest_httpserver.httpserver import UNDEFINED
from pytest_httpserver.httpserver import HTTPServerBase
from pytest_httpserver.httpserver import QueryMatcher
from pytest_httpserver.httpserver import RequestHandlerBase
from pytest_httpserver.httpserver import URIPattern
if TYPE_CHECKING:
from collections.abc import Mapping
from re import Pattern
from ssl import SSLContext
from werkzeug import Request
from werkzeug import Response
class BlockingRequestHandler(RequestHandlerBase):
"""
Provides responding to a request synchronously.
This class should only be instantiated inside the implementation of the :py:class:`BlockingHTTPServer`.
"""
def __init__(self) -> None:
self.response_queue: Queue[Response] = Queue()
def respond_with_response(self, response: Response) -> None:
self.response_queue.put_nowait(response)
class BlockingHTTPServer(HTTPServerBase):
"""
Server instance which enables synchronous matching for incoming requests.
:param host: the host or IP where the server will listen
:param port: the TCP port where the server will listen
:param ssl_context: the ssl context object to use for https connections
:param timeout: waiting time in seconds for matching and responding to an incoming request.
manager
.. py:attribute:: no_handler_status_code
Attribute containing the http status code (int) which will be the response
status when no matcher is found for the request. By default, it is set to *500*
but it can be overridden to any valid http status code such as *404* if needed.
"""
DEFAULT_LISTEN_HOST = "localhost"
DEFAULT_LISTEN_PORT = 0 # Use ephemeral port
def __init__(
self,
host: str = DEFAULT_LISTEN_HOST,
port: int = DEFAULT_LISTEN_PORT,
ssl_context: SSLContext | None = None,
timeout: int = 30,
) -> None:
super().__init__(host, port, ssl_context)
self.timeout = timeout
self.request_queue: Queue[Request] = Queue()
self.request_handlers: dict[Request, Queue[BlockingRequestHandler]] = {}
def assert_request(
self,
uri: str | URIPattern | Pattern[str],
method: str = METHOD_ALL,
data: str | bytes | None = None,
data_encoding: str = "utf-8",
headers: Mapping[str, str] | None = None,
query_string: QueryMatcher | str | bytes | Mapping[str, str] | None = None,
header_value_matcher: HVMATCHER_T | None = None,
json: Any = UNDEFINED,
timeout: int = 30,
data_form: Mapping[str, str] | None = None,
) -> BlockingRequestHandler:
"""
Wait for an incoming request and check whether it matches according to the given parameters.
If the incoming request matches, a request handler is created and registered,
otherwise assertion error is raised.
The request handler can be used once to respond for the request.
If no response is performed in the period given in the timeout parameter of the constructor
or no request arrives in the `timeout` period, assertion error is raised.
:param uri: URI of the request. This must be an absolute path starting with ``/``, a
:py:class:`URIPattern` object, or a regular expression compiled by :py:func:`re.compile`.
:param method: HTTP method of the request. If not specified (or `METHOD_ALL`
specified), all HTTP requests will match.
:param data: payload of the HTTP request. This could be a string (utf-8 encoded
by default, see `data_encoding`) or a bytes object.
:param data_encoding: the encoding used for data parameter if data is a string.
:param headers: dictionary of the headers of the request to be matched
:param query_string: the http query string, after ``?``, such as ``username=user``.
If string is specified it will be encoded to bytes with the encode method of
the string. If dict is specified, it will be matched to the ``key=value`` pairs
specified in the request. If multiple values specified for a given key, the first
value will be used. If multiple values needed to be handled, use ``MultiDict``
object from werkzeug.
:param header_value_matcher: callable that matches values of headers.
:param json: a python object (eg. a dict) whose value will be compared to the request body after it
is loaded as json. If load fails, this matcher will be failed also. *Content-Type* is not checked.
If that's desired, add it to the headers parameter.
:param timeout: waiting time in seconds for an incoming request.
:param data_form: expected form fields, parsed by werkzeug using the request's *Content-Type*.
A mapping matches the first value for each field; a ``MultiDict`` matches all values.
Mutually exclusive with `data` and `json`.
:return: Created and registered :py:class:`BlockingRequestHandler`.
Parameters `json`, `data`, and `data_form` are mutually exclusive.
"""
matcher = self.create_matcher(
uri,
method=method.upper(),
data=data,
data_encoding=data_encoding,
headers=headers,
query_string=query_string,
header_value_matcher=header_value_matcher,
json=json,
**({"data_form": data_form} if data_form is not None else {}),
)
try:
request = self.request_queue.get(timeout=timeout)
except Empty:
raise AssertionError(f"Waiting for request {matcher} timed out") # noqa: EM102
diff = matcher.difference(request)
request_handler = BlockingRequestHandler()
self.request_handlers[request].put_nowait(request_handler)
if diff:
request_handler.respond_with_response(self.respond_nohandler(request))
raise AssertionError(f"Request {matcher} does not match: {diff}") # noqa: EM102
return request_handler
def dispatch(self, request: Request) -> Response:
"""
Dispatch a request for synchronous matching.
This method queues the request for matching and waits for the request handler.
If there was no request handler, error is responded,
otherwise it waits for the response of request handler.
If no response arrives, assertion error is raised, otherwise the response is returned.
:param request: the request object from the werkzeug library.
:return: the response object what the handler responded, or a response which contains the error.
"""
self.request_handlers[request] = Queue()
try:
self.request_queue.put_nowait(request)
try:
request_handler = self.request_handlers[request].get(timeout=self.timeout)
except Empty:
return self.respond_nohandler(request)
try:
return request_handler.response_queue.get(timeout=self.timeout)
except Empty:
assertion = AssertionError(f"No response for request: {request}")
self.add_assertion(assertion)
raise assertion
finally:
del self.request_handlers[request]