Add intent_options_sso_bookmark_slug parameter on Portal.generate_link() method

Author: jonatascastro12

tests/test_portal.py

@@ -13,13 +13,26 @@ def setup(self, sync_http_client_for_test):
     def mock_portal_link(self):
         return {"link": "https://id.workos.com/portal/launch?secret=secret"}
 
-    def test_generate_link_sso(self, mock_portal_link, mock_http_client_with_response):
-        mock_http_client_with_response(self.http_client, mock_portal_link, 201)
+    def test_generate_link_sso(
+        self, mock_portal_link, capture_and_mock_http_client_request
+    ):
+        request_kwargs = capture_and_mock_http_client_request(
+            self.http_client, mock_portal_link, 201
+        )
 
         response = self.portal.generate_link(
-            intent="sso", organization_id="org_01EHQMYV6MBK39QC5PZXHY59C3"
+            intent="sso",
+            organization_id="org_01EHQMYV6MBK39QC5PZXHY59C3",
+            intent_options_sso_bookmark_slug="my_app",
         )
 
+        assert request_kwargs["url"].endswith("/portal/generate_link")
+        assert request_kwargs["method"] == "post"
+        assert request_kwargs["json"] == {
+            "intent": "sso",
+            "organization": "org_01EHQMYV6MBK39QC5PZXHY59C3",
+            "intent_options": {"sso": {"bookmark_slug": "my_app"}},
+        }
         assert response.link == "https://id.workos.com/portal/launch?secret=secret"
 
     def test_generate_link_domain_verification(

workos/portal.py

@@ -1,4 +1,4 @@
-from typing import Optional, Protocol
+from typing import Optional, Protocol, Dict, Literal
 from workos.types.portal.portal_link import PortalLink
 from workos.types.portal.portal_link_intent import PortalLinkIntent
 from workos.utils.http_client import SyncHTTPClient
@@ -16,6 +16,7 @@ def generate_link(
         organization_id: str,
         return_url: Optional[str] = None,
         success_url: Optional[str] = None,
+        intent_options_sso_bookmark_slug: Optional[str] = None,
     ) -> PortalLink:
         """Generate a link to grant access to an organization's Admin Portal
 
@@ -47,13 +48,22 @@ def generate_link(
         organization_id: str,
         return_url: Optional[str] = None,
         success_url: Optional[str] = None,
+        intent_options_sso_bookmark_slug: Optional[str] = None,
     ) -> PortalLink:
-        json = {
+        json: dict[str, str | None | dict[str, dict[str, str]]] = {
             "intent": intent,
             "organization": organization_id,
             "return_url": return_url,
             "success_url": success_url,
         }
+
+        if intent_options_sso_bookmark_slug:
+            json["intent_options"] = {
+                "sso": {
+                    "bookmark_slug": intent_options_sso_bookmark_slug,
+                }
+            }
+
         response = self._http_client.request(
             PORTAL_GENERATE_PATH, method=REQUEST_METHOD_POST, json=json
         )