fix: allow patch operations on resources and extensions path root

Author: azmeuk

doc/changelog.rst

@@ -7,6 +7,7 @@ Changelog
 Fixed
 ^^^^^
 - Attributes with ``None`` type are excluded from Schema generation.
+- Allow PATCH operations on resources and extensions root path.
 
 [0.4.2] - 2025-08-05
 --------------------

scim2_models/messages/patch_op.py

@@ -334,9 +334,18 @@ def _set_value_at_path(
         """Set a value at a specific path."""
         target, attr_path = _resolve_path_to_target(resource, path)
 
-        if not attr_path or not target:
+        if not target:
             raise ValueError(Error.make_invalid_path_error().detail)
 
+        if not attr_path:
+            if not isinstance(value, dict):
+                raise ValueError(Error.make_invalid_path_error().detail)
+
+            updated_data = {**target.model_dump(), **value}
+            updated_target = type(target).model_validate(updated_data)
+            target.__dict__.update(updated_target.__dict__)
+            return True
+
         path_parts = attr_path.split(".")
         if len(path_parts) == 1:
             return cls._set_simple_attribute(target, path_parts[0], value, is_add)

scim2_models/urn.py

@@ -29,6 +29,16 @@ def _normalize_path(model: Optional[type["BaseModel"]], path: str) -> tuple[str,
 
     # Absolute URN
     if ":" in path:
+        if (
+            model
+            and issubclass(model, Resource)
+            and (
+                path in model.get_extension_models()
+                or path == model.model_fields["schemas"].default[0]
+            )
+        ):
+            return path, ""
+
         parts = path.rsplit(":", 1)
         return parts[0], parts[1]
 
@@ -100,12 +110,13 @@ def _resolve_path_to_target(
     if not schema_urn:
         return resource, attr_path
 
+    if extension_class := resource.get_extension_model(schema_urn):
+        extension_instance = _get_or_create_extension_instance(
+            resource, extension_class
+        )
+        return extension_instance, attr_path
+
     if schema_urn in resource.schemas:
         return resource, attr_path
 
-    extension_class = resource.get_extension_model(schema_urn)
-    if not extension_class:
-        return (None, "")
-
-    extension_instance = _get_or_create_extension_instance(resource, extension_class)
-    return extension_instance, attr_path
+    return (None, "")

tests/test_patch_op_extensions.py

@@ -333,3 +333,76 @@ def test_complex_object_creation_and_basemodel_matching():
 
     result = patch.patch(group)
     assert result is True
+
+
+def test_patch_extension_schema_path_without_attribute():
+    """Test PATCH with extension schema URN as path (no specific attribute)."""
+    user = User[EnterpriseUser](
+        user_name="test",
+        schemas=[
+            "urn:ietf:params:scim:schemas:core:2.0:User",
+            "urn:ietf:params:scim:schemas:extension:enterprise:2.0:User",
+        ],
+    )
+    user[EnterpriseUser] = EnterpriseUser()
+
+    patch = PatchOp[User](
+        operations=[
+            PatchOperation(
+                op=PatchOperation.Op.add,
+                path="urn:ietf:params:scim:schemas:extension:enterprise:2.0:User",
+                value={
+                    "costCenter": "Engineering",
+                    "department": "IT",
+                    "employeeNumber": "12345",
+                },
+            )
+        ]
+    )
+
+    result = patch.patch(user)
+    assert result is True
+    assert user[EnterpriseUser].cost_center == "Engineering"
+
+
+def test_patch_main_schema_path_without_attribute():
+    """Test PATCH with main schema URN as path (no specific attribute)."""
+    user = User(user_name="original")
+
+    patch = PatchOp[User](
+        operations=[
+            PatchOperation(
+                op=PatchOperation.Op.add,
+                path="urn:ietf:params:scim:schemas:core:2.0:User",
+                value={
+                    "displayName": "Updated Name",
+                    "nickName": "Nick",
+                    "title": "Manager",
+                },
+            )
+        ]
+    )
+
+    result = patch.patch(user)
+    assert result is True
+    assert user.display_name == "Updated Name"
+    assert user.nick_name == "Nick"
+    assert user.title == "Manager"
+
+
+def test_patch_schema_path_with_invalid_value_type():
+    """Test PATCH with schema URN path and invalid value type (non-dict)."""
+    user = User(user_name="test")
+
+    patch = PatchOp[User](
+        operations=[
+            PatchOperation(
+                op=PatchOperation.Op.add,
+                path="urn:ietf:params:scim:schemas:core:2.0:User",
+                value="invalid string value",
+            )
+        ]
+    )
+
+    with pytest.raises(ValueError, match="path.*invalid.*malformed"):
+        patch.patch(user)