From 68ec6e8f2c26fef93557bada983536d51ddeaf96 Mon Sep 17 00:00:00 2001
From: T-256 <132141463+T-256@users.noreply.github.com>
Date: Tue, 16 Jan 2024 11:18:44 +0330
Subject: [PATCH] test `same_origin` via public api

---
 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 5391f9c22d..313266394e 100644
--- a/tests/test_utils.py
+++ b/tests/test_utils.py
@@ -14,7 +14,6 @@
     is_https_redirect,
     obfuscate_sensitive_headers,
     parse_header_links,
-    same_origin,
 )
 
 from .common import TESTS_DIR
@@ -222,15 +221,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():