Add tests for eu-central and creo deployment

Open-EO · Oct 5, 2023 · 36eaa96 · 36eaa96
1 parent 23e0f6d
commit 36eaa96
Showing 1 changed file with 68 additions and 3 deletions.
diff --git a/tests/test_units.py b/tests/test_units.py
@@ -986,9 +986,40 @@ def test_get_roles(filename, expected_roles):
     "endpoint,access_token,api_responses,min_exec_time,should_raise_error,expected_error",
     [
         ("https://services.sentinel-hub.com", None, [{"status": 200}], 0, False, None),
-        ("https://services.sentinel-hub.com", None, [{"status": 429}], 0, True, "Too Many Requests"),
-        ("https://services.sentinel-hub.com", "some-token", [{"status": 200}], 0, False, None),
-        ("https://services.sentinel-hub.com", "some-token", [{"status": 429}], 0, True, "Too Many Requests"),
+        (
+            "https://services.sentinel-hub.com",
+            None,
+            [
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "7"}},
+                {"status": 200},
+            ],
+            9,
+            False,
+            None,
+        ),
+        (
+            "https://services.sentinel-hub.com",
+            None,
+            [
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+            ],
+            0,
+            True,
+            "Out of retries.",
+        ),
+        (
+            "https://services.sentinel-hub.com",
+            None,
+            [{"status": 429, "headers": {"retry-after": "2"}}, {"status": 500}],
+            0,
+            True,
+            "Internal Server Error",
+        ),
+        ("https://services-uswest2.sentinel-hub.com", None, [{"status": 200}], 0, False, None),
         (
             "https://services-uswest2.sentinel-hub.com",
             None,
@@ -1022,6 +1053,40 @@ def test_get_roles(filename, expected_roles):
             True,
             "Internal Server Error",
         ),
+        ("https://creodias.sentinel-hub.com", None, [{"status": 200}], 0, False, None),
+        (
+            "https://creodias.sentinel-hub.com",
+            None,
+            [
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "7"}},
+                {"status": 200},
+            ],
+            9,
+            False,
+            None,
+        ),
+        (
+            "https://creodias.sentinel-hub.com",
+            None,
+            [
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+                {"status": 429, "headers": {"retry-after": "2"}},
+            ],
+            0,
+            True,
+            "Out of retries.",
+        ),
+        (
+            "https://creodias.sentinel-hub.com",
+            None,
+            [{"status": 429, "headers": {"retry-after": "2"}}, {"status": 500}],
+            0,
+            True,
+            "Internal Server Error",
+        ),
     ],
 )
 def test_processing_api_request(