def test_body_response(self) -> None:
e = HttpRequestRejected(
status_code=httpStatusCodes.NOT_FOUND, reason=b'NOT FOUND',
body=b'Nothing here')
self.assertEqual(
e.response(self.request),
build_http_response(httpStatusCodes.NOT_FOUND, reason=b'NOT FOUND', body=b'Nothing here'))
def test_status_code_response(self) -> None:
e = HttpRequestRejected(status_code=200, reason=b'OK')
self.assertEqual(
e.response(self.request),
CRLF.join([
b'HTTP/1.1 200 OK',
b'Connection: close',
CRLF,
]),
)
def before_upstream_connection(
self,
request: HttpParser,
) -> Optional[HttpParser]:
assert not self.flags.unix_socket_path and self.client.addr
if self.client.addr[0] in '127.0.0.1,::1'.split(','):
raise HttpRequestRejected(
status_code=httpStatusCodes.I_AM_A_TEAPOT,
reason=b'I\'m a tea pot',
)
return request
def test_empty_response(self) -> None:
e = HttpRequestRejected()
self.assertEqual(e.response(self.request), None)