Add the I wait for 2 seconds step

Author: kris-szlapa

features/cpts_ui/logout.feature

@@ -24,6 +24,7 @@ Feature: The user is able to logout of the application
     Scenario: User can log back in from the logout successful page
         Given I am on the logout successful page
         When I click the "log back in" button
+        Then I wait for 2 seconds
         Then I am on the login page
         And I can not see the RBAC banner
 

features/cpts_ui/search_for_a_prescription.feature

@@ -29,5 +29,7 @@ Feature: I can visit the Clinical Prescription Tracker Service Website
   @allure.tms:https://nhsd-jira.digital.nhs.uk/browse/AEA-4535
   @rbac_banner
   Scenario: User can see their RBAC banner when a role is selected
-    Given I am logged in with a single access role
+    Given I am logged in
+    When I have a selected role
+    Then I wait for 2 seconds
     Then I can see the RBAC banner

features/steps/cpts_ui/select_your_role_steps.py

@@ -1,4 +1,6 @@
 # pylint: disable=no-name-in-module
+import time
+
 from behave import when, then  # pyright: ignore [reportAttributeAccessIssue]
 from playwright.sync_api import expect
 
@@ -111,3 +113,10 @@ def i_see_logged_in_message(context):
         "eps_select_your_role_pre_role_selected"
     )
     expect(pre_selected_element).to_be_visible()
+
+
+@then("I wait for {seconds} seconds")
+def i_wait_for_seconds(_context, seconds):
+    wait_time = int(seconds)
+    time.sleep(wait_time)
+    print(f"Waited for {wait_time} seconds")