Fix get_downloads_list which is causing save_downloaded_file to fail (#…

…1098) (#1101) (cherry picked from commit b72a45d) Co-authored-by: Jameer Pathan <[email protected]>
SatelliteQE · Jan 4, 2024 · af7204f · af7204f
1 parent 04d3a4c
commit af7204f
Showing 1 changed file with 4 additions and 12 deletions.
diff --git a/airgun/browser.py b/airgun/browser.py
@@ -6,7 +6,6 @@
 from datetime import datetime
 import logging
 import os
-import time
 import urllib
 from urllib.parse import unquote
 
@@ -344,19 +343,12 @@ def get_downloads_list(self):
         downloads_uri = 'chrome://downloads'
         if not self.url.startswith(downloads_uri):
             self.url = downloads_uri
-        time.sleep(3)
         script = (
-            'return downloads.Manager.get().items_'
-            '.filter(e => e.state === "COMPLETE")'
-            '.map(e => e.file_url || e.fileUrl);'
+            'return document.querySelector("downloads-manager")'
+            '.shadowRoot.querySelector("#downloadsList")'
+            '.items.filter(e => e.state == "2")'
+            '.map(e => e.filePath || e.file_path || e.fileUrl || e.file_url);'
         )
-        if self.browser_type == 'chrome':
-            script = (
-                'return document.querySelector("downloads-manager")'
-                '.shadowRoot.querySelector("#downloadsList")'
-                '.items.filter(e => e.state === "COMPLETE")'
-                '.map(e => e.filePath || e.file_path || e.fileUrl || e.file_url);'
-            )
         return self.execute_script(script)
 
     def get_file_content(self, uri):