From ffa682eaa87a040457073a671a6c4fcf67454302 Mon Sep 17 00:00:00 2001
From: Matt Craddock <5796417+craddm@users.noreply.github.com>
Date: Wed, 15 May 2024 15:33:39 +0000
Subject: [PATCH] Use Pulumi error for consistency

---
 data_safe_haven/commands/shm.py | 4 ++--
 data_safe_haven/commands/sre.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/data_safe_haven/commands/shm.py b/data_safe_haven/commands/shm.py
index a4b228e85f..51fce90e6a 100644
--- a/data_safe_haven/commands/shm.py
+++ b/data_safe_haven/commands/shm.py
@@ -6,7 +6,7 @@
 
 from data_safe_haven.config import Config, DSHPulumiConfig
 from data_safe_haven.context import ContextSettings
-from data_safe_haven.exceptions import DataSafeHavenError, DataSafeHavenInputError
+from data_safe_haven.exceptions import DataSafeHavenError, DataSafeHavenPulumiError
 from data_safe_haven.external import GraphApi
 from data_safe_haven.infrastructure import SHMProjectManager
 
@@ -105,7 +105,7 @@ def teardown() -> None:
             stack.teardown()
         except Exception as exc:
             msg = f"Unable to teardown Pulumi infrastructure.\n{exc}"
-            raise DataSafeHavenInputError(msg) from exc
+            raise DataSafeHavenPulumiError(msg) from exc
 
         # Remove information from config file
         del pulumi_config[context.shm_name]
diff --git a/data_safe_haven/commands/sre.py b/data_safe_haven/commands/sre.py
index 6e8aefed14..936c2c1d24 100644
--- a/data_safe_haven/commands/sre.py
+++ b/data_safe_haven/commands/sre.py
@@ -6,7 +6,7 @@
 
 from data_safe_haven.config import Config, DSHPulumiConfig
 from data_safe_haven.context import ContextSettings
-from data_safe_haven.exceptions import DataSafeHavenError, DataSafeHavenInputError
+from data_safe_haven.exceptions import DataSafeHavenError, DataSafeHavenPulumiError
 from data_safe_haven.external import GraphApi
 from data_safe_haven.functions import sanitise_sre_name
 from data_safe_haven.infrastructure import SHMProjectManager, SREProjectManager
@@ -176,7 +176,7 @@ def teardown(
             stack.teardown()
         except Exception as exc:
             msg = f"Unable to teardown Pulumi infrastructure.\n{exc}"
-            raise DataSafeHavenInputError(msg) from exc
+            raise DataSafeHavenPulumiError(msg) from exc
 
         # Remove Pulumi project from Pulumi config file
         del pulumi_config[name]