Add typing and async support for SSO.

tests/conftest.py

@@ -1,4 +1,4 @@
-from typing import Mapping, Union
+from typing import Mapping, Optional, Union
 from unittest.mock import AsyncMock, MagicMock
 
 import httpx
@@ -128,9 +128,9 @@ def mock(*args, **kwargs):
 def mock_http_client_with_response(monkeypatch):
     def inner(
         http_client: Union[SyncHTTPClient, AsyncHTTPClient],
-        response_dict: dict,
+        response_dict: Optional[dict] = None,
         status_code: int = 200,
-        headers: Mapping[str, str] = None,
+        headers: Optional[Mapping[str, str]] = None,
     ):
         mock_class = (
             AsyncMock if isinstance(http_client, AsyncHTTPClient) else MagicMock
@@ -145,6 +145,39 @@ def inner(
     return inner
 
 
+@pytest.fixture
+def capture_and_mock_http_client_request(monkeypatch):
+    def inner(
+        http_client: Union[SyncHTTPClient, AsyncHTTPClient],
+        response_dict: dict,
+        status_code: int = 200,
+        headers: Optional[Mapping[str, str]] = None,
+    ):
+        request_args = []
+        request_kwargs = {}
+
+        def capture_and_mock(*args, **kwargs):
+            request_args.extend(args)
+            request_kwargs.update(kwargs)
+
+            return httpx.Response(
+                status_code=status_code,
+                headers=headers,
+                json=response_dict,
+            )
+
+        mock_class = (
+            AsyncMock if isinstance(http_client, AsyncHTTPClient) else MagicMock
+        )
+        mock = mock_class(side_effect=capture_and_mock)
+
+        monkeypatch.setattr(http_client._client, "request", mock)
+
+        return (request_args, request_kwargs)
+
+    return inner
+
+
 @pytest.fixture
 def mock_pagination_request_for_http_client(monkeypatch):
     # Mocking pagination correctly requires us to index into a list of data
@@ -153,7 +186,7 @@ def inner(
         http_client: Union[SyncHTTPClient, AsyncHTTPClient],
         data_list: list,
         status_code: int = 200,
-        headers: Mapping[str, str] = None,
+        headers: Optional[Mapping[str, str]] = None,
     ):
         # For convenient index lookup, store the list of object IDs.
         data_ids = list(map(lambda x: x["id"], data_list))

tests/test_client.py

@@ -171,3 +171,38 @@ def test_initialize_events_missing_api_key(self):
         message = str(ex)
 
         assert "api_key" in message
+
+    def test_initialize_sso(self, set_api_key_and_client_id):
+        assert bool(async_client.sso)
+
+    def test_initialize_sso_missing_api_key(self, set_client_id):
+        with pytest.raises(ConfigurationException) as ex:
+            async_client.sso
+
+        message = str(ex)
+
+        assert "api_key" in message
+        assert "client_id" not in message
+
+    def test_initialize_sso_missing_client_id(self, set_api_key):
+        with pytest.raises(ConfigurationException) as ex:
+            async_client.sso
+
+        message = str(ex)
+
+        assert "client_id" in message
+        assert "api_key" not in message
+
+    def test_initialize_sso_missing_api_key_and_client_id(self):
+        with pytest.raises(ConfigurationException) as ex:
+            async_client.sso
+
+        message = str(ex)
+
+        assert all(
+            setting in message
+            for setting in (
+                "api_key",
+                "client_id",
+            )
+        )

tests/test_directory_sync.py

@@ -197,7 +197,6 @@ def test_delete_directory(self, mock_http_client_with_response):
         mock_http_client_with_response(
             http_client=self.http_client,
             status_code=202,
-            response_dict=None,
             headers={"content-type": "text/plain; charset=utf-8"},
         )
 
@@ -432,7 +431,6 @@ async def test_delete_directory(self, mock_http_client_with_response):
         mock_http_client_with_response(
             http_client=self.http_client,
             status_code=202,
-            response_dict=None,
             headers={"content-type": "text/plain; charset=utf-8"},
         )
 

tests/test_organizations.py

@@ -1,11 +1,8 @@
 import datetime
-from typing import Dict, List, Union, cast
 
 import pytest
-import requests
 
-from tests.conftest import MockResponse
-from tests.utils.list_resource import list_data_to_dicts, list_response_of
+from tests.utils.list_resource import list_data_to_dicts
 from workos.organizations import Organizations
 from tests.utils.fixtures.mock_organization import MockOrganization