increase wait time and verify everything is checked

Author: jf-x-dev

mavis/test/pages/sessions/sessions_patient_page.py

@@ -220,7 +220,7 @@ def confirm_pre_screening_checks(
             expect(locator).to_be_visible()
 
         # need to wait for checkbox to load properly
-        time.sleep(1)
+        time.sleep(3)
 
         expect(self.pre_screening_checkbox).to_be_editable()
         self.pre_screening_checkbox.check()
@@ -282,13 +282,43 @@ def set_up_vaccination(
         self.select_ready_for_vaccination(vaccination_record.consent_option)
         if vaccination_record.consent_option is ConsentOption.INJECTION:
             self.select_delivery_site(vaccination_record.delivery_site)
+
+        self.verify_all_required_radios_and_checkboxes_selected(vaccination_record)
+
         self.click_continue_button()
 
         if len(notes) > MAVIS_NOTE_LENGTH_LIMIT:
             expect(self.notes_length_error).to_be_visible()
             self.pre_screening_notes.fill("Prescreening notes")
             self.click_continue_button()
 
+    @step("Verify all required radios/checkboxes are selected")
+    def verify_all_required_radios_and_checkboxes_selected(
+        self, vaccination_record: VaccinationRecord
+    ) -> None:
+        child = vaccination_record.child
+
+        expect(self.pre_screening_checkbox).to_be_checked()
+
+        expect(
+            self.page.get_by_role(
+                "group",
+                name=f"Has {child.first_name} confirmed their identity?",
+            ).get_by_label("Yes")
+        ).to_be_checked()
+
+        if vaccination_record.consent_option is ConsentOption.INJECTION:
+            expect(self.ready_for_injection_radio).to_be_checked()
+        else:
+            expect(self.ready_for_nasal_spray_radio).to_be_checked()
+
+        if vaccination_record.consent_option is ConsentOption.INJECTION:
+            expect(
+                self.page.get_by_role(
+                    "radio", name=str(vaccination_record.delivery_site)
+                )
+            ).to_be_checked()
+
     @step("Click on Session activity and notes tab")
     def click_session_activity_and_notes(self) -> None:
         self._select_tab("Session activity and notes")

mavis/test/pages/sessions/sessions_vaccination_wizard_page.py

@@ -28,6 +28,9 @@ def __init__(self, page: Page) -> None:
             page.locator("div").filter(has_text="There is a problemEnter").nth(3)
         )
         self.continue_button = self.page.get_by_role("button", name="Continue")
+        self.select_batch_heading = self.page.get_by_role(
+            "heading", name="Which batch did you use?"
+        )
 
     @step("Click on Confirm")
     def click_confirm_button(self) -> None:
@@ -52,6 +55,10 @@ def choose_batch(self, batch_name: str) -> None:
 
         self.page.wait_for_load_state()
 
+        expect(
+            self.page.get_by_role("heading", name="Which batch did you use?")
+        ).to_be_visible()
+
         reload_until_element_is_visible(
             self.page,
             batch_radio,