Improved testing for backend/app/mentorship

backend/apps/github/models/__init__.py

@@ -2,6 +2,7 @@
 
 from .comment import Comment
 from .commit import Commit
+from .issue import Issue
 from .label import Label
 from .milestone import Milestone
 from .pull_request import PullRequest

backend/pyproject.toml

@@ -154,7 +154,6 @@ omit = [
   "**/admin.py",
   "**/apps.py",
   "**/migrations/*",
-  "**/mentorship/*", # TODO: work in progress
   "manage.py",
   "settings/*",
   "tests/*",

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

@@ -0,0 +1,21 @@
+from apps.mentorship.api.internal.nodes.enum import ExperienceLevelEnum, ProgramStatusEnum
+from apps.mentorship.models import Program
+from apps.mentorship.models.common.experience_level import ExperienceLevel
+
+
+def test_experience_level_enum_values():
+    """Test that ExperienceLevelEnum maps correctly to model choices."""
+    assert ExperienceLevelEnum.BEGINNER.value == ExperienceLevel.ExperienceLevelChoices.BEGINNER
+    assert (
+        ExperienceLevelEnum.INTERMEDIATE.value
+        == ExperienceLevel.ExperienceLevelChoices.INTERMEDIATE
+    )
+    assert ExperienceLevelEnum.ADVANCED.value == ExperienceLevel.ExperienceLevelChoices.ADVANCED
+    assert ExperienceLevelEnum.EXPERT.value == ExperienceLevel.ExperienceLevelChoices.EXPERT
+
+
+def test_program_status_enum_values():
+    """Test that ProgramStatusEnum maps correctly to model choices."""
+    assert ProgramStatusEnum.DRAFT.value == Program.ProgramStatus.DRAFT
+    assert ProgramStatusEnum.PUBLISHED.value == Program.ProgramStatus.PUBLISHED
+    assert ProgramStatusEnum.COMPLETED.value == Program.ProgramStatus.COMPLETED

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

@@ -0,0 +1,56 @@
+import pytest
+
+from apps.mentorship.api.internal.nodes.enum import ExperienceLevelEnum
+from apps.mentorship.api.internal.nodes.mentee import MenteeNode
+
+
+@pytest.fixture
+def mock_mentee_node():
+    """Fixture for a mock MenteeNode instance."""
+    return MenteeNode(
+        id="1",
+        login="test_mentee",
+        name="Test Mentee",
+        avatar_url="https://example.com/avatar.jpg",
+        bio="A test mentee",
+        experience_level=ExperienceLevelEnum.BEGINNER,
+        domains=["python"],
+        tags=["backend"],
+    )
+
+
+def test_mentee_node_fields(mock_mentee_node):
+    """Test that MenteeNode fields are correctly assigned."""
+    assert mock_mentee_node.id == "1"
+    assert mock_mentee_node.login == "test_mentee"
+    assert mock_mentee_node.name == "Test Mentee"
+    assert mock_mentee_node.avatar_url == "https://example.com/avatar.jpg"
+    assert mock_mentee_node.bio == "A test mentee"
+    assert mock_mentee_node.experience_level == ExperienceLevelEnum.BEGINNER
+    assert mock_mentee_node.domains == ["python"]
+    assert mock_mentee_node.tags == ["backend"]
+
+
+def test_mentee_node_resolve_avatar_url(mock_mentee_node):
+    """Test the resolve_avatar_url method."""
+    assert mock_mentee_node.resolve_avatar_url() == "https://example.com/avatar.jpg"
+
+
+def test_mentee_node_resolve_experience_level(mock_mentee_node):
+    """Test the resolve_experience_level method."""
+    assert mock_mentee_node.resolve_experience_level() == ExperienceLevelEnum.BEGINNER
+
+
+def test_mentee_node_resolve_experience_level_none():
+    """Test the resolve_experience_level method when experience_level is None."""
+    mentee_node_no_exp = MenteeNode(
+        id="2",
+        login="no_exp_mentee",
+        name="No Experience Mentee",
+        avatar_url="https://example.com/noexp.jpg",
+        bio=None,
+        experience_level=None,  # type: ignore[assignment]
+        domains=None,
+        tags=None,
+    )
+    assert mentee_node_no_exp.resolve_experience_level() == "beginner"

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

@@ -0,0 +1,66 @@
+from unittest.mock import MagicMock
+
+import pytest
+
+from apps.mentorship.api.internal.nodes.mentor import MentorNode
+
+
+@pytest.fixture
+def mock_github_user():
+    """Fixture for a mock GithubUser."""
+    mock = MagicMock()
+    mock.avatar_url = "https://example.com/mentor_avatar.jpg"
+    mock.name = "Mentor Name"
+    mock.login = "mentor_login"
+    return mock
+
+
+@pytest.fixture
+def mock_mentor_node(mock_github_user):
+    """Fixture for a mock MentorNode instance."""
+    mentor_node = MentorNode(id="1")
+    mentor_node.github_user = mock_github_user
+    return mentor_node
+
+
+@pytest.fixture
+def mock_mentor_node_no_github_user():
+    """Fixture for a mock MentorNode instance without a GitHub user."""
+    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() == "mentor_login"
+
+
+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() == ""