commit 870578735024512322fb0a5e94c09711c4eae1c6
author Jacek Klimkowicz <klimkowicz@google.com> Wed Nov 19 13:26:59 2025
committer Chromeos LUCI <chromeos-scoped@luci-project-accounts.iam.gserviceaccount.com> Thu Nov 20 08:55:04 2025
tree 1a9fd4a1d1f0b7289357a9202f513cbc16fdcb9d
parent ff8cbd0889e3d260afa1396b18c9ec80d0db4bb5

labstation: increase timeout in image_downloader

Due to the current test image size 15 mins timeout is not enough
anymore.

BUG=b:461376696
TEST=manual

Change-Id: I1799033987501bd869263c3c9092644e0c75d480
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/labstation/+/7168363
Reviewed-by: Garry Wang <xianuowang@chromium.org>
Commit-Queue: Jacek Klimkowicz <klimkowicz@google.com>
Auto-Submit: Jacek Klimkowicz <klimkowicz@google.com>
Tested-by: Jacek Klimkowicz <klimkowicz@google.com>

image_downloader/image_downloader.py

diff --git a/image_downloader/image_downloader.py b/image_downloader/image_downloader.py
index 2fb0e68..8ff892d 100644
--- a/image_downloader/image_downloader.py
+++ b/image_downloader/image_downloader.py
@@ -13,12 +13,13 @@
 import shutil
 import subprocess
 import sys
+from urllib.error import ContentTooShortError
 from urllib.error import URLError
 from urllib.request import urlopen
 
 import requests
 from requests.exceptions import RequestException
-from urllib.error import ContentTooShortError
+
 
 LOG_FORMAT = "%(levelname)s - %(filename)s:%(lineno)d:%(funcName)s : %(message)s"
 
@@ -155,15 +156,18 @@
         """Downloads the image file from the network to the USB device."""
 
         # Check the webserver is working by getting the first 100 bytes of the file.
+        # If the image path requests extraction use the archive instead.
         self._logger.debug("Testing webserver")
-        response = requests.head(self.image_path, timeout=900)
+        base_image_path = re.sub('\?file=.*$', '', self.image_path)
+        base_image_path = re.sub('/extract/', '/download/', base_image_path)
+        response = requests.head(base_image_path, timeout=900)
         response.raise_for_status()
         self._logger.debug("Test Headers %s" % response.headers)
         self._logger.debug("Webserver test pass")
 
         def progress_bar(block_num, block_size, total_size):
             """Displays a progress bar in the logs."""
-            if block_num and block_num % 1000 == 0:
+            if block_num and block_num % 10000 == 0:
                 self._logger.debug(
                     "Show progress Block Num %d Block Size %d  Total %d"
                     % (block_num, block_size, total_size)
@@ -197,7 +201,7 @@
             bs = 4096
 
         with contextlib.closing(
-            requests.get(self.image_path, timeout=900, stream=True)
+            requests.get(self.image_path, timeout=3000, stream=True)
         ) as resp:
             headers = resp.headers
             self._logger.debug("Block size %d", bs)