superb-ai-onprem 0.1.6__py3-none-any.whl → 0.2.0__py3-none-any.whl

tests/activities/real_test.py

@@ -0,0 +1,66 @@
+from spb_onprem import (
+    DatasetService,
+    Dataset,
+    ActivityService,
+    Activity,
+    ActivityHistory,
+    ActivityStatus,
+    ActivitySchema,
+    SchemaType,
+)
+
+
+def test_activity_service():
+    dataset_service = DatasetService()
+    dataset = dataset_service.get_dataset(
+        dataset_id="01JPM6NR1APMBXJNC0YW72S1FN"
+    )
+
+    print(dataset)
+    
+    activity_service = ActivityService()
+    cursor = None
+    assign_data_slice_activity = None
+    while True:
+        (described_activities, cursor, _) = activity_service.get_activities(
+            dataset_id=dataset.id,
+            cursor=cursor,
+            length=50
+        )
+        for activity in described_activities:
+            if activity.activity_type == "ASSIGN_DATA_TO_SLICE":
+                assign_data_slice_activity = activity
+                break
+        if cursor is None:
+            break
+    
+    print(assign_data_slice_activity.parameter_schema)
+
+    activity_history = activity_service.start_activity(
+        dataset_id=dataset.id,
+        activity_id=assign_data_slice_activity.id,
+        parameters={
+            "dataset_id": dataset.id,
+            "slice_name": "sdk_test_slice",
+            "slice_description": "sdk_test_slice_description",
+            "filter": {
+                "must": {
+                    "keyContains": "validation_v5/87d37e96-0bab3a63.jpg"
+                }
+            },
+            "filter_type": "DATA",
+            "is_new_slice": True,
+        }
+    )
+    print(activity_history)
+    
+    activity_service.update_activity_history_status(
+        activity_history_id=activity_history.id,
+        status=ActivityStatus.FAILED,
+        meta={
+            "sdk_test": "sdk_test_value"
+        }
+    )
+
+if __name__ == "__main__":
+    test_activity_service()

tests/activities/test_params.py

@@ -0,0 +1,67 @@
+import pytest
+from spb_onprem.activities.params.update_activity_history import update_activity_history_params
+from spb_onprem.activities.entities import ActivityStatus
+from spb_onprem.exceptions import BadParameterError
+from spb_onprem.base_types import Undefined
+
+
+class TestUpdateActivityHistoryParams:
+    def test_update_activity_history_params_with_status(self):
+        # Given
+        activity_history_id = "test_history_id"
+        status = ActivityStatus.SUCCESS
+        meta = {"key": "value"}
+        
+        # When
+        params = update_activity_history_params(
+            activity_history_id=activity_history_id,
+            status=status,
+            meta=meta
+        )
+        
+        # Then
+        assert params["activity_history_id"] == activity_history_id
+        assert params["status"] == status
+        assert params["meta"] == meta
+        assert "progress" not in params
+    
+    def test_update_activity_history_params_with_progress(self):
+        # Given
+        activity_history_id = "test_history_id"
+        progress = {"current": 50, "total": 100}
+        
+        # When
+        params = update_activity_history_params(
+            activity_history_id=activity_history_id,
+            progress=progress
+        )
+        
+        # Then
+        assert params["activity_history_id"] == activity_history_id
+        assert params["progress"] == progress
+        assert "status" not in params
+        assert "meta" not in params
+    
+    def test_update_activity_history_params_without_required_fields(self):
+        # Given
+        activity_history_id = None
+        
+        # When/Then
+        with pytest.raises(BadParameterError) as exc_info:
+            update_activity_history_params(
+                activity_history_id=activity_history_id
+            )
+        assert str(exc_info.value) == "Activity history ID is required"
+    
+    def test_update_activity_history_params_without_status_or_progress(self):
+        # Given
+        activity_history_id = "test_history_id"
+        
+        # When/Then
+        with pytest.raises(BadParameterError) as exc_info:
+            update_activity_history_params(
+                activity_history_id=activity_history_id,
+                status=Undefined,
+                progress=Undefined
+            )
+        assert str(exc_info.value) == "Either status or progress must be provided" 

