From fc84f7f6eb77fe5d4428261c837ac8016ec77a28 Mon Sep 17 00:00:00 2001 From: T-256 <132141463+T-256@users.noreply.github.com> Date: Fri, 23 Feb 2024 17:46:03 +0330 Subject: [PATCH] test `same_origin` via public api (#3062) Co-authored-by: Tom Christie --- tests/test_utils.py | 21 ++++++++++++++------- 1 file changed, 14 insertions(+), 7 deletions(-) diff --git a/tests/test_utils.py b/tests/test_utils.py index 2341a7c7..f98a18f2 100644 --- a/tests/test_utils.py +++ b/tests/test_utils.py @@ -11,7 +11,6 @@ from httpx._utils import ( URLPattern, get_ca_bundle_from_env, get_environment_proxies, - same_origin, ) from .common import TESTS_DIR @@ -224,15 +223,23 @@ def test_obfuscate_sensitive_headers(headers, output): def test_same_origin(): - origin1 = httpx.URL("https://example.com") - origin2 = httpx.URL("HTTPS://EXAMPLE.COM:443") - assert same_origin(origin1, origin2) + origin = httpx.URL("https://example.com") + request = httpx.Request("GET", "HTTPS://EXAMPLE.COM:443") + + client = httpx.Client() + headers = client._redirect_headers(request, origin, "GET") + + assert headers["Host"] == request.url.netloc.decode("ascii") def test_not_same_origin(): - origin1 = httpx.URL("https://example.com") - origin2 = httpx.URL("HTTP://EXAMPLE.COM") - assert not same_origin(origin1, origin2) + origin = httpx.URL("https://example.com") + request = httpx.Request("GET", "HTTP://EXAMPLE.COM:80") + + client = httpx.Client() + headers = client._redirect_headers(request, origin, "GET") + + assert headers["Host"] == origin.netloc.decode("ascii") def test_is_https_redirect(): -- 2.47.3