feat: add support for extensions

Author: azmeuk

doc/changelog.rst

@@ -1,6 +1,13 @@
 Changelog
 =========
 
+[0.1.9] - Unreleased
+--------------------
+
+Added
+^^^^^
+- Added support for :class:`Extensions <scim2_models.Extension>`.
+
 [0.1.8] - 2024-11-29
 --------------------
 

scim2_tester/resource.py

@@ -6,6 +6,8 @@
 
 from pydantic import EmailStr
 from scim2_models import ComplexAttribute
+from scim2_models import EnterpriseUser
+from scim2_models import Extension
 from scim2_models import Group
 from scim2_models import Meta
 from scim2_models import Reference
@@ -20,9 +22,13 @@
 from scim2_tester.utils import checker
 
 
-def model_from_resource_type(resource_type: ResourceType):
+def model_from_resource_type(conf: CheckConfig, resource_type: ResourceType):
     if resource_type.id == "User":
-        return User
+        return (
+            User[EnterpriseUser]
+            if User[EnterpriseUser] in conf.client.resource_models
+            else User
+        )
 
     if resource_type.id == "Group":
         return Group
@@ -59,6 +65,10 @@ def fill_with_random_values(obj) -> Resource:
             value = field_type()
             fill_with_random_values(value)
 
+        elif isclass(field_type) and issubclass(field_type, Extension):
+            value = field_type()
+            fill_with_random_values(value)
+
         else:
             value = str(uuid.uuid4())
 
@@ -67,6 +77,7 @@ def fill_with_random_values(obj) -> Resource:
 
         else:
             setattr(obj, field_name, value)
+
     return obj
 
 
@@ -181,7 +192,7 @@ def check_resource_type(
 ) -> list[CheckResult]:
     results = []
 
-    model = model_from_resource_type(resource_type)
+    model = model_from_resource_type(conf, resource_type)
     obj = model()
     fill_with_random_values(obj)
 

tests/conftest.py

@@ -1,5 +1,6 @@
 import pytest
 from scim2_client.engines.werkzeug import TestSCIMClient
+from scim2_models import EnterpriseUser
 from scim2_models import Group
 from scim2_models import User
 from scim2_server.backend import InMemoryBackend
@@ -8,7 +9,6 @@
 from scim2_server.utils import load_default_schemas
 
 from scim2_tester import check_server
-from scim2_tester.utils import Status
 
 TestSCIMClient.__test__ = False
 
@@ -28,6 +28,5 @@ def scim2_server():
 
 
 def test_scim2_server(scim2_server):
-    scim = TestSCIMClient(scim2_server, resource_models=(User, Group))
-    results = check_server(scim)
-    assert all(result.status == Status.SUCCESS for result in results)
+    scim = TestSCIMClient(scim2_server, resource_models=(User[EnterpriseUser], Group))
+    check_server(scim, raise_exceptions=True)