def test_delete_workload_failure(self) -> None: """Test workload deletion - error due to exception.""" with patch("os.path.isfile") as fake_isfile, patch( "shutil.rmtree", ) as fake_rmtree, patch.object( Workdir, "load", autospec=True) as fake_load, patch.object( Workdir, "dump", ) as fake_dump, self.assertRaises(InternalException, ): fake_isfile.return_value = True fake_rmtree.side_effect = _fake_raise_exception_mock fake_load.side_effect = _fake_load_mock fake_dump.return_value = None workdir = Workdir(workspace_path="/tmp/lpot") workdir.delete_workload("workload_1")
def test_delete_workload_success(self) -> None: """Test workload deletion - success.""" with patch("os.path.isfile") as fake_isfile, patch( "shutil.rmtree", ) as fake_rmtree, patch.object( Workdir, "load", autospec=True) as fake_load, patch.object( Workdir, "dump", ) as fake_dump: fake_isfile.return_value = True fake_rmtree.return_value = None fake_load.side_effect = _fake_load_mock fake_dump.return_value = None workdir = Workdir(workspace_path="/tmp/lpot") workdir.delete_workload("workload_1") self.assertNotIn("workload_1", workdir.workloads_data.get("workloads", {})) self.assertIn("workload_2", workdir.workloads_data.get("workloads", {}))
def delete_workload(data: dict) -> Dict[str, Any]: """Delete workload based on ID.""" workdir = Workdir(workspace_path=os.environ["HOME"]) workdir.delete_workload(data["request_id"]) return {"message": "SUCCESS"}