Add mock GetCurrentState method to SqlAudit, SqlDatabasePermission, and SqlPermission tests

Author: johlju

tests/Unit/Classes/SqlAudit.Tests.ps1

@@ -243,6 +243,14 @@ Describe 'SqlAudit\Set()' -Tag 'Set' {
                 InstanceName = 'NamedInstance'
                 Path         = 'C:\Temp'
             } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockAuditName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru |
                 # Mock method Modify which is called by the base method Set().
                 Add-Member -Force -MemberType 'ScriptMethod' -Name 'Modify' -Value {
                     $script:mockMethodModifyCallCount += 1
@@ -314,7 +322,15 @@ Describe 'SqlAudit\Test()' -Tag 'Test' {
                 Name         = 'MockAuditName'
                 InstanceName = 'NamedInstance'
                 Path         = 'C:\Temp'
-            }
+            } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockAuditName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru
         }
     }
 

tests/Unit/Classes/SqlDatabasePermission.Tests.ps1

@@ -924,6 +924,15 @@ Describe 'SqlDatabasePermission\Set()' -Tag 'Set' {
                     }
                 )
             } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockUserName'
+                        DatabaseName = 'MockDatabaseName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru |
                 # Mock method Modify which is called by the base method Set().
                 Add-Member -Force -MemberType 'ScriptMethod' -Name 'Modify' -Value {
                     $script:mockMethodModifyCallCount += 1
@@ -1013,7 +1022,16 @@ Describe 'SqlDatabasePermission\Test()' -Tag 'Test' {
                         Permission = @()
                     }
                 )
-            }
+            } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockUserName'
+                        DatabaseName = 'MockDatabaseName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru
         }
     }
 

tests/Unit/Classes/SqlPermission.Tests.ps1

@@ -902,6 +902,14 @@ Describe 'SqlPermission\Set()' -Tag 'Set' {
                     }
                 )
             } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockUserName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru |
                 # Mock method Modify which is called by the base method Set().
                 Add-Member -Force -MemberType 'ScriptMethod' -Name 'Modify' -Value {
                     $script:mockMethodModifyCallCount += 1
@@ -990,7 +998,15 @@ Describe 'SqlPermission\Test()' -Tag 'Test' {
                         Permission = @()
                     }
                 )
-            }
+            } |
+                # Mock method GetCurrentState() which is called by the base method Get()
+                Add-Member -Force -MemberType 'ScriptMethod' -Name 'GetCurrentState' -Value {
+                    return [System.Collections.Hashtable] @{
+                        Name         = 'MockUserName'
+                        InstanceName = 'NamedInstance'
+                        ServerName   = Get-ComputerName
+                    }
+                } -PassThru
         }
     }