diff --git a/py/test/selenium/webdriver/remote/remote_downloads_tests.py b/py/test/selenium/webdriver/remote/remote_downloads_tests.py index 5a8ed7f1fcbd4..152e0f4383a2f 100644 --- a/py/test/selenium/webdriver/remote/remote_downloads_tests.py +++ b/py/test/selenium/webdriver/remote/remote_downloads_tests.py @@ -35,11 +35,15 @@ def test_get_downloadable_files(driver, pages): def test_download_file(driver, pages): _browser_downloads(driver, pages) - file_name = driver.get_downloadable_files()[0] + # Get a list of downloadable files and find the txt file + downloadable_files = driver.get_downloadable_files() + text_file_name = next((file for file in downloadable_files if file.endswith(".txt")), None) + assert text_file_name is not None, "Could not find a .txt file in downloadable files" + with tempfile.TemporaryDirectory() as target_directory: - driver.download_file(file_name, target_directory) + driver.download_file(text_file_name, target_directory) - target_file = os.path.join(target_directory, file_name) + target_file = os.path.join(target_directory, text_file_name) with open(target_file, "r") as file: assert "Hello, World!" in file.read()