Refactor Suspend-SqlDscDatabase tests to track KillAllProcesses calls and update mock behavior

Author: johlju

tests/Unit/Public/Suspend-SqlDscDatabase.Tests.ps1

@@ -223,33 +223,35 @@ Describe 'Suspend-SqlDscDatabase' -Tag 'Public' {
             $mockDatabaseObject = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Database' -ArgumentList @($mockServerObject, 'TestDatabase')
             $mockDatabaseObject.Status = [Microsoft.SqlServer.Management.Smo.DatabaseStatus]::Normal
 
-            $script:setOfflineForceValue = $null
+            $script:killAllProcessesCalled = $false
 
-            # Override SetOffline to track the Force parameter
-            $mockDatabaseObject | Add-Member -MemberType ScriptMethod -Name SetOffline -Value {
-                param($ForceDisconnect = $false)
+            # Override KillAllProcesses to track if it was called
+            $mockServerObject | Add-Member -MemberType ScriptMethod -Name KillAllProcesses -Value {
+                param($DatabaseName)
 
-                $script:setOfflineForceValue = $ForceDisconnect
-                $this.Status = [Microsoft.SqlServer.Management.Smo.DatabaseStatus]::Offline
+                $script:killAllProcessesCalled = $true
             } -Force
 
             Mock -CommandName 'Get-SqlDscDatabase' -MockWith {
                 return $mockDatabaseObject
             }
         }
 
-        It 'Should call SetOffline with Force when Force parameter is specified' {
+        It 'Should call KillAllProcesses when Force parameter is specified' {
+            $script:killAllProcessesCalled = $false
+
             Suspend-SqlDscDatabase -ServerObject $mockServerObject -Name 'TestDatabase' -Force -Confirm:$false
 
-            $script:setOfflineForceValue | Should -BeTrue
+            $script:killAllProcessesCalled | Should -BeTrue
         }
 
-        It 'Should call SetOffline without Force when Force parameter is not specified' {
+        It 'Should not call KillAllProcesses when Force parameter is not specified' {
             $mockDatabaseObject.Status = [Microsoft.SqlServer.Management.Smo.DatabaseStatus]::Normal
+            $script:killAllProcessesCalled = $false
 
             Suspend-SqlDscDatabase -ServerObject $mockServerObject -Name 'TestDatabase' -Confirm:$false
 
-            $script:setOfflineForceValue | Should -BeFalse
+            $script:killAllProcessesCalled | Should -BeFalse
         }
     }
 
@@ -292,7 +294,16 @@ Describe 'Suspend-SqlDscDatabase' -Tag 'Public' {
         }
 
         It 'Should throw a terminating error when SetOffline fails' {
-            { Suspend-SqlDscDatabase -ServerObject $mockServerObject -Name 'TestDatabase' -Force } | Should -Throw -ExpectedMessage '*Failed to take database*offline*'
+            { Suspend-SqlDscDatabase -ServerObject $mockServerObject -Name 'TestDatabase' -Confirm:$false } | Should -Throw -ExpectedMessage '*Failed to take database*offline*'
+        }
+
+        It 'Should throw a terminating error when KillAllProcesses fails' {
+            # Override KillAllProcesses to throw an exception
+            $mockServerObject | Add-Member -MemberType ScriptMethod -Name KillAllProcesses -Value {
+                throw 'Failed to kill processes'
+            } -Force
+
+            { Suspend-SqlDscDatabase -ServerObject $mockServerObject -Name 'TestDatabase' -Force -Confirm:$false } | Should -Throw -ExpectedMessage '*Failed to kill processes*'
         }
     }
 }

tests/Unit/Stubs/SMO.cs

@@ -517,6 +517,10 @@ public void Revoke( Microsoft.SqlServer.Management.Smo.ServerPermissionSet permi
         {
         }
 
+        public void KillAllProcesses( string databaseName )
+        {
+        }
+
         // Property for SQL Agent support
         public Microsoft.SqlServer.Management.Smo.Agent.JobServer JobServer { get; set; }
 
@@ -985,7 +989,7 @@ public void SetOnline()
             this.Status = DatabaseStatus.Normal;
         }
 
-        public void SetOffline(bool forceDisconnect = false)
+        public void SetOffline()
         {
             this.Status = DatabaseStatus.Offline;
         }