pygeai 0.5.0__py3-none-any.whl → 0.6.0b3__py3-none-any.whl

pygeai/tests/lab/processes/test_mappers.py

@@ -14,9 +14,9 @@ class TestAgenticProcessMapper(unittest.TestCase):
     """
 
     def setUp(self):
-        self.client = AgenticProcessClient()
-        self.client.api_service = MagicMock()
         self.project_id = "test-project-id"
+        self.client = AgenticProcessClient(api_key="test_key", base_url="https://test.url", project_id=self.project_id)
+        self.client.api_service = MagicMock()
         self.process_id = "test-process-id"
         self.process_name = "test-process-name"
         self.task_id = "test-task-id"
@@ -124,7 +124,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.put.return_value = self.mock_response
 
         result = self.client.update_process(
-            project_id=self.project_id,
             name="Test Process",
             key="updated-key",
             automatic_publish=False,
@@ -138,7 +137,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.put.return_value = self.mock_response
 
         result = self.client.update_process(
-            project_id=self.project_id,
             process_id=self.process_id,
             name="Upserted Process",
             automatic_publish=True,
@@ -156,7 +154,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.update_process(
-                project_id=self.project_id,
                 process_id=self.process_id,
                 name="Updated Process"
             )
@@ -169,7 +166,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_process(
-            project_id=self.project_id,
             process_name="Test Process",
             revision="1",
             version=1,
@@ -185,7 +181,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_process(
-                project_id=self.project_id,
                 process_id=self.process_id
             )
         except InvalidAPIResponseException as e:
@@ -198,9 +193,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         try:
-            result = self.client.list_processes(
-                project_id=self.project_id
-            )
+            result = self.client.list_processes()
         except InvalidAPIResponseException as e:
             result = self.mock_response.text
 
@@ -210,7 +203,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.list_process_instances(
-            project_id=self.project_id,
             process_id=self.process_id,
             is_active=False,
             start="5",
@@ -226,7 +218,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.list_process_instances(
-                project_id=self.project_id,
                 process_id=self.process_id
             )
         except InvalidAPIResponseException as e:
@@ -235,15 +226,15 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.assertEqual(result, "success text")
 
     def test_delete_process_success_with_name(self):
+        self.mock_response.status_code = 204
         self.client.api_service.delete.return_value = self.mock_response
 
         result = self.client.delete_process(
-            project_id=self.project_id,
             process_name="Test Process"
         )
 
         self.client.api_service.delete.assert_called_once()
-        self.assertEqual(result, {"status": "success"})
+        self.assertEqual(result, {})
 
     def test_delete_process_json_decode_error(self):
         self.mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
@@ -251,7 +242,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.delete_process(
-                project_id=self.project_id,
                 process_id=self.process_id
             )
         except InvalidAPIResponseException as e:
@@ -263,7 +253,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.publish_process_revision(
-            project_id=self.project_id,
             process_name="Test Process",
             revision="2"
         )
@@ -274,7 +263,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_publish_process_revision_missing_identifier(self):
         with self.assertRaises(ValueError) as context:
             self.client.publish_process_revision(
-                project_id=self.project_id,
                 revision="2"
             )
 
@@ -286,7 +274,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.publish_process_revision(
-                project_id=self.project_id,
                 process_id=self.process_id,
                 revision="2"
             )
@@ -301,7 +288,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.create_task(
-                project_id=self.project_id,
                 name="Test Task"
             )
         except InvalidAPIResponseException as e:
@@ -313,7 +299,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.create_task(
-            project_id=self.project_id,
             name="Test Task",
             description="Task Description",
             title_template="Task for {{issue}}",
@@ -330,7 +315,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_task(
-            project_id=self.project_id,
             task_id="",
             task_name="Test Task"
         )
@@ -344,7 +328,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_task(
-                project_id=self.project_id,
                 task_id=self.task_id
             )
         except InvalidAPIResponseException as e:
@@ -356,7 +339,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.list_tasks(
-            project_id=self.project_id,
             id="task1",
             start="10",
             count="50",
@@ -371,9 +353,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         try:
-            result = self.client.list_tasks(
-                project_id=self.project_id
-            )
+            result = self.client.list_tasks()
         except InvalidAPIResponseException as e:
             result = self.mock_response.text
 
@@ -382,7 +362,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_update_task_missing_id(self):
         with self.assertRaises(ValueError) as context:
             self.client.update_task(
-                project_id=self.project_id,
                 task_id=""
             )
 
@@ -392,7 +371,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.put.return_value = self.mock_response
 
         result = self.client.update_task(
-            project_id=self.project_id,
             task_id=self.task_id,
             name="Updated Task",
             description="Updated Description",
@@ -407,7 +385,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.put.return_value = self.mock_response
 
         result = self.client.update_task(
-            project_id=self.project_id,
             task_id=self.task_id,
             name="Upserted Task",
             upsert=True
@@ -424,7 +401,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.update_task(
-                project_id=self.project_id,
                 task_id=self.task_id,
                 name="Updated Task"
             )
@@ -434,16 +410,16 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.assertEqual(result, "success text")
 
     def test_delete_task_success_with_name(self):
+        self.mock_response.status_code = 204
         self.client.api_service.delete.return_value = self.mock_response
 
         result = self.client.delete_task(
-            project_id=self.project_id,
             task_id="",
             task_name="Test Task"
         )
 
         self.client.api_service.delete.assert_called_once()
-        self.assertEqual(result, {"status": "success"})
+        self.assertEqual(result, {})
 
     def test_delete_task_json_decode_error(self):
         self.mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
@@ -451,7 +427,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.delete_task(
-                project_id=self.project_id,
                 task_id=self.task_id
             )
         except InvalidAPIResponseException as e:
@@ -462,7 +437,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_delete_task_missing_identifier(self):
         with self.assertRaises(ValueError) as context:
             self.client.delete_task(
-                project_id=self.project_id,
                 task_id="",
                 task_name=""
             )
@@ -473,7 +447,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.publish_task_revision(
-            project_id=self.project_id,
             task_id="",
             task_name="Test Task",
             revision="3"
@@ -485,7 +458,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_publish_task_revision_missing_identifier(self):
         with self.assertRaises(ValueError) as context:
             self.client.publish_task_revision(
-                project_id=self.project_id,
                 task_id="",  # Explicitly empty to trigger the ValueError
                 task_name="",  # Explicitly empty to trigger the ValueError
                 revision="3"
@@ -496,7 +468,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_publish_task_revision_missing_revision(self):
         with self.assertRaises(ValueError) as context:
             self.client.publish_task_revision(
-                project_id=self.project_id,
                 task_id=self.task_id
             )
 
@@ -508,7 +479,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.publish_task_revision(
-                project_id=self.project_id,
                 task_id=self.task_id,
                 revision="3"
             )
@@ -523,7 +493,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.start_instance(
-                project_id=self.project_id,
                 process_name="Test Process"
             )
         except InvalidAPIResponseException as e:
@@ -535,7 +504,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.start_instance(
-            project_id=self.project_id,
             process_name="Test Process",
             subject="Test Subject",
             variables=[{"key": "var1", "value": "value1"}]
@@ -550,7 +518,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.abort_instance(
-            project_id=self.project_id,
             instance_id=self.instance_id
         )
 
@@ -563,7 +530,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.abort_instance(
-                project_id=self.project_id,
                 instance_id=self.instance_id
             )
         except InvalidAPIResponseException as e:
@@ -575,7 +541,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_instance(
-            project_id=self.project_id,
             instance_id=self.instance_id
         )
 
@@ -588,7 +553,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_instance(
-                project_id=self.project_id,
                 instance_id=self.instance_id
             )
         except InvalidAPIResponseException as e:
@@ -599,7 +563,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_get_instance_missing_id(self):
         with self.assertRaises(ValueError) as context:
             self.client.get_instance(
-                project_id=self.project_id,
                 instance_id=""
             )
 
@@ -609,7 +572,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_instance_history(
-            project_id=self.project_id,
             instance_id=self.instance_id
         )
 
@@ -622,7 +584,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_instance_history(
-                project_id=self.project_id,
                 instance_id=self.instance_id
             )
         except InvalidAPIResponseException as e:
@@ -633,7 +594,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_get_instance_history_missing_id(self):
         with self.assertRaises(ValueError) as context:
             self.client.get_instance_history(
-                project_id=self.project_id,
                 instance_id=""
             )
 
@@ -643,7 +603,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_thread_information(
-            project_id=self.project_id,
             thread_id=self.thread_id
         )
 
@@ -656,7 +615,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_thread_information(
-                project_id=self.project_id,
                 thread_id=self.thread_id
             )
         except InvalidAPIResponseException as e:
@@ -667,7 +625,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_get_thread_information_missing_id(self):
         with self.assertRaises(ValueError) as context:
             self.client.get_thread_information(
-                project_id=self.project_id,
                 thread_id=""
             )
 
@@ -677,7 +634,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.send_user_signal(
-            project_id=self.project_id,
             instance_id=self.instance_id,
             signal_name="approval"
         )
@@ -691,7 +647,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.send_user_signal(
-                project_id=self.project_id,
                 instance_id=self.instance_id,
                 signal_name="approval"
             )
@@ -703,7 +658,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_send_user_signal_missing_instance_id(self):
         with self.assertRaises(ValueError) as context:
             self.client.send_user_signal(
-                project_id=self.project_id,
                 instance_id="",
                 signal_name="approval"
             )
@@ -716,7 +670,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.create_kb(
-                project_id=self.project_id,
                 name="Test KB"
             )
         except InvalidAPIResponseException as e:
@@ -728,7 +681,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.post.return_value = self.mock_response
 
         result = self.client.create_kb(
-            project_id=self.project_id,
             name="Test KB",
             artifacts=["art1", "art2"],
             metadata=["meta1", "meta2"]
@@ -744,7 +696,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.get_kb(
-            project_id=self.project_id,
             kb_id="",
             kb_name="Test KB"
         )
@@ -758,7 +709,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.get_kb(
-                project_id=self.project_id,
                 kb_id=self.kb_id
             )
         except InvalidAPIResponseException as e:
@@ -769,7 +719,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_get_kb_missing_identifier(self):
         with self.assertRaises(ValueError) as context:
             self.client.get_kb(
-                project_id=self.project_id,
                 kb_id="",
                 kb_name=""
             )
@@ -780,7 +729,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.list_kbs(
-            project_id=self.project_id,
             name="Test KB",
             start="5",
             count="25"
@@ -794,25 +742,23 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         try:
-            result = self.client.list_kbs(
-                project_id=self.project_id
-            )
+            result = self.client.list_kbs()
         except InvalidAPIResponseException as e:
             result = self.mock_response.text
 
         self.assertEqual(result, "success text")
 
     def test_delete_kb_success_with_name(self):
+        self.mock_response.status_code = 204
         self.client.api_service.delete.return_value = self.mock_response
 
         result = self.client.delete_kb(
-            project_id=self.project_id,
             kb_id="",
             kb_name="Test KB"
         )
 
         self.client.api_service.delete.assert_called_once()
-        self.assertEqual(result, {"status": "success"})
+        self.assertEqual(result, {})
 
     def test_delete_kb_json_decode_error(self):
         self.mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
@@ -820,7 +766,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
 
         try:
             result = self.client.delete_kb(
-                project_id=self.project_id,
                 kb_id=self.kb_id
             )
         except InvalidAPIResponseException as e:
@@ -831,7 +776,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
     def test_delete_kb_missing_identifier(self):
         with self.assertRaises(ValueError) as context:
             self.client.delete_kb(
-                project_id=self.project_id,
                 kb_id="",
                 kb_name=""
             )
@@ -843,9 +787,7 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         try:
-            result = self.client.list_jobs(
-                project_id=self.project_id
-            )
+            result = self.client.list_jobs()
         except InvalidAPIResponseException as e:
             result = self.mock_response.text
 
@@ -855,7 +797,6 @@ class TestAgenticProcessMapper(unittest.TestCase):
         self.client.api_service.get.return_value = self.mock_response
 
         result = self.client.list_jobs(
-            project_id=self.project_id,
             start="10",
             count="30",
             topic="test-topic",