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)