From 1f13ae03aa146c59b6df90345c3d5238b43de48f Mon Sep 17 00:00:00 2001
From: Vermaas <vermaas@astron.nl>
Date: Mon, 19 Feb 2024 14:27:06 +0100
Subject: [PATCH] add unit tests
---
.../tests/test_update_activity.py | 48 +++++++++++++++++++
1 file changed, 48 insertions(+)
diff --git a/atdb/taskdatabase/tests/test_update_activity.py b/atdb/taskdatabase/tests/test_update_activity.py
index e69de29b..c4992ebf 100644
--- a/atdb/taskdatabase/tests/test_update_activity.py
+++ b/atdb/taskdatabase/tests/test_update_activity.py
@@ -0,0 +1,48 @@
+from django.test import TestCase
+import json
+from taskdatabase.models import Task, Workflow, Activity
+
+class TestUpdateActivity(TestCase):
+
+ def setUp(self):
+ self.workflow_requantisation = Workflow(id=22, workflow_uri="psrfits_requantisation")
+ self.workflow_requantisation.save()
+
+ self.task1 = Task.objects.create(sas_id=12345,
+ status='stored',
+ workflow=self.workflow_requantisation,
+ calculated_qualities={"per_task": "good", "per_sasid": "good"})
+ self.task2 = Task.objects.create(sas_id=12345,
+ status='archived',
+ workflow=self.workflow_requantisation,
+ archive={
+ "path_to_lta": "https://lta.lofar.eu//Lofar?project=ALL&mode=show_dataproducts_pipe&product=PulsarPipeline&pipeline_object_id=1101AB934B386BD5E063164A17AC38B9",
+ "lta_object_id": "1101AB934B386BD5E063164A17AC38B9",
+ "sas_id_archived": "1219995"
+ })
+
+ def test_created_activity(self):
+
+ activity = self.task1.activity
+
+ # test if an activity with the correct sas_id was created
+ actual = activity.sas_id
+ self.assertEqual(actual, 12345)
+
+ def test_stored(self):
+
+ # test if the activity gets the calculated quality of the sas_id of the stored task
+ activity = self.task1.activity
+
+ actual = activity.calculated_quality
+ self.assertEqual(actual, "good")
+
+ def test_archived(self):
+ # test if the activity gets the ingested_fraction of an archived task
+ activity = self.task2.activity
+
+ actual = activity.ingestq_status
+ self.assertEqual(actual, {'archived': 1})
+
+ actual = activity.archive['sas_id_archived']
+ self.assertEqual(actual, "1219995")
--
GitLab