fix: custom headers not setting (#1155)

Author: silentworks

supabase/_async/client.py

@@ -64,7 +64,10 @@ def __init__(
         self.supabase_url = supabase_url
         self.supabase_key = supabase_key
         self.options = copy.copy(options)
-        self.options.headers = copy.copy(self._get_auth_headers())
+        self.options.headers = {
+            **options.headers,
+            **self._get_auth_headers(),
+        }
 
         self.rest_url = f"{supabase_url}/rest/v1"
         self.realtime_url = f"{supabase_url}/realtime/v1".replace("http", "ws")

supabase/_sync/client.py

@@ -63,7 +63,10 @@ def __init__(
         self.supabase_url = supabase_url
         self.supabase_key = supabase_key
         self.options = copy.copy(options)
-        self.options.headers = copy.copy(self._get_auth_headers())
+        self.options.headers = {
+            **options.headers,
+            **self._get_auth_headers(),
+        }
 
         self.rest_url = f"{supabase_url}/rest/v1"
         self.realtime_url = f"{supabase_url}/realtime/v1".replace("http", "ws")

tests/test_client.py

@@ -132,3 +132,41 @@ def test_global_authorization_header_issue():
     client = create_client(url, key, options)
 
     assert client.options.headers.get("apiKey") == key
+
+
+def test_custom_headers():
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    options = ClientOptions(
+        headers={
+            "x-app-name": "apple",
+            "x-version": "1.0",
+        }
+    )
+
+    client = create_client(url, key, options)
+
+    assert client.options.headers.get("x-app-name") == "apple"
+    assert client.options.headers.get("x-version") == "1.0"
+
+
+def test_custom_headers_immutable():
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    options = ClientOptions(
+        headers={
+            "x-app-name": "apple",
+            "x-version": "1.0",
+        }
+    )
+
+    client1 = create_client(url, key, options)
+    client2 = create_client(url, key, options)
+
+    client1.options.headers["x-app-name"] = "grapes"
+
+    assert client1.options.headers.get("x-app-name") == "grapes"
+    assert client1.options.headers.get("x-version") == "1.0"
+    assert client2.options.headers.get("x-app-name") == "apple"