add: finally block to always disconnect

source/Public/Invoke-M365SecurityAudit.ps1

@@ -248,19 +248,11 @@ function Invoke-M365SecurityAudit {
         $currentTestIndex = 0
 
         # Establishing connections if required
-        try {
         $actualUniqueConnections = Get-UniqueConnection -Connections $requiredConnections
         if (!($DoNotConnect) -and $PSCmdlet.ShouldProcess("Establish connections to Microsoft 365 services: $($actualUniqueConnections -join ', ')", "Connect")) {
+            try {
                 Write-Information "Establishing connections to Microsoft 365 services: $($actualUniqueConnections -join ', ')" -InformationAction Continue
                 Connect-M365Suite -TenantAdminUrl $TenantAdminUrl -RequiredConnections $requiredConnections -SkipConfirmation:$DoNotConfirmConnections
-            }
-        }
-        catch {
-            Write-Host "Execution aborted: $_" -ForegroundColor Red
-            break
-        }
-
-
 
                 Write-Information "A total of $($totalTests) tests were selected to run..." -InformationAction Continue
                 # Import the test functions
@@ -291,12 +283,19 @@ function Invoke-M365SecurityAudit {
                     }
                 }
             }
-
-    End {
+            catch {
+                Write-Error "An error occurred: $_"
+            }
+            finally {
                 if (!($DoNotDisconnect) -and $PSCmdlet.ShouldProcess("Disconnect from Microsoft 365 services: $($actualUniqueConnections -join ', ')", "Disconnect")) {
                     # Clean up sessions
                     Disconnect-M365Suite -RequiredConnections $requiredConnections
                 }
+            }
+        }
+    }
+
+    End {
         if ($PSCmdlet.ShouldProcess("Measure and display audit results for $($totalTests) tests", "Measure")) {
             # Call the private function to calculate and display results
             Measure-AuditResult -AllAuditResults $allAuditResults -FailedTests $script:FailedTests
@@ -316,3 +315,4 @@ function Invoke-M365SecurityAudit {
     }
 }
 
+