Merge pull request #162 from mozilla/vs/testPDFdownload

Vs/test pdf download

Author: vsangereanMOZ

conftest.py

@@ -167,6 +167,20 @@ def sys_platform():
     return platform.system()
 
 
+@pytest.fixture()
+def downloads_folder(sys_platform):
+    """Return the downloads folder location for this OS"""
+    if sys_platform == "Windows":
+        user = os.environ.get("USERNAME")
+        return f"C:\\Users\\{user}\\Downloads"
+    elif sys_platform == "Darwin":  # MacOS
+        user = os.environ.get("USER")
+        return f"/Users/{user}/Downloads"
+    elif sys_platform == "Linux":
+        user = os.environ.get("USER")
+        return f"/home/{user}/Downloads"
+
+
 @pytest.fixture()
 def fx_executable(request, sys_platform):
     """Get the Fx executable path based on platform and edition request."""

modules/data/generic_pdf.components.json

@@ -31,5 +31,11 @@
         "selectorData": "zoomIn",
         "strategy": "id",
         "groups": []
+    },
+
+    "download-button": {
+        "selectorData": "download",
+        "strategy": "id",
+        "groups": []
     }
 }

tests/pdf_viewer/test_pdf_download.py

@@ -0,0 +1,62 @@
+import os
+import time
+
+import pytest
+
+from selenium.webdriver import Firefox
+from modules.page_object import GenericPdf
+
+
+@pytest.fixture()
+def add_prefs():
+    return []
+
+
+@pytest.mark.headed
+def test_pdf_download(driver: Firefox, fillable_pdf_url: str, downloads_folder: str, sys_platform):
+    """
+    C3932: PDF files can be successfully downloaded via pdf.js
+    """
+    from pynput.keyboard import Controller, Key
+
+    pdf = GenericPdf(driver, pdf_url=fillable_pdf_url).open()
+    keyboard = Controller()
+
+    # Click the download button
+    download_button = pdf.get_element("download-button")
+    download_button.click()
+
+    # Allow time for the download dialog m to appear and pressing enter to download
+    time.sleep(2)
+
+    if sys_platform == "Linux":
+        keyboard.press(Key.alt)
+        keyboard.press(Key.tab)
+        keyboard.release(Key.tab)
+        keyboard.release(Key.alt)
+        time.sleep(1)
+        keyboard.press(Key.alt)
+        keyboard.press(Key.tab)
+        keyboard.release(Key.tab)
+        keyboard.release(Key.alt)
+        time.sleep(1)
+        keyboard.press(Key.tab)
+        keyboard.release(Key.tab)
+        time.sleep(1)
+        keyboard.press(Key.tab)
+        keyboard.release(Key.tab)
+
+    keyboard.press(Key.enter)
+    keyboard.release(Key.enter)
+
+    # Allow time for the download to complete
+    time.sleep(2)
+
+    # Set the expected download path and the expected PDF name
+    file_name = "i-9.pdf"
+    saved_pdf_location = os.path.join(downloads_folder, file_name)
+
+    # Verify if the file exists
+    assert os.path.exists(saved_pdf_location), f"The file was not downloaded to {saved_pdf_location}."
+
+    print(f"Test passed: The file {file_name} has been downloaded and is present at {saved_pdf_location}.")