File tree Expand file tree Collapse file tree 2 files changed +3
-2
lines changed
pages/screening_practitioner_appointments Expand file tree Collapse file tree 2 files changed +3
-2
lines changed Original file line number Diff line number Diff line change 33from utils .table_util import TableUtils
44from datetime import datetime
55from utils .calendar_picker import CalendarPicker
6+ from typing import Optional
67
78
89class BookAppointmentPage (BasePage ):
@@ -49,7 +50,7 @@ def click_save_button(self) -> None:
4950 """Clicks the save button."""
5051 self .safe_accept_dialog (self .save_button )
5152
52- def click_save_button_and_return_message (self ) -> str | None :
53+ def click_save_button_and_return_message (self ) -> Optional [ str ] :
5354 """
5455 Clicks the save button and returns the dialog message if a dialog appears.
5556 Returns None if no dialog appears.
Original file line number Diff line number Diff line change @@ -168,7 +168,7 @@ def book_post_investigation_appointment(
168168 current_time = f"{ hour :02d} :{ minute :02d} "
169169 book_appointments_page .enter_appointment_start_time (current_time )
170170 dialog_message = book_appointments_page .click_save_button_and_return_message ()
171- if dialog_message is None or overlap_message not in dialog_message :
171+ if dialog_message is None or overlap_message not in ( dialog_message or "" ) :
172172 # Success or no overlap dialog
173173 break
174174 # Increase time by 15 minutes
You can’t perform that action at this time.
0 commit comments