Add factories for opinions and image reading sessions

Author: MatMoore

manage_breast_screening/dicom/tests/factories.py

@@ -1,7 +1,10 @@
 import uuid
 
-from factory.declarations import Sequence, SubFactory
+from factory.declarations import RelatedFactory, Sequence, SubFactory, Trait
 from factory.django import DjangoModelFactory, FileField
+from factory.helpers import post_generation
+
+from manage_breast_screening.users.tests.factories import UserFactory
 
 from .. import models
 
@@ -17,6 +20,22 @@ class Meta:
     description = "Test Study"
 
 
+class StudyWithImagesFactory(StudyFactory):
+    class Meta:
+        skip_postgeneration_save = True
+
+    @post_generation
+    def num_images(obj, created, extracted, **kwargs):
+        if not created:
+            return
+
+        if not extracted:
+            extracted = 1
+
+        for _ in range(extracted):
+            ImageFactory(series__study=obj)
+
+
 class SeriesFactory(DjangoModelFactory):
     class Meta:
         model = models.Series
@@ -37,3 +56,62 @@ class Meta:
     laterality = "L"
     view_position = "CC"
     image_file = FileField(filename="image.jpg")
+
+
+class RecallForAssessmentDetailsFactory(DjangoModelFactory):
+    class Meta:
+        model = models.RecallForAssessmentDetails
+
+    right_breast_opinion = models.BreastOpinions.ABNORMAL
+    left_breast_opinion = models.BreastOpinions.NORMAL
+
+
+class RetakeRequestFactory(DjangoModelFactory):
+    class Meta:
+        model = models.RetakeRequest
+
+    laterality = "R"
+    view_position = "MLO"
+
+
+class ReadingFactory(DjangoModelFactory):
+    class Meta:
+        model = models.Reading
+        skip_postgeneration_save = True
+
+    study = SubFactory(StudyWithImagesFactory)
+    reader = SubFactory(UserFactory)
+    opinion = models.Opinions.NORMAL
+
+    class Params:
+        normal = Trait(opinion=models.Opinions.NORMAL)
+        technical_recall = Trait(
+            opinion=models.Opinions.TECHNICAL_RECALL,
+            retake_request_1=RelatedFactory(
+                RetakeRequestFactory, factory_related_name="reading"
+            ),
+        )
+        recall = Trait(
+            opinion=models.Opinions.RECALL,
+            recall_for_assessment_details=RelatedFactory(
+                RecallForAssessmentDetailsFactory, factory_related_name="reading"
+            ),
+        )
+
+
+class ReadingSessionItemFactory(DjangoModelFactory):
+    class Meta:
+        model = models.ReadingSessionItem
+
+    study = SubFactory(StudyFactory)
+    order = Sequence(lambda i: i)
+
+
+class ReadingSessionFactory(DjangoModelFactory):
+    class Meta:
+        model = models.ReadingSession
+        skip_postgeneration_save = True
+
+    reader = SubFactory(UserFactory)
+    session_size = 50
+    item_1 = RelatedFactory(ReadingSessionItemFactory, factory_related_name="session")

manage_breast_screening/dicom/tests/test_factories.py

@@ -0,0 +1,35 @@
+import pytest
+
+from ..models import BreastOpinions, Opinions
+from .factories import ReadingFactory, ReadingSessionFactory
+
+
+@pytest.mark.django_db
+def test_technical_recall_trait():
+    opinion = ReadingFactory(technical_recall=True)
+    assert opinion.opinion == Opinions.TECHNICAL_RECALL
+    assert opinion.retake_requests.count() == 1
+
+    retake_request = opinion.retake_requests.first()
+    assert retake_request.view_position == "MLO"
+    assert retake_request.laterality == "R"
+
+
+@pytest.mark.django_db
+def test_recall_trait():
+    opinion = ReadingFactory(recall=True)
+    assert opinion.opinion == Opinions.RECALL
+    assert (
+        opinion.recall_for_assessment_details.right_breast_opinion
+        == BreastOpinions.ABNORMAL
+    )
+    assert (
+        opinion.recall_for_assessment_details.left_breast_opinion
+        == BreastOpinions.NORMAL
+    )
+
+
+@pytest.mark.django_db
+def test_reading_session():
+    session = ReadingSessionFactory()
+    assert session.items.count() == 1