Added test_reprint_and_archive_letter_batch and added/updated relevant utils/functions

Author: Andyg79

pages/communication_production/batch_list_page.py

@@ -232,3 +232,45 @@ def select_first_archived_batch(self) -> None:
         ).first
         assert first_batch_link.count() > 0, "No archived batch links found"
         first_batch_link.click()
+
+    def get_archived_batch_row(
+        self, batch_type: str, event_code: str, description: str
+    ) -> Locator | None:
+        """
+        Returns the first archived batch row matching the specified Type, Event Code, and Description.
+        Assumes columns appear in the following order:
+        1. Batch ID
+        2. Type
+        3. Event Code
+        4. Description
+        5+ ...
+
+        Args:
+            batch_type (str): The target value in the 'Type' column (e.g., 'Original').
+            event_code (str): The target value in the 'Event Code' column (e.g., 'S1').
+            description (str): The target text in the 'Description' column (e.g., 'Pre-invitation (FIT)').
+
+        Returns:
+            Locator: The first <tr> element matching all criteria, or None if not found.
+        """
+        rows = self.page.locator(f"{self.table_selector} tbody tr")
+        row_count = rows.count()
+        if row_count == 0:
+            return None
+
+        for i in range(row_count):
+            row = rows.nth(i)
+
+            try:
+                type_text = row.locator("td").nth(1).inner_text().strip()
+                event_code_text = row.locator("td").nth(4).inner_text().strip()
+                description_text = row.locator("td").nth(5).inner_text().strip()
+
+                if (
+                    batch_type.lower() in type_text.lower()
+                    and event_code.lower() in event_code_text.lower()
+                    and description.lower() in description_text.lower()
+                ):
+                    return row
+            except IndexError:
+                return None

pages/communication_production/manage_active_batch_page.py

@@ -32,7 +32,7 @@ def click_confirm_button(self) -> None:
         """Click the Confirm Printed button"""
         self.click(self.confirm_button)
 
-    def assert_batch_details_visible(self) -> None:
+    def assert_active_batch_details_visible(self) -> None:
         """Asserts that the Manage Active Batch screen has loaded by checking the page title."""
         page_title = self.page.locator("#page-title")
         expect(page_title).to_have_text("Manage Active Batch")

pages/communication_production/manage_archived_batch_page.py

@@ -9,7 +9,20 @@ class ManageArchivedBatchPage(BasePage):
     def __init__(self, page: Page):
         super().__init__(page)
 
-    def assert_batch_details_visible(self) -> None:
+    def assert_archived_batch_details_visible(self) -> None:
         """Verifies the Manage Archived Batch page has loaded."""
         header = self.page.locator("#page-title")
         expect(header).to_have_text("Manage Archived Batch")
+
+    def click_reprint_button(self) -> None:
+        """Clicks the 'Reprint' button on the Archived Batch details page."""
+        reprint_button = self.page.locator("input.ReprintButton[value='Reprint Batch']")
+        expect(reprint_button).to_be_visible()
+        reprint_button.click()
+
+    def confirm_archived_message_visible(self) -> None:
+        """Verifies that the batch was successfully archived and confirmation message is shown."""
+        confirmation_msg = self.page.locator(
+            'text="Batch Successfully Archived and Printed"'
+        )
+        expect(confirmation_msg).to_be_visible()

tests/regression/communications_production/test_basic_manage_active_batch_functionality.py

@@ -9,7 +9,6 @@
 from pages.communication_production.manage_active_batch_page import (
     ManageActiveBatchPage,
 )
-from pages.communication_production.batch_list_page import ActiveBatchListPage
 from utils.batch_processing import prepare_and_print_batch
 
 
@@ -56,7 +55,7 @@ def test_prepare_retrieve_and_confirm_active_letter_batch(select_user) -> None:
 
     # Step 5: Assert that Manage Active Batch page has loaded
     manage_page = ManageActiveBatchPage(page)
-    manage_page.assert_batch_details_visible()
+    manage_page.assert_active_batch_details_visible()
 
     # Step 6: Prepare, retrieve and confirm the batch using utility method
     prepare_and_print_batch(page, link_text=batch_id)

