107 lines
5.1 KiB
Python
107 lines
5.1 KiB
Python
"""Use SeleniumBase to download files and verify."""
|
|
import math
|
|
from seleniumbase import BaseCase
|
|
BaseCase.main(__name__, __file__)
|
|
|
|
|
|
class DownloadTests(BaseCase):
|
|
def test_download_chromedriver_notes(self):
|
|
if self._multithreaded:
|
|
self.open_if_not_url("about:blank")
|
|
self.skip("Skipping test in multi-threaded mode.")
|
|
self.open("https://chromedriver.chromium.org/downloads")
|
|
notes_file = "notes.txt"
|
|
notes_link = (
|
|
"https://chromedriver.storage.googleapis.com"
|
|
"/101.0.4951.41/%s" % notes_file
|
|
)
|
|
self.download_file(notes_link)
|
|
self.assert_downloaded_file(notes_file)
|
|
notes_path = self.get_path_of_downloaded_file(notes_file)
|
|
with open(notes_path, "r") as f:
|
|
notes_data = f.read()
|
|
self.assert_true(len(notes_data) > 100) # Verify file not empty
|
|
text = "Switching to nested frame fails with chrome/chromedriver 100"
|
|
self.assert_true(text in notes_data) # Verify file has expected data
|
|
|
|
def test_download_files_from_pypi(self):
|
|
self.open("https://pypi.org/project/sbvirtualdisplay/#files")
|
|
self.assert_element("span#pip-command")
|
|
self.assert_text("Download files", "div#files h2.page-title")
|
|
self.assert_text("Download files", "a#files-tab")
|
|
pkg_header = self.get_text("h1.package-header__name").strip()
|
|
pkg_name = pkg_header.replace(" ", "-")
|
|
whl_file = pkg_name + "-py2.py3-none-any.whl"
|
|
tar_gz_file = pkg_name + ".tar.gz"
|
|
|
|
# Click the links to download the files into: "./downloaded_files/"
|
|
# (If using Safari, IE, or Chromium Guest Mode: download directly.)
|
|
# (The default Downloads Folder can't be changed when using those.)
|
|
# (Use self.get_browser_downloads_folder() to get the folder used.)
|
|
whl_selector = 'div#files a[href$="%s"]' % whl_file
|
|
tar_selector = 'div#files a[href$="%s"]' % tar_gz_file
|
|
if (
|
|
self.browser == "safari"
|
|
or self.browser == "ie"
|
|
or self.browser == "edge"
|
|
or (self.is_chromium() and self.guest_mode)
|
|
or (self.is_chromium() and (self.headless or self.headless2))
|
|
):
|
|
whl_href = self.get_attribute(whl_selector, "href")
|
|
tar_href = self.get_attribute(tar_selector, "href")
|
|
self.download_file(whl_href)
|
|
self.download_file(tar_href)
|
|
else:
|
|
self.js_click(whl_selector) # Download the "whl" file
|
|
self.sleep(0.1)
|
|
self.js_click(tar_selector) # Download the "tar" file
|
|
|
|
# Verify that the downloaded files appear in the [Downloads Folder]
|
|
# (This only guarantees that the exact file name is in the folder.)
|
|
# (This does not guarantee that the downloaded files are complete.)
|
|
# (Later, we'll check that the files were downloaded successfully.)
|
|
self.assert_downloaded_file(whl_file)
|
|
self.assert_downloaded_file(tar_gz_file)
|
|
|
|
self.sleep(1) # Add more time to make sure downloads have completed
|
|
|
|
# Get the actual size of the downloaded files (in bytes)
|
|
whl_path = self.get_path_of_downloaded_file(whl_file)
|
|
with open(whl_path, "rb") as f:
|
|
whl_file_bytes = len(f.read())
|
|
print("\n%s | Download = %s bytes." % (whl_file, whl_file_bytes))
|
|
tar_gz_path = self.get_path_of_downloaded_file(tar_gz_file)
|
|
with open(tar_gz_path, "rb") as f:
|
|
tar_gz_file_bytes = len(f.read())
|
|
print("%s | Download = %s bytes." % (tar_gz_file, tar_gz_file_bytes))
|
|
|
|
# Check to make sure the downloaded files are not empty or too small
|
|
self.assert_true(whl_file_bytes > 5000)
|
|
self.assert_true(tar_gz_file_bytes > 5000)
|
|
|
|
# Get file sizes in kB to compare actual values with displayed values
|
|
whl_file_kb = whl_file_bytes / 1000.0
|
|
whl_line_fi = self.get_text('a[href$=".whl"]').strip()
|
|
whl_line = self.get_text('div.file:contains("%s")' % whl_line_fi)
|
|
whl_display_kb = float(whl_line.split("(")[1].split(" ")[0])
|
|
tar_gz_file_kb = tar_gz_file_bytes / 1000.0
|
|
tar_gz_line_fi = self.get_text('a[href$=".tar.gz"]').strip()
|
|
tar_gz_line = self.get_text('div.file:contains("%s")' % tar_gz_line_fi)
|
|
tar_gz_display_kb = float(tar_gz_line.split("(")[1].split(" ")[0])
|
|
|
|
# Verify downloaded files are the correct size (account for rounding)
|
|
self.assert_true(
|
|
abs(math.floor(whl_file_kb) - math.floor(whl_display_kb)) < 2
|
|
)
|
|
self.assert_true(
|
|
abs(math.floor(tar_gz_file_kb) - math.floor(tar_gz_display_kb)) < 2
|
|
)
|
|
|
|
# Delete the downloaded files from the [Downloads Folder]
|
|
self.delete_downloaded_file_if_present(whl_file)
|
|
self.delete_downloaded_file_if_present(tar_gz_file)
|
|
|
|
# Verify that the downloaded files have been successfully deleted
|
|
self.assert_false(self.is_downloaded_file_present(whl_file))
|
|
self.assert_false(self.is_downloaded_file_present(tar_gz_file))
|