From 9e754f280d1a4c0f99bb8066bf3c73c8a0c58ea7 Mon Sep 17 00:00:00 2001
From: Austin Gibbons <agibbons@mitre.org>
Date: Fri, 26 Apr 2024 17:15:52 -0400
Subject: [PATCH] Retry failed downloads.

---
 tests/download-test-images.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/download-test-images.py b/tests/download-test-images.py
index b4eaf54..cabe465 100644
--- a/tests/download-test-images.py
+++ b/tests/download-test-images.py
@@ -1,7 +1,6 @@
 import os
 import shutil
 import sys
-import time
 
 from urllib.parse import urlparse
 from zipfile import ZipFile
@@ -88,11 +87,14 @@ def download_dftt_images():
     for url in dftt_urls:
         download(url)
 
-    time.sleep(15)
-
     for filename in os.listdir("."):
         _, ext = os.path.splitext(filename)
         if ext == ".zip":
+            while os.path.getsize(filename) == 154:
+                for url in dftt_urls:
+                    if filename in url:
+                        print(f"Retrying download of {filename}")
+                        download(url)
             print(f"unzipping: {filename}")
             unzip(filename)