fix: async set_auth for realtime in auth event listener (#930)

Makefile

@@ -17,3 +17,4 @@ tests_only:
 
 build_sync:
 	poetry run unasync supabase tests
+	sed -i 's/asyncio.create_task(self.realtime.set_auth(access_token))//g' supabase/_sync/client.py

pyproject.toml

@@ -38,6 +38,10 @@ tests = 'poetry_scripts:run_tests'
 
 [tool.poetry.group.dev.dependencies]
 unasync-cli = { git = "https://github.com/supabase-community/unasync-cli.git", branch = "main" }
+pytest-asyncio = "^0.24.0"
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]

supabase/_async/client.py

@@ -1,3 +1,4 @@
+import asyncio
 import re
 from typing import Any, Dict, List, Optional, Union
 
@@ -296,9 +297,7 @@ def _listen_to_auth_events(
             access_token = session.access_token if session else self.supabase_key
 
         self.options.headers["Authorization"] = self._create_auth_header(access_token)
-
-        # set_auth is a coroutine, how to handle this?
-        self.realtime.set_auth(access_token)
+        asyncio.create_task(self.realtime.set_auth(access_token))
 
 
 async def create_client(

supabase/_sync/client.py

@@ -297,9 +297,6 @@ def _listen_to_auth_events(
 
         self.options.headers["Authorization"] = self._create_auth_header(access_token)
 
-        # set_auth is a coroutine, how to handle this?
-        self.realtime.set_auth(access_token)
-
 
 def create_client(
     supabase_url: str,

tests/_async/test_client.py

@@ -0,0 +1,38 @@
+import os
+from unittest.mock import AsyncMock, MagicMock
+
+from supabase import create_async_client
+
+
+async def test_updates_the_authorization_header_on_auth_events() -> None:
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    client = await create_async_client(url, key)
+
+    assert client.options.headers.get("apiKey") == key
+    assert client.options.headers.get("Authorization") == f"Bearer {key}"
+
+    mock_session = MagicMock(access_token="secretuserjwt")
+    realtime_mock = AsyncMock()
+    client.realtime = realtime_mock
+
+    client._listen_to_auth_events("SIGNED_IN", mock_session)
+
+    updated_authorization = f"Bearer {mock_session.access_token}"
+
+    assert client.options.headers.get("apiKey") == key
+    assert client.options.headers.get("Authorization") == updated_authorization
+
+    assert client.postgrest.session.headers.get("apiKey") == key
+    assert (
+        client.postgrest.session.headers.get("Authorization") == updated_authorization
+    )
+
+    assert client.auth._headers.get("apiKey") == key
+    assert client.auth._headers.get("Authorization") == updated_authorization
+
+    assert client.storage.session.headers.get("apiKey") == key
+    assert client.storage.session.headers.get("Authorization") == updated_authorization
+
+    realtime_mock.set_auth.assert_called_once_with(mock_session.access_token)

tests/test_client.py

@@ -92,5 +92,3 @@ def test_updates_the_authorization_header_on_auth_events() -> None:
 
     assert client.storage.session.headers.get("apiKey") == key
     assert client.storage.session.headers.get("Authorization") == updated_authorization
-
-    realtime_mock.set_auth.assert_called_once_with(mock_session.access_token)