📝 Add docstrings to test_mentorship_backend

backend/apps/mentorship/api/internal/nodes/mentee.py

@@ -26,5 +26,10 @@ def resolve_avatar_url(self) -> str:
 
     @strawberry.field(name="experienceLevel")
     def resolve_experience_level(self) -> str:
-        """Get the experience level of the mentee."""
-        return self.experience_level if self.experience_level else "beginner"
+        """
+        Return the mentee's experience level, defaulting to "beginner" when not set.
+
+        Returns:
+            experience_level (str): The mentee's experience level, or "beginner" if the field is empty or falsy.
+        """
+        return self.experience_level if self.experience_level else "beginner"

backend/tests/apps/mentorship/api/internal/nodes/test_api_internal_mentor.py

@@ -0,0 +1,91 @@
+"""Pytest for mentorship mentor nodes."""
+
+from unittest.mock import MagicMock
+
+import pytest
+
+from apps.mentorship.api.internal.nodes.mentor import MentorNode
+
+
+@pytest.fixture
+def mock_github_user():
+    """
+    Create a MagicMock that simulates a GitHub user for tests.
+
+    The mock has the following attributes set:
+    - `avatar_url`: "https://example.com/mentor_avatar.jpg"
+    - `name`: "Mentor Name"
+    - `login`: "mentorlogin"
+
+    Returns:
+        MagicMock: A mock object representing a GitHub user with the above attributes.
+    """
+    mock = MagicMock()
+    mock.avatar_url = "https://example.com/mentor_avatar.jpg"
+    mock.name = "Mentor Name"
+    mock.login = "mentorlogin"
+    return mock
+
+
+@pytest.fixture
+def mock_mentor_node(mock_github_user):
+    """
+    Create a MentorNode with its `github_user` set to the provided mock.
+
+    Parameters:
+        mock_github_user: Mocked GitHub user object (e.g., MagicMock) providing attributes like `avatar_url`, `name`, and `login`.
+
+    Returns:
+        MentorNode: A MentorNode with id `"1"` and `github_user` assigned to the provided mock.
+    """
+    mentor_node = MentorNode(id="1")
+    mentor_node.github_user = mock_github_user
+    return mentor_node
+
+
+@pytest.fixture
+def mock_mentor_node_no_github_user():
+    """
+    Create a MentorNode test fixture with no associated GitHub user.
+
+    Returns:
+        MentorNode: a MentorNode with id "2" and github_user set to None.
+    """
+    mentor_node = MentorNode(id="2")
+    mentor_node.github_user = None
+    return mentor_node
+
+
+def test_mentor_node_id(mock_mentor_node):
+    """Test that MentorNode id is correctly assigned."""
+    assert str(mock_mentor_node.id) == "1"
+
+
+def test_mentor_node_avatar_url(mock_mentor_node):
+    """Test the avatar_url field resolver."""
+    assert mock_mentor_node.avatar_url() == "https://example.com/mentor_avatar.jpg"
+
+
+def test_mentor_node_avatar_url_no_github_user(mock_mentor_node_no_github_user):
+    """Test avatar_url when no github_user is associated."""
+    assert mock_mentor_node_no_github_user.avatar_url() == ""
+
+
+def test_mentor_node_name(mock_mentor_node):
+    """Test the name field resolver."""
+    assert mock_mentor_node.name() == "Mentor Name"
+
+
+def test_mentor_node_name_no_github_user(mock_mentor_node_no_github_user):
+    """Test name when no github_user is associated."""
+    assert mock_mentor_node_no_github_user.name() == ""
+
+
+def test_mentor_node_login(mock_mentor_node):
+    """Test the login field resolver."""
+    assert mock_mentor_node.login() == "mentorlogin"
+
+
+def test_mentor_node_login_no_github_user(mock_mentor_node_no_github_user):
+    """Test login when no github_user is associated."""
+    assert mock_mentor_node_no_github_user.login() == ""