TMSS-692: Add test case for Project report action

SAS/TMSS/backend/src/tmss/tmssapp/viewsets/specification.py

@@ -349,12 +349,11 @@ class ProjectViewSet(LOFARViewSet):
 
         return queryset
 
-    @swagger_auto_schema(responses={200: 'The Report',
+    @swagger_auto_schema(responses={200: 'The Report information',
                                     403: 'forbidden'},
-                         operation_description="Get Report for the project.")
+                         operation_description="Get Report information for the project.")
     @action(methods=['get'], detail=True, url_name="report", name="Get Report")
     def report(self, request, pk=None):
-        # TODO: Discuss about the endpoint choose. ATM is /project/<project-name>/report
         project = get_object_or_404(models.Project, pk=pk)
         result = create_project_report(request, project)
         return Response(result, status=status.HTTP_200_OK)

SAS/TMSS/backend/test/t_adapter.py

@@ -507,6 +507,42 @@ _isCobalt=T
         self.assertEqual(self.feedback_pipe_complete.strip(), subtask_pipe.raw_feedback.strip())
 
 
+class ProjectReportTest(unittest.TestCase):
+    def setUp(self):
+        # Create requirements
+        self.project = models.Project.objects.create(**Project_test_data(name='test_for_report'))
+        self.scheduling_set = models.SchedulingSet.objects.create(**SchedulingSet_test_data(project=self.project))
+        self.scheduling_unit_draft = models.SchedulingUnitDraft.objects.create(
+            **SchedulingUnitDraft_test_data(scheduling_set=self.scheduling_set))
+        self.task_draft = models.TaskDraft.objects.create(
+            **TaskDraft_test_data(scheduling_unit_draft=self.scheduling_unit_draft))
+        self.scheduling_unit_blueprint = models.SchedulingUnitBlueprint.objects.create(
+            **SchedulingUnitBlueprint_test_data(draft=self.scheduling_unit_draft))
+        self.task_blueprint = models.TaskBlueprint.objects.create(
+            **TaskBlueprint_test_data(task_draft=self.task_draft, scheduling_unit_blueprint=self.scheduling_unit_blueprint))
+
+    def test_create_report(self):
+        # TODO: Add ProjectQuota sample.
+
+        # Create Subtask of type 'ingest'
+        subtask_template = models.SubtaskTemplate.objects.create(**SubtaskTemplate_test_data(subtask_type_value='ingest'))
+        subtask = models.Subtask.objects.create(**Subtask_test_data(task_blueprint=self.task_blueprint, subtask_template=subtask_template))
+
+        # Set Subtask status to 'finished'
+        with tmss_test_env.create_tmss_client() as client:
+            client.set_subtask_status(subtask.pk, 'finished')
+
+        # Refreshing Subtask from cache.
+        subtask = models.Subtask.objects.get(pk=subtask.pk)
+        while subtask.state.value != 'finished':
+            subtask = models.Subtask.objects.get(pk=subtask.pk)
+
+        # Create SubtaskOutput and Dataproduct
+        subtask_output = models.SubtaskOutput.objects.create(**SubtaskOutput_test_data(subtask=subtask))
+        dataproduct = models.Dataproduct.objects.create(**Dataproduct_test_data(producer=subtask_output))
+
+        # TODO: Assert we get the expected object.
+
 
 if __name__ == "__main__":
     os.environ['TZ'] = 'UTC'