diff --git a/atdb/taskdatabase/tests/test_path_to_lta.py b/atdb/taskdatabase/tests/test_path_to_lta.py index fa4b0a9399b3b6f3e26e17318f3effe20aad14ae..5f650159ca48f4946b1d5c96c1a8ef004ff4f7a2 100644 --- a/atdb/taskdatabase/tests/test_path_to_lta.py +++ b/atdb/taskdatabase/tests/test_path_to_lta.py @@ -1,5 +1,5 @@ from django.test import TestCase -from taskdatabase.models import Task, Workflow +from taskdatabase.models import Task class TaskModelTestCase(TestCase): def setUp(self): @@ -10,22 +10,26 @@ class TaskModelTestCase(TestCase): self.task2 = Task.objects.create(sas_id=12345,archive={'path_to_lta': None}) # this task has a valid path_to_lta set - self.task3 = Task.objects.create(sas_id=12345,archive={'path_to_lta': '/sample/path'}) + self.task3 = Task.objects.create(sas_id=12345,archive={'path_to_lta': '/sample/path', 'sas_id_archived': 54321}) # this sasid has no path_to_lta set at all self.task4 = Task.objects.create(sas_id=66666,archive={}) self.task5 = Task.objects.create(sas_id=66666,archive={}) def test_path_to_lta_with_path(self): - # if only one of the tasks has a path_to_lta, then the other tasks should also return that path for task in Task.objects.filter(sas_id=12345): result = task.sasid_path_to_lta self.assertEqual(result, '/sample/path') def test_path_to_lta_without_path(self): - # if one of the tasks has 'path_to_lta' set, then return None for task in Task.objects.filter(sas_id=66666): result = task.sasid_path_to_lta - self.assertEqual(result, None) \ No newline at end of file + self.assertEqual(result, None) + + def test_sas_id_has_archived(self): + # if only one of the tasks has a sas_id_has_archived, then the other tasks should also return that path + for task in Task.objects.filter(sas_id=12345): + result = task.sas_id_has_archived + self.assertEqual(result, 54321) \ No newline at end of file