From: T-256 <132141463+T-256@users.noreply.github.com>
Date: Fri, 23 Feb 2024 14:16:03 +0000 (+0330)
Subject: test `same_origin` via public api (#3062)
X-Git-Tag: 0.27.1~28
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=fc84f7f6eb77fe5d4428261c837ac8016ec77a28;p=thirdparty%2Fhttpx.git

test `same_origin` via public api (#3062)

Co-authored-by: Tom Christie <tom@tomchristie.com>
---

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():