Add test 1756752

Author: soncuteanca

SELECTOR_INFO.md

@@ -1151,6 +1151,13 @@ Description: The dropdown menu for default zoom selection
 Location: about:preferences - Zoom settings
 Path to .json: modules/data/about_prefs.components.json
 ```
+```
+Selector Name: unknown-content-type-dialog
+Selector Data: unknownContentTypeWindo
+Description: The unknown content type dialog
+Location: about:preferences#general Applications subsection
+Path to .json: modules/data/about_prefs.components.json
+```
 #### about_profiles
 ```
 Selector Name: profile-container

modules/data/about_prefs.components.json

@@ -541,5 +541,13 @@
     "selectorData": "menuitem[data-l10n-id='preferences-default-zoom-value'][value='{.*}']",
     "strategy": "css",
     "groups": []
-  }
+  },
+
+    "unknown-content-type-dialog": {
+        "selectorData": "unknownContentTypeWindow",
+        "strategy": "id",
+        "groups": [
+            "doNotCache"
+        ]
+    }
 }

modules/page_object_prefs.py

@@ -657,6 +657,31 @@ def get_app_name_for_mime_type(self, mime_type: str) -> str:
         mime_type_data = json.loads(action_description.get_attribute("data-l10n-args"))
         return mime_type_data["app-name"]
 
+    def set_pdf_handling_to_always_ask(self) -> BasePage:
+        """
+        Set PDF content type handling to "Always ask" in Applications settings.
+        """
+        self.click_on("pdf-content-type")
+        self.click_on("pdf-actions-menu")
+        menu = self.get_element("pdf-actions-menu")
+        menu.send_keys(Keys.DOWN)
+        menu.send_keys(Keys.ENTER)
+        return self
+
+    @BasePage.context_chrome
+    def handle_unknown_content_dialog(self) -> BasePage:
+        """
+        Wait for the unknown content type dialog to appear and close it with Escape.
+        """
+        self.wait.until(lambda _: len(self.driver.window_handles) > 1)
+        self.driver.switch_to.window(self.driver.window_handles[-1])
+        self.wait.until(lambda _: self.get_element("unknown-content-type-dialog"))
+
+        # Close the dialog with Escape
+        dialog = self.get_element("unknown-content-type-dialog")
+        dialog.send_keys(Keys.ESCAPE)
+        return self
+
 
 class AboutAddons(BasePage):
     """

tests/downloads/test_set_always_ask_file_type.py

@@ -27,21 +27,21 @@ def delete_files_regex_string():
 @pytest.mark.ci
 def test_set_always_ask_file_type(driver: Firefox, delete_files):
     """
-    C1756752: Ensure that the Always ask option in Firefox Applications settings
+    C1756752 - Ensure that the Always ask option in Firefox Applications settings
     leads to a dialog asking "What should Firefox do with this file?" when the file type
     is downloaded.
     """
+
+    # Initialize page objects
     nav = Navigation(driver)
-    about_prefs = AboutPrefs(driver, category="general").open()
+    about_prefs = AboutPrefs(driver, category="general")
 
-    about_prefs.click_on("pdf-content-type")
-    about_prefs.click_on("pdf-actions-menu")
-    menu = about_prefs.get_element("pdf-actions-menu")
-    menu.send_keys(Keys.DOWN)
-    menu.send_keys(Keys.ENTER)
+    # Set PDF handling to "Always ask"
+    about_prefs.open()
+    about_prefs.set_pdf_handling_to_always_ask()
 
+    # Navigate to download URL and verify dialog appears
     nav.search(CONTENT_DISPOSITION_ATTACHMENT_URL)
-    sleep(2)
-    with driver.context(driver.CONTEXT_CHROME):
-        driver.switch_to.window(driver.window_handles[-1])
-        driver.find_element(By.ID, "unknownContentTypeWindow").send_keys(Keys.ESCAPE)
+
+    # Wait for and handle the unknown content type dialog
+    about_prefs.handle_unknown_content_dialog()