From 395d624e120643d51fd4693346bafce40fe6a48a Mon Sep 17 00:00:00 2001 From: David Robertson Date: Sat, 11 Jun 2022 19:45:35 +0100 Subject: Suppress checking of a mock --- tests/utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'tests/utils.py') diff --git a/tests/utils.py b/tests/utils.py index d3da559200..abdc0eaadd 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -221,10 +221,15 @@ def default_config( return config_dict -def mock_getRawHeaders(headers=None): +def mock_getRawHeaders(headers=None): # type: ignore[no-untyped-def] headers = headers if headers is not None else {} - def getRawHeaders(name, default=None): + def getRawHeaders(name, default=None): # type: ignore[no-untyped-def] + # If the requested header is present, the real twisted function returns + # List[str] if name is a str and List[bytes] if name is a bytes. + # This mock doesn't support that behaviour. + # Fortunately, none of the current callers of mock_getRawHeaders() provide a + # headers dict, so we don't encounter this discrepancy in practice. return headers.get(name, default) return getRawHeaders -- cgit 1.5.1