Added loop for generate invitation page and few lines for next compartment

Author: Megha Prasannan

tests/Smokescreen/test_compartment_1.py

@@ -27,23 +27,57 @@ def test_example(page: Page) -> None:
     page.get_by_placeholder("Enter note").fill("test data")
     page.locator("#saveNote").get_by_role("button", name="Save").click()
 
-     # Generate Invitations
+    # Generate Invitations
     page.get_by_role("link", name="Main Menu").click()
     page.get_by_role("link", name="Call and Recall").click()
     page.get_by_role("link", name="Generate Invitations").click()
     page.get_by_role("button", name="Generate Invitations").click()
     # #TODO Add loop for below steps
+    # expect(page.locator("#displayRS")).to_contain_text("Queued")
+    # page.get_by_role("button", name="Refresh").click()
+    # expect(page.locator("#displayRS")).to_contain_text("Completed")
+
+    page.wait_for_selector("#displayRS", timeout=5000)
+
+    # Initially, ensure the table contains "Queued"
     expect(page.locator("#displayRS")).to_contain_text("Queued")
-    page.get_by_role("button", name="Refresh").click()
-    expect(page.locator("#displayRS")).to_contain_text("Completed")
 
+    # Set timeout parameters
+    timeout = 120000  # Total timeout of 120 seconds (in milliseconds)
+    wait_interval = 10000  # Wait 10 seconds between refreshes (in milliseconds)
+    elapsed = 0
 
-    #Print the batch of Pre-Invitation Letters
+    # Loop until the table no longer contains "Queued"
+    while elapsed < timeout:
+        table_text = page.locator("#displayRS").text_content()
+        if "Queued" in table_text:
+            # Click the Refresh button
+            page.get_by_role("button", name="Refresh").click()
+            page.wait_for_timeout(wait_interval)
+            elapsed += wait_interval
+        else:
+            break
+
+    # Final check: ensure that the table now contains "Completed"
+    expect(page.locator("#displayRS")).to_contain_text("Completed")
+
+    # Print the batch of Pre-Invitation Letters
+    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()
     page.locator("#eventCodeFilter").click()
     page.locator("#eventCodeFilter").fill("S1")
     page.locator("#eventCodeFilter").press("Enter")
-    page.get_by_role("link", name="8843").click()
+    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:
+            # 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()
     page.get_by_text("Open In-Queue Processing Prepared Type : Original Priority : High Deadline : 17").click()