aboutsummaryrefslogtreecommitdiff
path: root/test/test_networking.py
diff options
context:
space:
mode:
authorbashonly <88596187+bashonly@users.noreply.github.com>2024-05-04 17:19:42 -0500
committerGitHub <noreply@github.com>2024-05-04 22:19:42 +0000
commitbec9a59e8ec82c18e3bf9268eaa436793dd52e35 (patch)
tree181c9801cd2ae5edda75c59cde2ab9d86d89312b /test/test_networking.py
parent036e0d92c6052465673d459678322ea03e61483d (diff)
[networking] Add `extensions` attribute to `Response` (#9756)
CurlCFFIRH now provides an `impersonate` field in its responses' extensions Authored by: bashonly
Diffstat (limited to 'test/test_networking.py')
-rw-r--r--test/test_networking.py19
1 files changed, 19 insertions, 0 deletions
diff --git a/test/test_networking.py b/test/test_networking.py
index b50f70d08..d613cb568 100644
--- a/test/test_networking.py
+++ b/test/test_networking.py
@@ -785,6 +785,25 @@ class TestHTTPImpersonateRequestHandler(TestRequestHandlerBase):
assert res.status == 200
assert std_headers['user-agent'].lower() not in res.read().decode().lower()
+ def test_response_extensions(self, handler):
+ with handler() as rh:
+ for target in rh.supported_targets:
+ request = Request(
+ f'http://127.0.0.1:{self.http_port}/gen_200', extensions={'impersonate': target})
+ res = validate_and_send(rh, request)
+ assert res.extensions['impersonate'] == rh._get_request_target(request)
+
+ def test_http_error_response_extensions(self, handler):
+ with handler() as rh:
+ for target in rh.supported_targets:
+ request = Request(
+ f'http://127.0.0.1:{self.http_port}/gen_404', extensions={'impersonate': target})
+ try:
+ validate_and_send(rh, request)
+ except HTTPError as e:
+ res = e.response
+ assert res.extensions['impersonate'] == rh._get_request_target(request)
+
class TestRequestHandlerMisc:
"""Misc generic tests for request handlers, not related to request or validation testing"""