tests/activities/test_service.py

@@ -0,0 +1,139 @@
+import pytest
+from unittest.mock import MagicMock, patch
+
+from spb_onprem.activities.service import ActivityService
+from spb_onprem.activities.entities import Activity, ActivityHistory, ActivityStatus
+from spb_onprem.base_types import Undefined
+
+
+@pytest.fixture
+def activity_service():
+    return ActivityService()
+
+
+class TestActivityService:
+    def test_create_activity(self, activity_service):
+        # Given
+        mock_response = {
+            "createJob": {
+                "id": "test_id",
+                "type": "test_type",
+                "name": "test_name",
+                "description": "test_description"
+            }
+        }
+        activity_service.request_gql = MagicMock(return_value=mock_response)
+        
+        # When
+        activity = activity_service.create_activity(
+            activity_type="test_type",
+            name="test_name",
+            description="test_description"
+        )
+        
+        # Then
+        assert isinstance(activity, Activity)
+        assert activity.id == "test_id"
+        assert activity.type == "test_type"
+        assert activity.name == "test_name"
+        assert activity.description == "test_description"
+    
+    def test_get_activities(self, activity_service):
+        # Given
+        mock_response = {
+            "jobs": {
+                "activities": [
+                    {
+                        "id": "test_id_1",
+                        "type": "test_type",
+                        "name": "test_name_1"
+                    },
+                    {
+                        "id": "test_id_2",
+                        "type": "test_type",
+                        "name": "test_name_2"
+                    }
+                ],
+                "next": "next_cursor",
+                "totalCount": 2
+            }
+        }
+        activity_service.request_gql = MagicMock(return_value=mock_response)
+        
+        # When
+        activities, next_cursor, total_count = activity_service.get_activities(
+            dataset_id="test_dataset_id"
+        )
+        
+        # Then
+        assert len(activities) == 2
+        assert all(isinstance(activity, Activity) for activity in activities)
+        assert next_cursor == "next_cursor"
+        assert total_count == 2
+    
+    def test_start_activity(self, activity_service):
+        # Given
+        mock_response = {
+            "startJob": {
+                "id": "test_history_id",
+                "jobId": "test_activity_id",
+                "status": ActivityStatus.RUNNING.value
+            }
+        }
+        activity_service.request_gql = MagicMock(return_value=mock_response)
+        
+        # When
+        activity_history = activity_service.start_activity(
+            dataset_id="test_dataset_id",
+            activity_id="test_activity_id"
+        )
+        
+        # Then
+        assert isinstance(activity_history, ActivityHistory)
+        assert activity_history.id == "test_history_id"
+        assert activity_history.activity_id == "test_activity_id"
+        assert activity_history.status == ActivityStatus.RUNNING
+    
+    def test_update_activity_history_status(self, activity_service):
+        # Given
+        mock_response = {
+            "updateJobHistory": {
+                "id": "test_history_id",
+                "jobId": "test_activity_id",
+                "status": ActivityStatus.SUCCESS.value
+            }
+        }
+        activity_service.request_gql = MagicMock(return_value=mock_response)
+        
+        # When
+        activity_history = activity_service.update_activity_history_status(
+            activity_history_id="test_history_id",
+            status=ActivityStatus.SUCCESS
+        )
+        
+        # Then
+        assert isinstance(activity_history, ActivityHistory)
+        assert activity_history.id == "test_history_id"
+        assert activity_history.status == ActivityStatus.SUCCESS
+    
+    def test_update_activity_history_progress(self, activity_service):
+        # Given
+        mock_response = {
+            "updateJobHistory": {
+                "id": "test_history_id",
+                "jobId": "test_activity_id",
+                "progress": {"current": 50, "total": 100}
+            }
+        }
+        activity_service.request_gql = MagicMock(return_value=mock_response)
+        
+        # When
+        activity_history = activity_service.update_activity_history_progress(
+            activity_history_id="test_history_id",
+            progress={"current": 50, "total": 100}
+        )
+        
+        # Then
+        assert isinstance(activity_history, ActivityHistory)
+        assert activity_history.id == "test_history_id"
+        assert activity_history.progress == {"current": 50, "total": 100}