Update test_client.py

Author: Sanidhyafeaturist

tests/test_client.py

@@ -1,12 +1,13 @@
 from __future__ import annotations
 
 import os
+import asyncio
 from typing import Any
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, AsyncMock
 
 import pytest
 
-from supabase import Client, ClientOptions, create_client
+from supabase import AsyncClient, ClientOptions, create_client
 
 
 @pytest.mark.xfail(
@@ -16,7 +17,7 @@
 @pytest.mark.parametrize("key", ["", None, "valeefgpoqwjgpj", 139, -1, {}, []])
 def test_incorrect_values_dont_instantiate_client(url: Any, key: Any) -> None:
     """Ensure we can't instantiate client with invalid values."""
-    _: Client = create_client(url, key)
+    _: AsyncClient = create_client(url, key)
 
 
 def test_uses_key_as_authorization_header_by_default() -> None:
@@ -91,3 +92,50 @@ 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
+
+
+@pytest.mark.asyncio
+async def test_connect_to_realtime_success(mocker):
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    client = create_client(url, key)
+    mock_connect = mocker.patch.object(client.realtime, 'connect', return_value=None)
+
+    await client.connect_to_realtime()
+    mock_connect.assert_called_once()
+
+
+@pytest.mark.asyncio
+async def test_connect_to_realtime_failure(mocker):
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    client = create_client(url, key)
+    mock_connect = mocker.patch.object(client.realtime, 'connect', side_effect=Exception("Connection failed"))
+
+    with pytest.raises(Exception, match="Connection failed"):
+        await client.connect_to_realtime()
+
+
+@pytest.mark.asyncio
+async def test_reconnect_logic(mocker):
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    client = create_client(url, key)
+    mock_reconnect = mocker.patch.object(client.realtime, 'connect', side_effect=[Exception("Failed"), None])
+
+    await client.connect_to_realtime()  # First attempt fails
+    await client.connect_to_realtime()  # Second attempt should succeed
+    assert mock_reconnect.call_count == 2  # Check that it tried to reconnect
+
+
+def test_logging_on_connection_failure(caplog):
+    url = os.environ.get("SUPABASE_TEST_URL")
+    key = os.environ.get("SUPABASE_TEST_KEY")
+
+    client = create_client(url, key)
+    with caplog.at_level(logging.ERROR):
+        await client.connect_to_realtime()  # Assume it fails
+    assert "Connection failed" in caplog.text