Started adding section 4 from the smoke screen excel document

Author: adrianoaru-nhs

tests/Smokescreen/test_compartment_1.py

@@ -80,4 +80,81 @@ def test_example(page: Page) -> None:
             link_text = link.inner_text()  # Get the link text dynamically
             link.click()
     page.get_by_text("Open In-Queue Processing Prepared Type : Original Priority : High Deadline : 17").click()
-    
+
+
+from pypdf import PdfReader
+
+@pytest.mark.wip2
+def test_example(page: Page) -> None:
+    page.goto("/")
+    page.get_by_role("textbox", name="Username").click()
+    page.get_by_role("textbox", name="Username").fill("BCSS401")
+    page.get_by_role("textbox", name="Username").press("Tab")
+    page.get_by_role("textbox", name="Password").fill("changeme")
+    page.get_by_role("button", name="submit").click()
+
+     # Print the batch of Pre-Invitation Letters
+    page.get_by_role("link", name="Communications Production").click()
+    page.get_by_role("link", name="Active Batch List").click()
+    page.locator("#eventCodeFilter").click()
+    page.locator("#eventCodeFilter").fill("S1")
+    page.locator("#eventCodeFilter").press("Enter")
+    pre_invitation_cells = page.locator("//td[text()='Pre-invitation (FIT)']")
+
+    for i in range(pre_invitation_cells.count()):
+        row = pre_invitation_cells.nth(i).locator("..")  # Get the parent row
+
+        # Check if the row contains "Prepared" or "Open"
+        if row.locator("td", has_text="Prepared").count() > 0 or row.locator("td", has_text="Open").count() > 0:
+            break
+            # # Find the first link in that row and click it
+            # link = row.locator("a").first
+            # link_text = link.inner_text()  # Get the link text dynamically
+            # link.click()
+        else:
+            page.get_by_role("button", name="Prepare Batch").click()
+            page.wait_for_timeout(10000)
+
+        # Start waiting for the download
+        with page.expect_download() as download_info:
+            # Perform the action that initiates download
+            page.get_by_role("button", name="Retrieve").click()
+        download = download_info.value
+
+        # Wait for the download process to complete and save the downloaded file somewhere
+        download.save_as(f"/temp/{download.suggested_filename}")
+
+
+        reader = PdfReader(f"/temp/{download.suggested_filename}")
+        #For loop looping through all pages of the file to find the NHS Number
+        for pages in reader.pages:
+            text = pages.extract_text()
+            if "NHS No" in text:
+                #If NHS number is found split the text by every new line into a list
+                text = text.splitlines(True)
+                for split_text in text:
+                    if "NHS No" in split_text:
+                        #If a string is found containing "NHS No" all characters but digits are stored into nhs_no
+                        nhs_no = res = "".join([ele for ele in split_text if ele.isdigit()])
+                        break
+                    # print(nhs_no) 9849028742
+    page.on("dialog", lambda dialog: dialog.accept())
+    page.get_by_role("button", name="Confirm Printed").click()
+    expect(page.get_by_text("Batch Successfully Archived")).to_be_visible()
+    page.get_by_role("link", name="Back").click()
+    page.get_by_role("link", name="Archived Batch List").click()
+    page.get_by_role("cell", name="Date Archived : Activate to").locator("span").nth(1).click()
+    page.get_by_role("cell", name="Date Archived : Activate to").click()
+    # Change below to check if first row is  the same as the batch we just archived
+    # expect(page.get_by_role("cell", name="S1").first).to_be_visible()
+    # expect(page.get_by_role("cell", name="Pre-invitation (FIT)").first).to_be_visible()
+    # expect(page.get_by_role("link", name="8848")).to_be_visible()
+    page.get_by_role("link", name="Main Menu").click()
+    page.get_by_role("link", name="Screening Subject Search").click()
+    page.get_by_role("textbox", name="NHS Number").click()
+    page.get_by_role("textbox", name="NHS Number").fill(nhs_no)
+    page.get_by_role("button", name="Search").click()
+    expect(page.get_by_role("cell", name="Subject Screening Summary", exact=True)).to_be_visible()
+    page.get_by_role("link", name="Main Menu").click()
+    page.get_by_role("link", name="Communications Production").click()
+    page.get_by_role("link", name="Active Batch List").click()