add unit tests

atdb/taskdatabase/tests/test_models_joins.py

+from django.test import TestCase
+import json
+
+from taskdatabase.models import Workflow,Task
+
+
+class TestJoinedTasks(TestCase):
+
+    @classmethod
+    def setUpTestData(cls):
+
+        # Set up non-modified objects used by all test methods
+        workflow = Workflow()
+        workflow.save()
+
+        # create a list of Tasks
+        Task.objects.get_or_create(sas_id=1, status='stored')
+        Task.objects.get_or_create(sas_id=2, status='stored')
+        Task.objects.get_or_create(sas_id=3, status='defined')
+
+
+    def test_add_input_tasks_to_task(self):
+        output_task = Task.objects.get(sas_id=1)
+        input_task_1 = Task.objects.get(sas_id=2)
+        input_task_2 = Task.objects.get(sas_id=3)
+        output_task.joined_input_tasks.set([input_task_1,input_task_2])
+        output_task.save()
+
+        # count number of input asks
+        count_input_tasks = output_task.joined_input_tasks.count()
+        self.assertEqual(count_input_tasks,2)
+
+
+    def test_join_status_none(self):
+        output_task = Task.objects.get(sas_id=1)
+        input_task_1 = Task.objects.get(sas_id=2) # stored
+        input_task_2 = Task.objects.get(sas_id=3) # defined
+        output_task.joined_input_tasks.set([input_task_1,input_task_2])
+        output_task.save()
+
+        # one input task on stored, and on on defined, should give: None
+
+        self.assertEqual(output_task.joined_status,None)
+
+
+    def test_join_status_stored(self):
+        output_task = Task.objects.get(sas_id=3)
+        input_task_1 = Task.objects.get(sas_id=2)  # stored
+        input_task_2 = Task.objects.get(sas_id=1)  # stored
+        output_task.joined_input_tasks.set([input_task_1, input_task_2])
+        output_task.save()
+
+        # two input tasks 'stored', should give: 'stored'
+        self.assertEqual(output_task.joined_status, "stored")
\ No newline at end of file