@@ -139,6 +139,11 @@ def __init__(self, page: Page):
139139 "button" , name = "Redirect to Establish Attendance at Appointment"
140140 )
141141 )
142+ self .high_risk_findings_result_from_symptomatic_procedure_button = (
143+ self .page .get_by_role (
144+ "button" , name = "High-risk findings Result from Symptomatic Procedure"
145+ )
146+ )
142147 self .date_of_symptomatic_procedure_input = self .page .get_by_role (
143148 "textbox" , name = "Date of Symptomatic Procedure"
144149 )
@@ -171,6 +176,9 @@ def __init__(self, page: Page):
171176 self .lnpcp_result_from_symptomatic_procedure_button = self .page .get_by_role (
172177 "button" , name = "LNPCP Result from Symptomatic Procedure"
173178 )
179+ self .other_post_investigation_contact_required_button = self .page .get_by_role (
180+ "button" , name = "Other Post-investigation Contact Required"
181+ )
174182
175183 def click_suitable_for_endoscopic_test_button (self ) -> None :
176184 """Click the 'Suitable for Endoscopic Test' button."""
@@ -518,3 +526,15 @@ def enter_date_of_symptomatic_procedure(self, date: datetime) -> None:
518526 CalendarPicker (self .page ).calendar_picker_ddmmyyyy (
519527 date , self .date_of_symptomatic_procedure_input
520528 )
529+
530+ def click_high_risk_findings_result_from_symptomatic_procedure_button (
531+ self ,
532+ ) -> None :
533+ """Click the 'High-risk findings Result from Symptomatic Procedure' button"""
534+ self .safe_accept_dialog (
535+ self .high_risk_findings_result_from_symptomatic_procedure_button
536+ )
537+
538+ def click_other_post_investigation_contact_required_button (self ) -> None :
539+ """Click the 'Other Post Investigation Contact Required' button."""
540+ self .safe_accept_dialog (self .other_post_investigation_contact_required_button )
0 commit comments