handling review comments

atdb/taskdatabase/tests/test_path_to_lta.py

 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