Compare commits
2
Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9aa84cc2c3 | ||
|
|
658174518a |
+1
-1
@@ -28,7 +28,7 @@ dev = [
|
||||
|
||||
[tool.ruff.lint]
|
||||
select = ["ALL"]
|
||||
ignore = ["D", "E501", "S104", "PLR2004", "ANN401", "BLE001", "COM812", "S603", "PLR0911", "S101", "PT"]
|
||||
ignore = ["D", "E501", "S104", "PLR2004", "ANN401", "BLE001", "COM812", "S603", "PLR0911", "S101", "PT", "E722"]
|
||||
|
||||
[tool.coverage.run]
|
||||
source = ["stapler"]
|
||||
|
||||
@@ -73,6 +73,10 @@ class BaseHandler(abc.ABC, http.server.BaseHTTPRequestHandler):
|
||||
else:
|
||||
self.send_status_only(code, message)
|
||||
|
||||
@typing.override
|
||||
def address_string(self) -> str: # pragma: no cover
|
||||
return re.sub(r"[^\x00-\x7F]+", "", super().address_string())
|
||||
|
||||
@typing.override
|
||||
def log_message(self, format: str, *args: typing.Any) -> None: # pragma: no cover
|
||||
fmt = "%s - " + format
|
||||
@@ -275,7 +279,10 @@ class RequestHandler(http.server.SimpleHTTPRequestHandler, BaseHandler):
|
||||
self.__target_redirect: str | None = None
|
||||
self.__target_proxy: str | None = None
|
||||
self.__target_spa: str | None = None
|
||||
try:
|
||||
super().__init__(*args, directory=params.data_dir, **kwargs, params=params) # ty:ignore[unknown-argument]
|
||||
except:
|
||||
self.logger.exception("Could not handle request")
|
||||
|
||||
@property
|
||||
def token(self) -> str:
|
||||
|
||||
@@ -162,6 +162,21 @@ class TestRequestHandler(BaseHandlerTestCase):
|
||||
handler.data_dir = self.data_dir
|
||||
return handler
|
||||
|
||||
def test_handle_errors_silently(self) -> None:
|
||||
with self.patch("http.server.BaseHTTPRequestHandler.__init__") as mock:
|
||||
mock.side_effect = Exception
|
||||
logging.basicConfig(level=logging.CRITICAL)
|
||||
RequestHandler(
|
||||
unittest.mock.MagicMock(),
|
||||
"127.0.0.1",
|
||||
unittest.mock.MagicMock(),
|
||||
params=Parameters(
|
||||
data_dir=self.get_tmp_dir(), certbot_www=str(self.certbot_www)
|
||||
),
|
||||
registry=self.registry,
|
||||
token_manager=self.token_manager,
|
||||
)
|
||||
|
||||
def test_do_head_forward(self) -> None:
|
||||
handler = self._get_handler()
|
||||
with (
|
||||
@@ -1253,6 +1268,7 @@ class TestUpgradeHandler(BaseHandlerTestCase):
|
||||
handler.headers = collections.defaultdict(lambda: None, headers) # ty:ignore[invalid-assignment]
|
||||
handler.rfile = rfile if rfile is not None else io.BytesIO()
|
||||
handler.wfile = io.BytesIO()
|
||||
handler.client_address = ("127.0.0.1", 12345)
|
||||
handler.logger = unittest.mock.Mock(logging.Logger)
|
||||
handler.data_dir = self.data_dir
|
||||
return handler
|
||||
|
||||
Reference in New Issue
Block a user