diff options
author | bashonly <88596187+bashonly@users.noreply.github.com> | 2024-05-04 17:19:42 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-04 22:19:42 +0000 |
commit | bec9a59e8ec82c18e3bf9268eaa436793dd52e35 (patch) | |
tree | 181c9801cd2ae5edda75c59cde2ab9d86d89312b /test/test_networking.py | |
parent | 036e0d92c6052465673d459678322ea03e61483d (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.py | 19 |
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""" |