Add: error handling to tests

source/tests/Test-SafeAttachmentsPolicy.ps1

@@ -9,39 +9,48 @@ function Test-SafeAttachmentsPolicy {
         # Dot source the class script if necessary
         #. .\source\Classes\CISAuditResult.ps1
         # Initialization code, if needed
+        $recnum = "2.1.4"
     }
 
     process {
-        # 2.1.4 (L2) Ensure Safe Attachments policy is enabled
+        try {
+            # 2.1.4 (L2) Ensure Safe Attachments policy is enabled
 
-        # Retrieve all Safe Attachment policies where Enable is set to True
-        $safeAttachmentPolicies = Get-SafeAttachmentPolicy | Where-Object { $_.Enable -eq $true }
+            # Retrieve all Safe Attachment policies where Enable is set to True
+            $safeAttachmentPolicies = Get-SafeAttachmentPolicy | Where-Object { $_.Enable -eq $true }
 
-        # Determine result and details based on the presence of enabled policies
-        $result = $null -ne $safeAttachmentPolicies -and $safeAttachmentPolicies.Count -gt 0
-        $details = if ($result) {
-            "Enabled Safe Attachments Policies: $($safeAttachmentPolicies.Name -join ', ')"
-        }
-        else {
-            "No Safe Attachments Policies are enabled."
-        }
+            # Determine result and details based on the presence of enabled policies
+            $result = $null -ne $safeAttachmentPolicies -and $safeAttachmentPolicies.Count -gt 0
+            $details = if ($result) {
+                "Enabled Safe Attachments Policies: $($safeAttachmentPolicies.Name -join ', ')"
+            }
+            else {
+                "No Safe Attachments Policies are enabled."
+            }
 
-        $failureReasons = if ($result) {
-            "N/A"
-        }
-        else {
-            "Safe Attachments policy is not enabled."
-        }
+            $failureReasons = if ($result) {
+                "N/A"
+            }
+            else {
+                "Safe Attachments policy is not enabled."
+            }
 
-        # Create and populate the CISAuditResult object
-        $params = @{
-            Rec            = "2.1.4"
-            Result         = $result
-            Status         = if ($result) { "Pass" } else { "Fail" }
-            Details        = $details
-            FailureReason  = $failureReasons
+            # Create and populate the CISAuditResult object
+            $params = @{
+                Rec           = $recnum
+                Result        = $result
+                Status        = if ($result) { "Pass" } else { "Fail" }
+                Details       = $details
+                FailureReason = $failureReasons
+            }
+            $auditResult = Initialize-CISAuditResult @params
+        }
+        catch {
+            Write-Error "An error occurred during the test: $_"
+
+            # Call Initialize-CISAuditResult with error parameters
+            $auditResult = Initialize-CISAuditResult -Rec $recnum -Failure
         }
-        $auditResult = Initialize-CISAuditResult @params
     }
 
     end {