Test for appointment status bar

Author: swebberuk

manage_breast_screening/participants/tests/factories.py

@@ -160,6 +160,14 @@ def current_status(obj, create, extracted, **kwargs):
             AppointmentStatusFactory.create(state=extracted, appointment=obj)
         )
 
+    # Allow passing an explicit status and created_by user
+    # e.g. `current_status_params={'state': AppointmentStatus.IN_PROGRESS, 'created_by': self.current_user}`
+    @post_generation
+    def current_status_params(obj, create, extracted, **kwargs):
+        if not create or not extracted:
+            return
+        obj.statuses.add(AppointmentStatusFactory.create(appointment=obj, **extracted))
+
 
 class AppointmentNoteFactory(DjangoModelFactory):
     class Meta: