feat: add deleting status support (#341)

Author: stepansergeevitch

src/firebolt/model/V2/engine.py

@@ -245,7 +245,7 @@ def update(
     def delete(self) -> None:
         """Delete an engine."""
         self.refresh()
-        if self.current_status == EngineStatus.DROPPING:
+        if self.current_status in [EngineStatus.DROPPING, EngineStatus.DELETING]:
             return
         with self._service._connection.cursor() as c:
             c.execute(self.DROP_SQL.format(self.name))

src/firebolt/service/V2/types.py

@@ -48,6 +48,7 @@ class EngineStatus(Enum):
     DROPPING = "Dropping"
     REPAIRING = "Repairing"
     FAILED = "Failed"
+    DELETING = "Deleting"
 
     def __str__(self) -> str:
         return self.value

tests/unit/service/V2/test_engine.py

@@ -4,13 +4,14 @@
 from pytest_httpx import HTTPXMock
 
 from firebolt.model.V2.database import Database
-from firebolt.model.V2.engine import Engine
+from firebolt.model.V2.engine import Engine, EngineStatus
 from firebolt.service.manager import ResourceManager
 from firebolt.service.V2.types import EngineType
 from firebolt.utils.exception import (
     EngineNotFoundError,
     NoAttachedDatabaseError,
 )
+from tests.unit.service.V2.conftest import get_objects_from_db_callback
 
 
 def test_engine_create(
@@ -195,3 +196,22 @@ def test_engine_get_by_name(
     engine = resource_manager.engines.get_by_name(mock_engine.name)
 
     assert engine == mock_engine
+
+
+def test_engine_deleteting(
+    httpx_mock: HTTPXMock,
+    resource_manager: ResourceManager,
+    instance_type_callback: Callable,
+    instance_type_url: str,
+    system_engine_no_db_query_url: str,
+    mock_engine: Engine,
+):
+    mock_engine.current_status = "Deleting"
+    get_engine_callback = get_objects_from_db_callback([mock_engine])
+
+    httpx_mock.add_callback(instance_type_callback, url=instance_type_url)
+    httpx_mock.add_callback(get_engine_callback, url=system_engine_no_db_query_url)
+
+    engine = resource_manager.engines.get_by_name(mock_engine.name)
+
+    assert engine.current_status == EngineStatus.DELETING