From 40590fb681222fd79e563c54152dcb1d8d59d44e Mon Sep 17 00:00:00 2001
From: Taylor <taylor.walton@socfortress.co>
Date: Fri, 30 Aug 2024 13:16:18 -0500
Subject: [PATCH] chore: Skip decommissioning if Wazuh Worker Provisioning
 connector is not verified

---
 .../customer_provisioning/services/decommission.py  | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/backend/app/customer_provisioning/services/decommission.py b/backend/app/customer_provisioning/services/decommission.py
index 41418c53..e91937ee 100644
--- a/backend/app/customer_provisioning/services/decommission.py
+++ b/backend/app/customer_provisioning/services/decommission.py
@@ -98,8 +98,19 @@ async def decommission_wazuh_worker(
         ProvisionWorkerResponse: The response object indicating the success or failure of the provisioning operation.
     """
     logger.info(f"Decommissioning Wazuh worker {request}")
+    # Check if the connector is verified
+    if await get_connector_attribute(
+        connector_name="Wazuh Worker Provisioning",
+        column_name="connector_verified",
+        session=session,
+    ) is False:
+        logger.info("Wazuh Worker Provisioning connector is not verified, skipping ...")
+        return DecommissionWorkerResponse(
+            success=False,
+            message="Wazuh Worker Provisioning connector is not verified",
+        )
     api_endpoint = await get_connector_attribute(
-        connector_id=13,
+        connector_name="Wazuh Worker Provisioning",
         column_name="connector_url",
         session=session,
     )