Add Delete ETD Custom Module sample

Author: vijaykanthm

securitycenter/snippets_management_api/event_threat_detection_custom_modules.py

@@ -18,9 +18,10 @@
 import time
 from typing import Dict
 
+from google.api_core.exceptions import NotFound
 from google.cloud import securitycentermanagement_v1
-from google.protobuf.struct_pb2 import Struct
 from google.protobuf.field_mask_pb2 import FieldMask
+from google.protobuf.struct_pb2 import Struct
 
 
 # [START securitycenter_create_event_threat_detection_custom_module]
@@ -89,6 +90,7 @@ def create_event_threat_detection_custom_module(parent: str) -> Dict:
 
 # [END securitycenter_create_event_threat_detection_custom_module]
 
+
 # [START securitycenter_get_event_threat_detection_custom_module]
 def get_event_threat_detection_custom_module(parent: str, module_id: str):
     """
@@ -118,6 +120,7 @@ def get_event_threat_detection_custom_module(parent: str, module_id: str):
         raise e
 # [END securitycenter_get_event_threat_detection_custom_module]
 
+
 # [START securitycenter_list_event_threat_detection_custom_module]
 def list_event_threat_detection_custom_module(parent: str):
     """
@@ -155,6 +158,7 @@ def list_event_threat_detection_custom_module(parent: str):
         raise e
 # [END securitycenter_list_event_threat_detection_custom_module]
 
+
 # [START securitycenter_update_event_threat_detection_custom_module]
 def update_event_threat_detection_custom_module(parent: str, module_id: str):
     """
@@ -194,3 +198,32 @@ def update_event_threat_detection_custom_module(parent: str, module_id: str):
         raise
 
 # [END securitycenter_update_event_threat_detection_custom_module]
+
+
+# [START securitycenter_delete_event_threat_detection_custom_module]
+def delete_event_threat_detection_custom_module(parent: str, module_id: str):
+    """
+    Deletes an Event Threat Detection custom module.
+    Args:
+        parent: Use any one of the following options:
+                - organizations/{organization_id}/locations/{location_id}
+                - folders/{folder_id}/locations/{location_id}
+                - projects/{project_id}/locations/{location_id}
+    Returns:
+        Message that Event Threat Detection custom module is deleted.
+    Raises:
+        NotFound: If the specified custom module does not exist.
+    """
+    client = securitycentermanagement_v1.SecurityCenterManagementClient()
+
+    try:
+        request = securitycentermanagement_v1.DeleteEventThreatDetectionCustomModuleRequest(
+            name=f"{parent}/eventThreatDetectionCustomModules/{module_id}",
+        )
+
+        client.delete_event_threat_detection_custom_module(request=request)
+        print(f"Deleted Event Threat Detection Custom Module Successfully: {module_id}")
+    except NotFound as e:
+        print(f"Custom Module not found: {module_id}")
+        raise e
+# [END securitycenter_delete_event_threat_detection_custom_module]

securitycenter/snippets_management_api/event_threat_detection_custom_modules_test.py

@@ -131,6 +131,7 @@ def test_create_event_threat_detection_custom_module():
     assert response.display_name.startswith(PREFIX)
     assert response.enablement_state == securitycentermanagement_v1.EventThreatDetectionCustomModule.EnablementState.ENABLED
 
+
 @backoff.on_exception(
     backoff.expo, (InternalServerError, ServiceUnavailable, NotFound), max_tries=3
 )
@@ -148,6 +149,7 @@ def test_get_event_threat_detection_custom_module():
     assert response.enablement_state == securitycentermanagement_v1.EventThreatDetectionCustomModule.EnablementState.ENABLED
     print(f"Retrieved Custom Module: {response.name}")
 
+
 @backoff.on_exception(
     backoff.expo, (InternalServerError, ServiceUnavailable, NotFound), max_tries=3
 )
@@ -172,6 +174,7 @@ def test_list_event_threat_detection_custom_module():
         == securitycentermanagement_v1.EventThreatDetectionCustomModule.EnablementState.ENABLED
     )
 
+
 @backoff.on_exception(
     backoff.expo, (InternalServerError, ServiceUnavailable, NotFound), max_tries=3
 )
@@ -187,3 +190,20 @@ def test_update_event_threat_detection_custom_module():
     # Verify that the custom module was created
     assert response.display_name.startswith(PREFIX)
     assert response.enablement_state == securitycentermanagement_v1.EventThreatDetectionCustomModule.EnablementState.DISABLED
+
+
+@backoff.on_exception(
+    backoff.expo, (InternalServerError, ServiceUnavailable, NotFound), max_tries=3
+)
+def test_delete_event_threat_detection_custom_module():
+
+    module_name, module_id = add_custom_module(ORGANIZATION_ID)
+    parent = f"organizations/{ORGANIZATION_ID}/locations/{LOCATION}"
+    try:
+        response = event_threat_detection_custom_modules.delete_event_threat_detection_custom_module(parent, module_id)
+    except Exception as e:
+        pytest.fail(f"delete_event_threat_detection_custom_module() failed: {e}")
+        return
+    assert response is None
+
+    print(f"Custom module was deleted successfully: {module_id}")