From 94e711e7bcebc0f52a50f821ac5223db182a9e49 Mon Sep 17 00:00:00 2001 From: Navin Chandra Date: Tue, 20 May 2025 16:19:25 +0530 Subject: [PATCH] check for .txt file in test --- .../webdriver/remote/remote_downloads_tests.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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()