Update coverage/tree route to parse flags and components filters

api/internal/coverage/views.py

@@ -2,8 +2,8 @@
 from rest_framework.decorators import action
 from rest_framework.exceptions import NotFound
 from rest_framework.response import Response
-from rest_framework.views import APIView
 
+import services.components as components_service
 from api.shared.mixins import RepoPropertyMixin
 from api.shared.permissions import RepositoryArtifactPermissions
 from api.shared.report.serializers import TreeSerializer
@@ -20,31 +20,49 @@ def get_object(self):
             branch = self.repo.branches.filter(name=branch_name).first()
             if branch is None:
                 raise NotFound(
-                    f"The branch '{branch_name}' in not in our records. Please provide a valid branch name.",
+                    f"The branch '{branch_name}' is not in our records. Please provide a valid branch name.",
                     404,
                 )
             commit_sha = branch.head
 
         commit = self.repo.commits.filter(commitid=commit_sha).first()
         if commit is None:
             raise NotFound(
-                f"The commit {commit_sha} is not in our records. Please specify valid commit.",
+                f"The commit {commit_sha} is not in our records. Please specify a valid commit.",
                 404,
             )
 
         report = commit.full_report
         if report is None:
             raise NotFound(f"Coverage report for {commit_sha} not found")
 
-        return report
+        components = self.request.query_params.getlist("components")
+        component_paths = []
+        if components:
+            all_components = components_service.commit_components(
+                commit, self.request.user
+            )
+            filtered_components = components_service.filter_components_by_name(
+                all_components, components
+            )
+
+            if not filtered_components:
+                raise NotFound(
+                    f"Coverage report for components {filtered_components} not found"
+                )
+
+            for component in filtered_components:
+                component_paths.extend(component.paths)
+        flags = self.request.query_params.getlist("flags")
+
+        paths = ReportPaths(
+            report=report, filter_flags=flags, filter_paths=component_paths
+        )
+
+        return paths
 
-    @action(
-        detail=False,
-        methods=["get"],
-        url_path="tree",
-    )
+    @action(detail=False, methods=["get"], url_path="tree")
     def tree(self, request, *args, **kwargs):
-        report = self.get_object()
-        paths = ReportPaths(report)
+        paths = self.get_object()
         serializer = TreeSerializer(paths.single_directory(), many=True)
         return Response(serializer.data)

api/internal/tests/views/test_coverage_viewset.py

@@ -9,6 +9,7 @@
 
 from codecov_auth.tests.factories import OwnerFactory
 from core.tests.factories import BranchFactory, CommitFactory, RepositoryFactory
+from services.components import Component
 from utils.test_utils import Client
 
 
@@ -87,10 +88,19 @@ def setUp(self):
         self.client.force_login_owner(self.current_owner)
 
     @patch("shared.reports.api_report_service.build_report_from_commit")
-    def test_tree(self, build_report_from_commit):
+    @patch("services.components.commit_components")
+    def test_tree(self, commit_components_mock, build_report_from_commit):
+        commit_components_mock.return_value = [
+            Component.from_dict(
+                {
+                    "component_id": "global",
+                    "name": "Global",
+                    "paths": [".*/*.py"],
+                }
+            ),
+        ]
         build_report_from_commit.return_value = sample_report()
-
-        res = self._tree()
+        res = self._tree(components="Global")
         assert res.status_code == 200
         assert res.json() == [
             {
@@ -286,3 +296,63 @@ def test_tree_missing_report(self, build_report_from_commit):
 
         res = self._tree()
         assert res.status_code == 404
+
+    @patch("shared.reports.api_report_service.build_report_from_commit")
+    @patch("services.components.commit_components")
+    def test_tree_no_data_for_components(
+        self, commit_components_mock, build_report_from_commit
+    ):
+        commit_components_mock.return_value = [
+            Component.from_dict(
+                {
+                    "component_id": "c1",
+                    "name": "ComponentOne",
+                    "paths": ["dne.py"],
+                }
+            ),
+        ]
+        build_report_from_commit.return_value = sample_report()
+        res = self._tree(components="ComponentOne")
+        assert res.json() == []
+
+    @patch("shared.reports.api_report_service.build_report_from_commit")
+    @patch("services.components.commit_components")
+    def test_tree_not_found_for_components(
+        self, commit_components_mock, build_report_from_commit
+    ):
+        commit_components_mock.return_value = [
+            Component.from_dict(
+                {
+                    "component_id": "c1",
+                    "name": "ComponentOne",
+                    "paths": ["dne.py"],
+                }
+            ),
+        ]
+        build_report_from_commit.return_value = sample_report()
+        res = self._tree(components="Does_not_exist")
+        assert res.status_code == 404
+
+    @patch("shared.reports.api_report_service.build_report_from_commit")
+    @patch("services.components.commit_components")
+    def test_tree_not_found_for_components(
+        self, commit_components_mock, build_report_from_commit
+    ):
+        commit_components_mock.return_value = [
+            Component.from_dict(
+                {
+                    "component_id": "c1",
+                    "name": "ComponentOne",
+                    "paths": ["dne.py"],
+                }
+            ),
+        ]
+        build_report_from_commit.return_value = sample_report()
+        res = self._tree(components="Does_not_exist")
+        assert res.status_code == 404
+
+    @patch("shared.reports.api_report_service.build_report_from_commit")
+    def test_tree_no_data_for_flags(self, build_report_from_commit):
+        build_report_from_commit.return_value = sample_report()
+        res = self._tree(flags="Does_not_exist")
+        assert res.json() == []