Add custom_attributes field to SSO Profile. (#434)

Author: mattgd

tests/test_sso.py

@@ -29,7 +29,7 @@ def mock_magic_link_profile(self):
             idp_id="",
             first_name=None,
             last_name=None,
-            profile=None,
+            role=None,
             groups=None,
             raw_attributes={},
         ).dict()

tests/utils/fixtures/mock_profile.py

@@ -6,7 +6,7 @@ class MockProfile(Profile):
     def __init__(self, id: str):
         super().__init__(
             object="profile",
-            id="prof_01DWAS7ZQWM70PV93BFV1V78QV",
+            id=id or "prof_01DWAS7ZQWM70PV93BFV1V78QV",
             email="[email protected]",
             first_name="WorkOS",
             last_name="Demo",
@@ -16,10 +16,14 @@ def __init__(self, id: str):
             connection_id="conn_01EMH8WAK20T42N2NBMNBCYHAG",
             connection_type="OktaSAML",
             idp_id="00u1klkowm8EGah2H357",
+            custom_attributes={
+                "license": "professional",
+            },
             raw_attributes={
                 "email": "[email protected]",
                 "first_name": "WorkOS",
                 "last_name": "Demo",
                 "groups": ["Admins", "Developers"],
+                "license": "professional",
             },
         )

workos/types/sso/profile.py

@@ -23,7 +23,8 @@ class Profile(WorkOSModel):
     idp_id: str
     role: Optional[ProfileRole] = None
     groups: Optional[Sequence[str]] = None
-    raw_attributes: Mapping[str, Any]
+    custom_attributes: Optional[Mapping[str, Any]] = None
+    raw_attributes: Optional[Mapping[str, Any]] = None
 
 
 class ProfileAndToken(WorkOSModel):