TestStatusBarPresenter

Author: swebberuk

manage_breast_screening/mammograms/tests/presenters/test_appointment_presenters.py

@@ -200,6 +200,68 @@ def test_status_attribution(
         assert presenter.status_attribution == expected_result
 
 
+class TestStatusBarPresenter:
+    @pytest.fixture
+    def mock_appointment(self):
+        mock = MagicMock(spec=Appointment)
+        mock.screening_episode.participant.nhs_number = "99900900829"
+        mock.screening_episode.participant.pk = uuid4()
+        return mock
+
+    @pytest.fixture
+    def mock_user(self):
+        return MagicMock(spec=User)
+
+    def test_show_status_bar_when_in_progress_and_user_is_owner(
+        self, mock_appointment, mock_user
+    ):
+        mock_appointment.current_status.state = AppointmentStatus.IN_PROGRESS
+        mock_user.nhs_uid = "user-123"
+        mock_appointment.current_status.created_by.nhs_uid = "user-123"
+        presenter = AppointmentPresenter(mock_appointment)
+        assert presenter.status_bar.show_status_bar_for(mock_user)
+
+    def test_show_status_bar_when_user_is_not_owner(self, mock_appointment, mock_user):
+        mock_appointment.current_status.state = AppointmentStatus.IN_PROGRESS
+        mock_user.nhs_uid = "user-123"
+        mock_appointment.current_status.created_by.nhs_uid = "user-456"
+        presenter = AppointmentPresenter(mock_appointment)
+        assert not presenter.status_bar.show_status_bar_for(mock_user)
+
+    @pytest.mark.parametrize(
+        "current_state",
+        [
+            AppointmentStatus.CONFIRMED,
+            AppointmentStatus.CHECKED_IN,
+            AppointmentStatus.DID_NOT_ATTEND,
+            AppointmentStatus.SCREENED,
+            AppointmentStatus.PARTIALLY_SCREENED,
+            AppointmentStatus.ATTENDED_NOT_SCREENED,
+        ],
+    )
+    def test_dont_show_status_bar_when_not_in_progress(
+        self, mock_appointment, mock_user
+    ):
+        mock_appointment.current_status.state = AppointmentStatus.CONFIRMED
+        mock_user.nhs_uid = "user-123"
+        mock_appointment.current_status.created_by.nhs_uid = "user-123"
+        presenter = AppointmentPresenter(mock_appointment)
+        assert not presenter.status_bar.show_status_bar_for(mock_user)
+
+    def test_tag_properties(self, mock_appointment):
+        presenter = AppointmentPresenter(mock_appointment)
+        assert presenter.status_bar.tag_properties == {
+            "classes": "nhsuk-tag--yellow nhsuk-u-margin-left-1",
+            "text": "Special appointment",
+        }
+
+    def test_attributes(self, mock_appointment):
+        presenter = AppointmentPresenter(mock_appointment)
+        assert presenter.status_bar.appointment == presenter
+        assert presenter.status_bar.clinic_slot == presenter.clinic_slot
+        assert presenter.status_bar.participant == presenter.participant
+
+
 class TestClinicSlotPresenter:
     @pytest.fixture
     def clinic_slot_mock(self):