Add tests

Author: ahmedxgouda

backend/apps/owasp/graphql/nodes/project_health_metrics.py

@@ -9,16 +9,11 @@
 @strawberry_django.type(
     ProjectHealthMetrics,
     fields=[
-        "age_days",
         "contributors_count",
         "forks_count",
         "is_funding_requirements_compliant",
         "is_project_leaders_requirements_compliant",
-        "last_commit_days",
-        "last_pull_request_days",
-        "last_release_days",
         "open_issues_count",
-        "owasp_page_last_update_days",
         "recent_releases_count",
         "score",
         "stars_count",
@@ -29,6 +24,31 @@
 class ProjectHealthMetricsNode:
     """Project health metrics node."""
 
+    @strawberry.field
+    def age_days(self) -> int:
+        """Resolve project age in days."""
+        return self.age_days
+
+    @strawberry.field
+    def last_commit_days(self) -> int:
+        """Resolve last commit age in days."""
+        return self.last_commit_days
+
+    @strawberry.field
+    def last_pull_request_days(self) -> int:
+        """Resolve last pull request age in days."""
+        return self.last_pull_request_days
+
+    @strawberry.field
+    def last_release_days(self) -> int:
+        """Resolve last release age in days."""
+        return self.last_release_days
+
+    @strawberry.field
+    def owasp_page_last_update_days(self) -> int:
+        """Resolve OWASP page last update age in days."""
+        return self.owasp_page_last_update_days
+
     @strawberry.field
     def project_name(self) -> str:
         """Resolve project node."""

backend/tests/apps/owasp/graphql/nodes/project_health_metrics_test.py

@@ -0,0 +1,67 @@
+"""Test cases for ProjectHealthMetricsNode."""
+
+import pytest
+
+from apps.owasp.graphql.nodes.project_health_metrics import ProjectHealthMetricsNode
+
+
+class TestProjectHealthMetricsNode:
+    def test_project_health_metrics_node_inheritance(self):
+        assert hasattr(ProjectHealthMetricsNode, "__strawberry_definition__")
+
+    def test_meta_configuration(self):
+        field_names = {
+            field.name for field in ProjectHealthMetricsNode.__strawberry_definition__.fields
+        }
+        expected_field_names = {
+            "age_days",
+            "contributors_count",
+            "forks_count",
+            "is_funding_requirements_compliant",
+            "is_project_leaders_requirements_compliant",
+            "last_commit_days",
+            "last_pull_request_days",
+            "last_release_days",
+            "open_issues_count",
+            "owasp_page_last_update_days",
+            "recent_releases_count",
+            "score",
+            "stars_count",
+            "unanswered_issues_count",
+            "unassigned_issues_count",
+        }
+        assert expected_field_names.issubset(field_names)
+
+    def _get_field_by_name(self, name):
+        return next(
+            (
+                f
+                for f in ProjectHealthMetricsNode.__strawberry_definition__.fields
+                if f.name == name
+            ),
+            None,
+        )
+
+    @pytest.mark.parametrize(
+        ("field_name", "expected_type"),
+        [
+            ("project_name", str),
+            ("age_days", int),
+            ("contributors_count", int),
+            ("forks_count", int),
+            ("is_funding_requirements_compliant", bool),
+            ("is_project_leaders_requirements_compliant", bool),
+            ("last_commit_days", int),
+            ("last_pull_request_days", int),
+            ("last_release_days", int),
+            ("open_issues_count", int),
+            ("owasp_page_last_update_days", int),
+            ("recent_releases_count", int),
+            ("unanswered_issues_count", int),
+            ("unassigned_issues_count", int),
+        ],
+    )
+    def test_field_types(self, field_name, expected_type):
+        field = self._get_field_by_name(field_name)
+        assert field is not None
+        assert field.type is expected_type

backend/tests/apps/owasp/graphql/nodes/project_test.py

@@ -6,6 +6,7 @@
 from apps.github.graphql.nodes.release import ReleaseNode
 from apps.github.graphql.nodes.repository import RepositoryNode
 from apps.owasp.graphql.nodes.project import ProjectNode
+from apps.owasp.graphql.nodes.project_health_metrics import ProjectHealthMetricsNode
 
 
 class TestProjectNode:
@@ -43,6 +44,11 @@ def _get_field_by_name(self, name):
             (f for f in ProjectNode.__strawberry_definition__.fields if f.name == name), None
         )
 
+    def test_resolve_health(self):
+        field = self._get_field_by_name("health")
+        assert field is not None
+        assert field.type.of_type is ProjectHealthMetricsNode
+
     def test_resolve_issues_count(self):
         field = self._get_field_by_name("issues_count")
         assert field is not None

backend/tests/apps/owasp/graphql/queries/project_health_metrics_test.py

@@ -0,0 +1,64 @@
+from unittest.mock import Mock, patch
+
+from apps.owasp.graphql.nodes.project_health_metrics import ProjectHealthMetricsNode
+from apps.owasp.graphql.queries.project_health_metrics import ProjectHealthMetricsQuery
+from apps.owasp.models.project_health_metrics import ProjectHealthMetrics
+
+
+class ProjectHealthMetricsQueryTest:
+    """Test cases for ProjectHealthMetricsQuery class."""
+
+    def test_project_health_metrics_query_has_strawberry_definition(self):
+        """Check if ProjectHealthMetricsQuery has valid Strawberry definition."""
+        assert hasattr(ProjectHealthMetricsQuery, "__strawberry_definition__")
+
+        field_names = [
+            field.name for field in ProjectHealthMetricsQuery.__strawberry_definition__.fields
+        ]
+        assert "project_health_metrics" in field_names
+
+    def test_project_health_metrics_field_configuration(self):
+        """Test if 'project_health_metrics' field is configured properly."""
+        health_metrics_field = next(
+            field
+            for field in ProjectHealthMetricsQuery.__strawberry_definition__.fields
+            if field.name == "project_health_metrics"
+        )
+
+        assert health_metrics_field.type.of_type is ProjectHealthMetricsNode
+
+        arg_names = [arg.python_name for arg in health_metrics_field.arguments]
+        assert "key" in arg_names
+
+        key_arg = next(arg for arg in health_metrics_field.arguments if arg.python_name == "key")
+        assert key_arg.type_annotation.annotation is str
+
+
+class ProjectHealthMetricsResolutionTest:
+    """Test cases for resolving the project_health_metrics field."""
+
+    @patch("apps.owasp.models.project_health_metrics.ProjectHealthMetrics.objects")
+    def test_resolve_unhealthy_projects(self, mocked_objects):
+        """Test the resolution of unhealthy projects."""
+        # Mock the queryset
+        mock_queryset = Mock(spec=ProjectHealthMetrics.objects)
+        mocked_objects.select_related.return_value.order_by.return_value = mock_queryset
+
+        # Create a mock instance of ProjectHealthMetricsNode
+        mock_node = Mock(spec=ProjectHealthMetricsNode)
+        mock_queryset.all.return_value = [mock_node]
+
+        # Create an instance of the query class
+        query_instance = ProjectHealthMetricsQuery()
+
+        # Call the method
+        result = query_instance.unhealthy_projects()
+
+        # Assert that the mocked queryset was called correctly
+        mocked_objects.select_related.assert_called_with("project")
+        mocked_objects.select_related.return_value.order_by.assert_called_with(
+            "project__key", "-nest_created_at", "-score"
+        )
+
+        # Assert that the result is as expected
+        assert result == [mock_node]