Added support for object integrity level boundary

src/pytmv1/model/common.py

@@ -213,6 +213,23 @@ class EndpointActivity(BaseConsumable):
     tags: List[str] = Field(default=[])
     uuid: Optional[str] = None
 
+    @field_validator("object_integrity_level", mode="before")
+    @classmethod
+    def map_object_integrity_level(
+        cls, value: Optional[int]
+    ) -> Optional[IntegrityLevel]:
+        if value:
+            if IntegrityLevel.UNTRUSTED <= value < IntegrityLevel.LOW:
+                return IntegrityLevel.UNTRUSTED
+            if IntegrityLevel.LOW <= value < IntegrityLevel.MEDIUM:
+                return IntegrityLevel.LOW
+            if IntegrityLevel.MEDIUM <= value < IntegrityLevel.HIGH:
+                return IntegrityLevel.MEDIUM
+            if IntegrityLevel.HIGH <= value < IntegrityLevel.SYSTEM:
+                return IntegrityLevel.HIGH
+            return IntegrityLevel.SYSTEM
+        return None
+
 
 class HostInfo(BaseModel):
     name: str

tests/integration/test_search.py

@@ -1,6 +1,7 @@
 from pytmv1 import (
     GetEmailActivitiesCountResp,
     GetEndpointActivitiesCountResp,
+    IntegrityLevel,
     ListEmailActivityResp,
     ListEndpointActivityResp,
     ListEndpointDataResp,
@@ -45,6 +46,10 @@ def test_list_endpoint_activities(client):
     assert result.result_code == ResultCode.SUCCESS
     assert isinstance(result.response, ListEndpointActivityResp)
     assert len(result.response.items) > 0
+    assert (
+        result.response.items[0].object_integrity_level
+        == IntegrityLevel.MEDIUM
+    )
 
 
 def test_get_endpoint_activities_count(client):