tests/regression/communications_production/test_basic_manage_archived_batch_functionality.py

@@ -1,15 +1,90 @@
-# @LettersTests
-# Feature: Basic Manage Archived Batch functionality
+import pytest
+from playwright.sync_api import Page
+from pages.base_page import BasePage
+from pages.communication_production.communications_production_page import (
+    CommunicationsProductionPage,
+)
+from pages.communication_production.batch_list_page import ArchivedBatchListPage
+from utils.user_tools import UserTools
+from pages.communication_production.manage_active_batch_page import (
+    ManageActiveBatchPage,
+)
+from pages.communication_production.manage_archived_batch_page import (
+    ManageArchivedBatchPage,
+)
+from utils.batch_processing import prepare_and_print_batch
 
 
-# Scenario: I can take an archived batch, reprint it, then archive that new batch
-# Given I log in to BCSS "England" as user role "HubManagerStateRegistered"
-# When I view the archived batch list
-# And I view the "Original" type archived letter batch for "S1" "Pre"
-# And I reprint the archived letter batch
-# And I prepare the letter batch
-# And I retrieve and confirm the letters
-# And my batch is now archived
+@pytest.fixture
+def select_user(page: Page):
+    def _login_as(user_role: str):
+        # Log in with the specified user
+        UserTools.user_login(page, user_role)
+        # Navigate to Active Batch List
+        BasePage(page).go_to_communications_production_page()
+        CommunicationsProductionPage(page).go_to_archived_batch_list_page()
+        return page
+
+    return _login_as
+
+
+@pytest.mark.wip
+@pytest.mark.letters_tests
+@pytest.mark.regression
+def test_reprint_and_archive_letter_batch(select_user) -> None:
+    """
+    Scenario: I can take an archived batch, reprint it, then archive that new batch
+    Given I log in to BCSS "England" as user role "HubManagerStateRegistered"
+    When I view the archived batch list
+    And I view the "Original" type archived letter batch for "S1" "Pre"
+    And I reprint the archived letter batch
+    And I prepare the letter batch
+    And I retrieve and confirm the letters
+    And my batch is now archived
+    """
+    # Step 1: Log in as user and navigate to Archived Batch List
+    page = select_user("Hub Manager State Registered at BCS01")
+    batch_list_page = ArchivedBatchListPage(page)
+
+    # Step 2: Ensure the archived batch list table is visible
+    batch_list_page.assert_batch_table_visible()
+    # Wait for at least one row to appear
+    page.wait_for_function(
+        "document.querySelectorAll('table#batchList tbody tr').length > 1", timeout=8000
+    )
+    rows = page.locator("table#batchList tbody tr")
+    row_count = rows.count()
+
+    for i in range(row_count):
+        row = rows.nth(i)
+
+    # Step 3: Find and open archived batch with Type "Original", Event Code "S1", and Description "Pre"
+    # You might want to use filters like:
+    row = batch_list_page.get_archived_batch_row(
+        "Original", "S1", "Pre-invitation (FIT)"
+    )
+    if not row:
+        pytest.skip("No archived 'Original' S1 Pre batches found to reprint.")
+
+    batch_id = row.locator("a").first.inner_text()
+    row.locator("a").first.click()
+
+    # Step 4: Perform reprint from Archived Batch detail screen
+    manage_archived_page = ManageArchivedBatchPage(page)  # Assuming reuse
+    manage_archived_page.assert_archived_batch_details_visible()
+    manage_archived_page.click_reprint_button()
+    BasePage(page).safe_accept_dialog(page.get_by_role("button", name="Reprint Batch"))
+
+    # Step 5: Wait for navigation to new Active Batch screen
+    # You’ll likely land back in Active Batch context
+    manage_active_page = ManageActiveBatchPage(page)
+    manage_active_page.assert_active_batch_details_visible()
+
+    # Step 6: Prepare, retrieve and confirm new batch
+    prepare_and_print_batch(page, link_text=batch_id)
+
+    # Step 7: Assert batch archived successfully
+    manage_archived_page.confirm_archived_message_visible()
 
 
 # Scenario: Check that S1 has supplementary batches