Merge pull request #697 from seleniumbase/improve-file-downloads

Improve file downloads - disable automation prompts

Author: mdmintz

examples/test_download_files.py

@@ -0,0 +1,15 @@
+from seleniumbase import BaseCase
+
+
+class DownloadTests(BaseCase):
+
+    def test_download_files(self):
+        self.open("https://pypi.org/project/seleniumbase/#files")
+        pkg_header = self.get_text("h1.package-header__name")
+        pkg_name = pkg_header.replace(" ", "-")
+        whl_file = pkg_name + "-py2.py3-none-any.whl"
+        self.click('div#files a[href$="%s"]' % whl_file)
+        self.assert_downloaded_file(whl_file)
+        tar_gz_file = pkg_name + ".tar.gz"
+        self.click('div#files a[href$="%s"]' % tar_gz_file)
+        self.assert_downloaded_file(tar_gz_file)

requirements.txt

@@ -54,12 +54,12 @@ ipython==7.16.1;python_version>="3.6" and python_version<"3.7"
 ipython==7.18.1;python_version>="3.7"
 colorama==0.4.3
 pymysql==0.10.1
-coverage==5.2.1
+coverage==5.3
 brython==3.8.10
 pyotp==2.4.0
 boto==2.49.0
 cffi==1.14.2
-rich==6.1.2;python_version>="3.6" and python_version<"4.0"
+rich==6.2.0;python_version>="3.6" and python_version<"4.0"
 flake8==3.7.9;python_version<"3.5"
 flake8==3.8.3;python_version>="3.5"
 pyflakes==2.1.1;python_version<"3.5"

seleniumbase/core/browser_launcher.py

@@ -145,7 +145,8 @@ def _set_chrome_options(
         "safebrowsing.enabled": False,
         "safebrowsing.disable_download_protection": True,
         "profile": {
-            "password_manager_enabled": False
+            "password_manager_enabled": False,
+            "default_content_setting_values.automatic_downloads": 1
         }
     }
     if locale_code:
@@ -349,11 +350,14 @@ def _create_firefox_profile(
         "browser.download.animateNotifications", False)
     profile.set_preference("browser.download.dir", downloads_path)
     profile.set_preference("browser.download.folderList", 2)
+    profile.set_preference("browser.helperApps.alwaysAsk.force", False)
+    profile.set_preference(
+        "browser.download.manager.showWhenStarting", False)
     profile.set_preference(
         "browser.helperApps.neverAsk.saveToDisk",
         ("application/pdf, application/zip, application/octet-stream, "
          "text/csv, text/xml, application/xml, text/plain, "
-         "text/octet-stream, "
+         "text/octet-stream, application/x-gzip, application/x-tar "
          "application/"
          "vnd.openxmlformats-officedocument.spreadsheetml.sheet"))
     return profile
@@ -761,7 +765,8 @@ def get_local_driver(
                 "safebrowsing.enabled": False,
                 "safebrowsing.disable_download_protection": True,
                 "profile": {
-                    "password_manager_enabled": False
+                    "password_manager_enabled": False,
+                    "default_content_setting_values.automatic_downloads": 1
                 }
             }
             if locale_code:

seleniumbase/fixtures/base_case.py

@@ -2853,8 +2853,8 @@ def is_downloaded_file_present(self, file):
     def assert_downloaded_file(self, file, timeout=None):
         """ Asserts that the file exists in the Downloads Folder. """
         if not timeout:
-            timeout = settings.SMALL_TIMEOUT
-        if self.timeout_multiplier and timeout == settings.SMALL_TIMEOUT:
+            timeout = settings.LARGE_TIMEOUT
+        if self.timeout_multiplier and timeout == settings.LARGE_TIMEOUT:
             timeout = self.__get_new_timeout(timeout)
         start_ms = time.time() * 1000.0
         stop_ms = start_ms + (timeout * 1000.0)

setup.py

@@ -54,7 +54,7 @@
 
 setup(
     name='seleniumbase',
-    version='1.49.12',
+    version='1.49.13',
     description='Web Automation and Test Framework - https://seleniumbase.io',
     long_description=long_description,
     long_description_content_type='text/markdown',
@@ -146,12 +146,12 @@
         'ipython==7.18.1;python_version>="3.7"',
         'colorama==0.4.3',
         'pymysql==0.10.1',
-        'coverage==5.2.1',
+        'coverage==5.3',
         'brython==3.8.10',
         'pyotp==2.4.0',
         'boto==2.49.0',
         'cffi==1.14.2',
-        'rich==6.1.2;python_version>="3.6" and python_version<"4.0"',
+        'rich==6.2.0;python_version>="3.6" and python_version<"4.0"',
         'flake8==3.7.9;python_version<"3.5"',
         'flake8==3.8.3;python_version>="3.5"',
         'pyflakes==2.1.1;python_version<"3.5"',