♻️ Add study attribute to IngestProcess

kids-first · Jun 1, 2021 · fd0e11b · fd0e11b
1 parent 66cb07b
commit fd0e11b
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/creator/ingest_runs/common/model.py b/creator/ingest_runs/common/model.py
@@ -109,6 +109,13 @@ def queue(self):
         """
         return django_rq.get_queue(name=self.__queue__)
 
+    @property
+    def study(self):
+        """
+        This should not be accessed.
+        """
+        raise NotImplementedError
+
     @transition(field=state, source=State.NOT_STARTED, target=State.RUNNING)
     def start(self):
         """

diff --git a/tests/ingest_runs/test_common.py b/tests/ingest_runs/test_common.py
@@ -6,6 +6,7 @@
     hash_versions,
     State,
     camel_to_snake,
+    IngestProcess,
 )
 from creator.ingest_runs.common.mutations import (
     cancel_duplicate_ingest_processes
@@ -25,6 +26,16 @@
 User = get_user_model()
 
 
+def test_study_not_implemented():
+    """
+    Test that accessing the study attribute of IngestProcess when it isn't
+    implemented yields a NotImplementedError.
+    """
+    ip = IngestProcess()
+    with pytest.raises(NotImplementedError):
+        ip.study
+
+
 def test_hash_versions(db, clients, prep_file):
     """
     Test hash_versions