set smoke tests to smoke and sanity

Author: Hani Yacoub

modules/testrail.py

@@ -160,6 +160,10 @@ def get_test_case(self, case_id):
         """Get a given Test Case"""
         return self.client.send_get(f"get_case/{case_id}")
 
+    def get_suites(self, project_id):
+        """Get all suites for project"""
+        return self.client.send_get(f"get_suites/{project_id}")
+
     def update_cases_in_suite(self, suite_id, case_ids, **kwargs):
         """Given a suite and a list of test cases, update all listed
         test cases according to keyword args"""

modules/testrail_set_smoke_tests_to_smoke_and_sanity.py

@@ -0,0 +1,119 @@
+import logging
+from dotenv import load_dotenv
+from modules.testrail_integration import testrail_init
+
+# Load environment variables from .env file
+load_dotenv()
+
+# Configuration Constants
+PROJECT_ID = 17  # The ID of the TestRail project to work with
+SUITE_NAMES = ["PDF Viewer"]  # List of suite names to process
+
+CUSTOM_SUB_TEST_SUITES = [2]  # Value to set for the 'custom_sub_test_suites' field
+TEST_SUB_SUITES = {
+    1: "functional",
+    2: "smoke",
+    3: "accessibility",
+    4: "l10n",
+    5: "security",
+    6: "fxa",
+    7: "other"
+}
+updated_count = 0
+DRY_RUN = True  # If True, only log actions without making changes
+
+
+def get_all_cases_from_suite(tr, project_id, suite_id):
+    """
+    Fetch all test cases from a given TestRail suite using pagination.
+
+    Args:
+        tr: TestRail API client instance.
+        project_id (int): The ID of the TestRail project.
+        suite_id (int): The ID of the suite to fetch cases from.
+
+    Returns:
+        list: A list of test case dictionaries.
+    """
+    all_cases = []  # List to store all fetched test cases
+    offset = 0  # Starting point for pagination
+    limit = 240  # Maximum number of cases to fetch in each API call
+
+    while True:
+        # Build the API endpoint with pagination parameters
+        endpoint = f"get_cases/{project_id}&suite_id={suite_id}&limit={limit}&offset={offset}"
+        response = tr.client.send_get(endpoint)  # Send API request to TestRail
+        cases = response.get("cases", [])  # Extract the list of test cases
+
+        if not cases:
+            # If no more cases are found, break the loop
+            break
+
+        # Add the fetched cases to the complete list
+        all_cases.extend(cases)
+
+        if len(cases) < limit:
+            # If the number of fetched cases is less than the limit, we are at the last page
+            break
+
+        # Increment offset to fetch the next batch of cases
+        offset += limit
+
+    # Log the total number of fetched cases
+    logging.info(f"Total cases fetched from suite {suite_id}: {len(all_cases)}")
+    return all_cases
+
+
+if __name__ == "__main__":
+    # Initialize logging to display info messages
+    logging.basicConfig(level=logging.INFO)
+
+    # Initialize the TestRail API client
+    tr = testrail_init()
+
+    # Fetch suite IDs matching the given suite names
+    suites = tr.get_suites(PROJECT_ID)
+    suite_ids = [suite["id"] for suite in suites if suite["name"] in SUITE_NAMES]
+    logging.info(f"Found suites: {suite_ids}")
+
+    # Fetch all test cases from the selected suites
+    all_cases = []  # List to store all fetched test cases
+    for suite_id in suite_ids:
+        # Fetch and add cases from each suite
+        cases = get_all_cases_from_suite(tr, PROJECT_ID, suite_id)
+        all_cases.extend(cases)
+
+    # Log the total number of test cases found
+    logging.info(f"Total test cases found for selected suites: {len(all_cases)}")
+
+    updated_count = 0  # Counter for the number of cases updated
+
+    # Iterate through all cases
+    for case in all_cases:
+        case_id = case["id"]
+        sub_test_suites_value = case.get("sub_test_suites", [])
+
+        logging.info(f"Case ID {case_id} sub_test_suites: {sub_test_suites_value}")
+
+        case_id = case["id"]
+        sub_test_suites_value = case.get("sub_test_suites", [])  # Fallback to empty list
+        custom_value = case.get("custom_sub_test_suites", "Not Set")
+
+        # Check if 2 is present in sub_test_suites
+        if 2 not in sub_test_suites_value:
+            logging.info(f"Skipping case {case_id} – sub_test_suites = {sub_test_suites_value}")
+            continue
+
+        logging.info(
+            f"Case ID {case_id} matches (contains 2 in sub_test_suites). Current custom_sub_test_suites: {custom_value}")
+
+        if not DRY_RUN:
+            # Perform the update
+            result = tr.update_case_field(case_id, "custom_sub_test_suites", CUSTOM_SUB_TEST_SUITES)
+            logging.info(f"Updated case {case_id} to '{CUSTOM_SUB_TEST_SUITES}', Result: {result}")
+        else:
+            logging.info(f"[DRY RUN] Would update case {case_id} from '{custom_value}' to '{CUSTOM_SUB_TEST_SUITES}'.")
+
+        updated_count += 1
+
+    logging.info(f"Total cases {'updated' if not DRY_RUN else 'to be updated'}: {updated_count}")