Make sure we do all in our power to release an environment

eiffel-community · Apr 25, 2024 · 08bacf7 · 08bacf7
1 parent 037a9b7
commit 08bacf7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/src/execution_space_provider/utilities/external_provider.py b/src/execution_space_provider/utilities/external_provider.py
@@ -25,7 +25,7 @@
 from etos_lib.lib.http import Http
 from jsontas.jsontas import JsonTas
 from packageurl import PackageURL
-from requests.exceptions import HTTPError
+from requests.exceptions import HTTPError, ConnectionError as RequestsConnectionError
 from urllib3.util import Retry
 
 from environment_provider.lib.encrypt import encrypt
@@ -132,6 +132,11 @@ def checkin(self, execution_space: ExecutionSpace) -> None:
                     raise ExecutionSpaceCheckinFailed(
                         f"Unable to check in {execution_spaces} " f"({response.get('error')})"
                     )
+            except RequestsConnectionError as error:
+                if "connection refused" in str(error).lower():
+                    self.logger.error("Error connecting to %r: %r", host, error)
+                    continue
+                raise
             except ConnectionError:
                 self.logger.error("Error connecting to %r", host)
                 continue

diff --git a/src/iut_provider/utilities/external_provider.py b/src/iut_provider/utilities/external_provider.py
@@ -25,7 +25,7 @@
 from etos_lib.lib.http import Http
 from jsontas.jsontas import JsonTas
 from packageurl import PackageURL
-from requests.exceptions import HTTPError
+from requests.exceptions import HTTPError, ConnectionError as RequestsConnectionError
 from urllib3.util import Retry
 
 from ..exceptions import IutCheckinFailed, IutCheckoutFailed, IutNotAvailable
@@ -122,6 +122,11 @@ def checkin(self, iut: Iut) -> None:
                 response = response.json()
                 if response.get("error") is not None:
                     raise IutCheckinFailed(f"Unable to check in {iuts} ({response.get('error')})")
+            except RequestsConnectionError as error:
+                if "connection refused" in str(error).lower():
+                    self.logger.error("Error connecting to %r: %r", host, error)
+                    continue
+                raise
             except ConnectionError:
                 self.logger.error("Error connecting to %r", host)
                 continue

diff --git a/src/log_area_provider/utilities/external_provider.py b/src/log_area_provider/utilities/external_provider.py
@@ -25,7 +25,7 @@
 from etos_lib.lib.http import Http
 from jsontas.jsontas import JsonTas
 from packageurl import PackageURL
-from requests.exceptions import HTTPError
+from requests.exceptions import HTTPError, ConnectionError as RequestsConnectionError
 from urllib3.util import Retry
 
 from ..exceptions import LogAreaCheckinFailed, LogAreaCheckoutFailed, LogAreaNotAvailable
@@ -126,6 +126,11 @@ def checkin(self, log_area: LogArea) -> None:
                     raise LogAreaCheckinFailed(
                         f"Unable to check in {log_areas} ({response.get('error')})"
                     )
+            except RequestsConnectionError as error:
+                if "connection refused" in str(error).lower():
+                    self.logger.error("Error connecting to %r: %r", host, error)
+                    continue
+                raise
             except ConnectionError:
                 self.logger.error("Error connecting to %r", host)
                 continue