feat: created get_last_failed_deployment

tdp/cli/commands/plan/resume.py

@@ -32,7 +32,7 @@ def resume(
     """Resume a failed deployment."""
     with Dao(db_engine, commit_on_exit=True) as dao:
         if id is None:
-            deployment_to_resume = dao.get_last_deployment()
+            deployment_to_resume = dao.get_last_failed_deployment()
             if not deployment_to_resume:
                 raise click.ClickException("No deployment found.")
             click.echo("Creating a deployment plan to resume latest deployment.")

tdp/dao.py

@@ -285,6 +285,16 @@ def get_planned_deployment(self) -> Optional[DeploymentModel]:
             self.session.query(DeploymentModel).filter_by(state="PLANNED").one_or_none()
         )
 
+    def get_last_failed_deployment(self) -> Optional[DeploymentModel]:
+        """Get the last failed deployment."""
+        self._check_session()
+        return (
+            self.session.query(DeploymentModel)
+            .filter_by(state="FAILURE")
+            .order_by(desc(DeploymentModel.id))
+            .first()
+        )
+
     def get_last_deployment(self) -> Optional[DeploymentModel]:
         """Get the last deployment."""
         self._check_session()

tests/unit/test_dao.py

@@ -229,6 +229,32 @@ def test_get_last_deployment(db_engine):
         )
 
 
+@pytest.mark.parametrize("db_engine", [True], indirect=True)
+def test_get_last_failed_deployment(db_engine):
+    with create_session(db_engine) as session:
+        session.add(
+            DeploymentModel(
+                id=2,
+                state=DeploymentStateEnum.FAILURE,
+            )
+        )
+        session.add(
+            DeploymentModel(
+                id=3,
+                state=DeploymentStateEnum.SUCCESS,
+            )
+        )
+        session.commit()
+    with Dao(db_engine) as dao:
+        assert assert_equal_values_in_model(
+            dao.get_last_failed_deployment(),
+            DeploymentModel(
+                id=2,
+                state=DeploymentStateEnum.FAILURE,
+            ),
+        )
+
+
 @pytest.mark.parametrize("db_engine", [True], indirect=True)
 def test_get_deployments(db_engine):
     with create_session(db_engine) as session: