feat: Add Project collaborator tests

Author: vamaq

tests/future/conftest.py

@@ -22,6 +22,7 @@
 from httpx import Request, Response
 
 from galileo.__future__ import Configuration
+from galileo.__future__.collaborator import CollaboratorRole
 from galileo.__future__.configuration import _CONFIGURATION_KEYS
 from galileo.config import GalileoPythonConfig
 from galileo.resources.models.messages_list_item import MessagesListItem
@@ -223,3 +224,18 @@ def mock_integration() -> MagicMock:
     mock_int.is_selected = True
     mock_int.permissions = ["read", "update"]
     return mock_int
+
+
+@pytest.fixture
+def mock_collaborator() -> MagicMock:
+    """Create a mock collaborator object for testing."""
+    mock_collab = MagicMock()
+    mock_collab.id = str(uuid4())
+    mock_collab.user_id = str(uuid4())
+    mock_collab.role = CollaboratorRole.VIEWER
+    mock_collab.created_at = None
+    mock_collab.email = "collaborator@test.com"
+    mock_collab.first_name = "Test"
+    mock_collab.last_name = "Collaborator"
+    mock_collab.permissions = []
+    return mock_collab

tests/future/test_project.py

@@ -4,6 +4,7 @@
 import pytest
 
 from galileo.__future__ import Project
+from galileo.__future__.collaborator import Collaborator, CollaboratorRole
 from galileo.__future__.shared.base import SyncState
 from galileo.__future__.shared.exceptions import APIError, ValidationError
 
@@ -302,3 +303,61 @@ def test_repr_representation(self, reset_configuration: None) -> None:
 
         assert "Test Project" in repr(project)
         assert "test-id-123" in repr(project)
+
+
+class TestProjectCollaborators:
+    """Test suite for Project collaborator management methods."""
+
+    @patch("galileo.__future__.project.Projects")
+    def test_add_update_remove_collaborator(
+        self,
+        mock_projects_class: MagicMock,
+        reset_configuration: None,
+        mock_project: MagicMock,
+        mock_collaborator: MagicMock,
+    ) -> None:
+        """Test add, update, and remove collaborator return Collaborator instances."""
+        mock_service = MagicMock()
+        mock_projects_class.return_value = mock_service
+        mock_service.get.return_value = mock_project
+        mock_service.share_project_with_user.return_value = mock_collaborator
+        mock_service.update_user_project_collaborator.return_value = mock_collaborator
+
+        project = Project.get(id=mock_project.id)
+
+        added = project.add_collaborator(user_id=mock_collaborator.user_id)
+        assert isinstance(added, Collaborator)
+
+        updated = project.update_collaborator(user_id=mock_collaborator.user_id, role=CollaboratorRole.EDITOR)
+        assert isinstance(updated, Collaborator)
+
+        project.remove_collaborator(user_id=mock_collaborator.user_id)
+        mock_service.unshare_project_with_user.assert_called_once()
+
+    @patch("galileo.__future__.project.Projects")
+    def test_collaborators_property_returns_same_as_list_method(
+        self,
+        mock_projects_class: MagicMock,
+        reset_configuration: None,
+        mock_project: MagicMock,
+        mock_collaborator: MagicMock,
+    ) -> None:
+        """Test collaborators property returns equivalent results to list_collaborators()."""
+        mock_service = MagicMock()
+        mock_projects_class.return_value = mock_service
+        mock_service.get.return_value = mock_project
+        mock_service.list_user_project_collaborators.return_value = [mock_collaborator]
+
+        project = Project.get(id=mock_project.id)
+
+        assert project.collaborators[0] == project.list_collaborators()[0]
+
+    def test_collaborator_methods_require_synced_project(self, reset_configuration: None) -> None:
+        """Test collaborator methods raise ValueError for local-only projects."""
+        project = Project(name="Test Project")
+
+        with pytest.raises(ValueError, match="Project ID is not set"):
+            project.list_collaborators()
+
+        with pytest.raises(ValueError, match="Project ID is not set"):
+            project.add_collaborator(user_id="user-123")