[py][tests]: check for .txt file in remote download test (#15758)

check for .txt file in test

Co-authored-by: Diego Molina <[email protected]>

Author: navin772

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()