From 66f6692591622a35ed130c78ea7d0f5d04bda9e5 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Fri, 13 Oct 2023 17:14:29 +0200
Subject: [PATCH] Fix unit test
---
tests/test_extract.py | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/tests/test_extract.py b/tests/test_extract.py
index f708c337..7ae10ecf 100644
--- a/tests/test_extract.py
+++ b/tests/test_extract.py
@@ -587,13 +587,16 @@ def test_download_image_error_try_max(responses):
responses.GET,
fixed_url,
status=200,
+ body=next((FIXTURES / "prediction" / "images").iterdir()).read_bytes(),
)
image = download_image(url)
assert image
- assert len(responses.calls) == 6
- assert list(map(attrgetter("request.url"), responses.calls)) == [url] * 5 + [
+ # We try 3 times with the first URL
+ # Then the first try with the new URL is successful
+ assert len(responses.calls) == 4
+ assert list(map(attrgetter("request.url"), responses.calls)) == [url] * 3 + [
fixed_url
]
--
GitLab