@@ -31,7 +31,7 @@ def test_example(page: Page) -> None:
3131 CreateAPlan (page ).click_save_button ()
3232 CreateAPlan (page ).fill_note_field ("test data" )
3333 CreateAPlan (page ).click_saveNote_button ()
34- page . locator ( '# page-title:has-text("Invitation Plans")' ) .wait_for ()
34+ InvitationsPlans ( page ). invitations_plans_title .wait_for ()
3535
3636 # Generate Invitations
3737 NavigationBar (page ).click_main_menu_link ()
@@ -54,7 +54,7 @@ def test_example(page: Page) -> None:
5454
5555 # Log out
5656 NavigationBar (page ).click_log_out_link ()
57- expect (page . get_by_role ( "heading" , name = "You have logged out" )). to_be_visible ()
57+ Logout (page ). verify_log_out_page ()
5858
5959def batch_processing (page : Page , batch_type : str , batch_description : str , latest_event_status : str ):
6060 NavigationBar (page ).click_main_menu_link ()
@@ -111,13 +111,13 @@ def batch_processing(page: Page, batch_type: str, batch_description: str, latest
111111 page .on ("dialog" , lambda dialog : dialog .accept ())
112112 ManageActiveBatch (page ).confirm_button .nth (0 ).click ()
113113
114- page . locator ( 'text="Batch Successfully Archived and Printed"' ) .wait_for ()
114+ ActiveBatchList ( page ). batch_successfully_archived_msg .wait_for ()
115115
116116 NavigationBar (page ).click_main_menu_link ()
117117 MainMenu (page ).go_to_communications_production_page ()
118118 CommunicationsProduction (page ).go_to_archived_batch_list_page ()
119119 ArchivedBatchList (page ).enter_id_filter (link_text )
120- expect (page . locator ( "td" ). filter ( has_text = link_text )). to_be_visible () # Checks to see if the batch is now archived
120+ ArchivedBatchList (page ). verify_table_data ( link_text )
121121
122122 subject_search_by_nhs_no (page , nhs_no , latest_event_status )
123123
0 commit comments