Mirrored_Repos/InvenTree
1
0
Fork 0
mirror of https://github.com/inventree/InvenTree synced 2024-08-30 18:33:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improvements for image download helper

Browse Source
This commit is contained in:
Oliver 2022-11-07 13:04:34 +11:00
parent 494a016dad
commit 43b0d3ca55
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
InvenTree/InvenTree/tests.py
View File

@ -275,8 +275,10 @@ class TestHelpers(TestCase):
we will simply try multiple times
"""
tries = 0
with self.assertRaises(expected_error):
while retries > 0:
while tries < retries:
try:
helpers.download_image_from_url(url, timeout=timeout)
@ -285,9 +287,11 @@ class TestHelpers(TestCase):
if type(exc) is expected_error:
# Re-throw this error
raise exc
else:
print("Unexpected error:", type(exc), exc)
time.sleep(30)
retries -= 1
tries += 1
time.sleep(10 * tries)
# Attempt to download an image which throws a 404
dl_helper("https://httpstat.us/404", requests.exceptions.HTTPError, timeout=10)
@ -295,9 +299,6 @@ class TestHelpers(TestCase):
# Attempt to download, but timeout
dl_helper("https://httpstat.us/200?sleep=5000", requests.exceptions.ReadTimeout, timeout=1)
# Attempt to download, but not a valid image
dl_helper("https://httpstat.us/200", TypeError, timeout=10)
large_img = "https://github.com/inventree/InvenTree/raw/master/InvenTree/InvenTree/static/img/paper_splash_large.jpg"
InvenTreeSetting.set_setting('INVENTREE_DOWNLOAD_IMAGE_MAX_SIZE', 1, change_user=None)