futurehouse-client 0.3.20.dev208__tar.gz → 0.3.20.dev215__tar.gz

{futurehouse_client-0.3.20.dev208/futurehouse_client.egg-info → futurehouse_client-0.3.20.dev215}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev208
+Version: 0.3.20.dev215
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License

{futurehouse_client-0.3.20.dev208 → futurehouse_client-0.3.20.dev215}/futurehouse_client/version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.20.dev208'
-__version_tuple__ = version_tuple = (0, 3, 20, 'dev208')
+__version__ = version = '0.3.20.dev215'
+__version_tuple__ = version_tuple = (0, 3, 20, 'dev215')

{futurehouse_client-0.3.20.dev208 → futurehouse_client-0.3.20.dev215/futurehouse_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: futurehouse-client
-Version: 0.3.20.dev208
+Version: 0.3.20.dev215
 Summary: A client for interacting with endpoints of the FutureHouse service.
 Author-email: FutureHouse technical staff <hello@futurehouse.org>
 License:                                  Apache License

{futurehouse_client-0.3.20.dev208 → futurehouse_client-0.3.20.dev215}/tests/test_rest.py

@@ -8,6 +8,7 @@ import time
 import types
 from pathlib import Path
 from unittest.mock import MagicMock, mock_open, patch
+from uuid import UUID, uuid4
 
 import pytest
 from futurehouse_client.clients import (
@@ -15,6 +16,7 @@ from futurehouse_client.clients import (
 )
 from futurehouse_client.clients.rest_client import (
     FileUploadError,
+    ProjectError,
     RestClient,
     RestClientError,
 )
@@ -745,3 +747,129 @@ def test_world_model_create_and_get(admin_client: RestClient):
     assert updated_model_by_id.name == model.name
     assert updated_model_by_id.content != model.content
     assert updated_model_by_name.content != model.content
+
+
+class TestProjectOperations:
+    @pytest.fixture
+    def test_project_name(self):
+        return f"test-project-{int(time.time())}"
+
+    def test_create_project_success(
+        self, admin_client: RestClient, test_project_name: str
+    ):
+        project_id = admin_client.create_project(test_project_name)
+        assert isinstance(project_id, str)
+        UUID(project_id)
+
+    def test_get_project_by_name_success(
+        self, admin_client: RestClient, test_project_name: str
+    ):
+        created_project_id = admin_client.create_project(test_project_name)
+        retrieved_project_id = admin_client.get_project_by_name(test_project_name)
+        assert str(retrieved_project_id) == created_project_id
+
+    def test_get_project_by_name_not_found(self, admin_client: RestClient):
+        with pytest.raises(ProjectError, match="No project found with name"):
+            admin_client.get_project_by_name("non-existent-project-12345")
+
+    def test_get_project_by_name_multiple_found(self, admin_client: RestClient):
+        with patch.object(admin_client.client, "get") as mock_get:
+            mock_response = MagicMock()
+            mock_response.raise_for_status.return_value = None
+            mock_response.json.return_value = [
+                {"id": "uuid1", "name": "test"},
+                {"id": "uuid2", "name": "test"},
+            ]
+            mock_get.return_value = mock_response
+
+            with pytest.raises(ProjectError, match="Multiple projects found"):
+                admin_client.get_project_by_name("test")
+
+    def test_add_task_to_project_success(
+        self, admin_client: RestClient, test_project_name: str, task_req: TaskRequest
+    ):
+        project_id = admin_client.create_project(test_project_name)
+        trajectory_id = admin_client.create_task(task_req)
+        admin_client.add_task_to_project(UUID(project_id), trajectory_id)
+
+    def test_add_task_to_project_not_found(self, admin_client: RestClient):
+        fake_project_id = uuid4()
+        fake_trajectory_id = str(uuid4())
+
+        with patch.object(admin_client.client, "post") as mock_post:
+            mock_response = MagicMock()
+            mock_response.raise_for_status.side_effect = Exception("404 Not Found")
+            mock_post.return_value = mock_response
+
+            with pytest.raises(
+                ProjectError, match="Error adding trajectory to project"
+            ):
+                admin_client.add_task_to_project(fake_project_id, fake_trajectory_id)
+
+    def test_create_project_with_http_error(self, admin_client: RestClient):
+        with patch.object(admin_client.client, "post") as mock_post:
+            mock_response = MagicMock()
+            mock_response.raise_for_status.side_effect = Exception("400 Bad Request")
+            mock_response.status_code = 400
+            mock_response.text = "Invalid project name"
+            mock_post.return_value = mock_response
+
+            with pytest.raises(ProjectError, match="Error creating project"):
+                admin_client.create_project("invalid project name")
+
+
+class TestAsyncProjectOperations:
+    @pytest.fixture
+    def test_project_name(self):
+        return f"test-async-project-{int(time.time())}"
+
+    @pytest.mark.asyncio
+    async def test_acreate_project_success(
+        self, admin_client: RestClient, test_project_name: str
+    ):
+        project_id = await admin_client.acreate_project(test_project_name)
+        assert isinstance(project_id, str)
+        UUID(project_id)
+
+    @pytest.mark.asyncio
+    async def test_aget_project_by_name_success(
+        self, admin_client: RestClient, test_project_name: str
+    ):
+        created_project_id = await admin_client.acreate_project(test_project_name)
+        retrieved_project_id = await admin_client.aget_project_by_name(
+            test_project_name
+        )
+        assert str(retrieved_project_id) == created_project_id
+
+    @pytest.mark.asyncio
+    async def test_aget_project_by_name_not_found(self, admin_client: RestClient):
+        with pytest.raises(ProjectError, match="No project found with name"):
+            await admin_client.aget_project_by_name("non-existent-async-project-12345")
+
+    @pytest.mark.asyncio
+    async def test_aadd_task_to_project_success(
+        self, admin_client: RestClient, test_project_name: str, task_req: TaskRequest
+    ):
+        project_id = await admin_client.acreate_project(test_project_name)
+        trajectory_id = await admin_client.acreate_task(task_req)
+        await admin_client.aadd_task_to_project(UUID(project_id), trajectory_id)
+
+    @pytest.mark.asyncio
+    async def test_aadd_task_to_project_permission_denied(
+        self, admin_client: RestClient
+    ):
+        fake_project_id = uuid4()
+        fake_trajectory_id = str(uuid4())
+
+        with patch.object(admin_client.async_client, "post") as mock_post:
+            mock_response = MagicMock()
+            mock_response.raise_for_status.side_effect = Exception("403 Forbidden")
+            mock_response.status_code = 403
+            mock_post.return_value = mock_response
+
+            with pytest.raises(
+                ProjectError, match="Error adding trajectory to project: 403 Forbidden"
+            ):
+                await admin_client.aadd_task_to_project(
+                    fake_project_id, fake_trajectory_id
+                )