Add refresh logic for database owner in Set-SqlDscDatabaseOwner function and mock implementation in tests

Author: johlju

source/Public/Set-SqlDscDatabaseOwner.ps1

@@ -150,6 +150,16 @@ function Set-SqlDscDatabaseOwner
                 try
                 {
                     $sqlDatabaseObject.SetOwner($OwnerName)
+
+                    <#
+                        Refresh the database object to get the updated owner property if:
+                        - PassThru is specified (user wants the updated object back)
+                        - Using DatabaseObject parameter set (user's object reference should be updated)
+                    #>
+                    if ($PassThru.IsPresent -or $PSCmdlet.ParameterSetName -eq 'DatabaseObjectSet')
+                    {
+                        $sqlDatabaseObject.Refresh()
+                    }
                 }
                 catch
                 {

tests/Unit/Public/Set-SqlDscDatabaseOwner.Tests.ps1

@@ -72,6 +72,9 @@ Describe 'Set-SqlDscDatabaseOwner' -Tag 'Public' {
                 param($OwnerName)
                 $this.Owner = $OwnerName
             } -Force
+            $mockDatabaseObject | Add-Member -MemberType 'ScriptMethod' -Name 'Refresh' -Value {
+                # Mock implementation - in real SMO this updates properties from server
+            } -Force
 
             $mockServerObject = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Server'
             $mockServerObject | Add-Member -MemberType 'NoteProperty' -Name 'InstanceName' -Value 'TestInstance' -Force
@@ -124,6 +127,9 @@ Describe 'Set-SqlDscDatabaseOwner' -Tag 'Public' {
                 param($OwnerName)
                 $this.Owner = $OwnerName
             } -Force
+            $mockDatabaseObject | Add-Member -MemberType 'ScriptMethod' -Name 'Refresh' -Value {
+                # Mock implementation - in real SMO this updates properties from server
+            } -Force
         }
 
         It 'Should set database owner successfully' {
@@ -158,6 +164,9 @@ Describe 'Set-SqlDscDatabaseOwner' -Tag 'Public' {
                 param($OwnerName)
                 $this.Owner = $OwnerName
             } -Force
+            $mockDatabaseObject | Add-Member -MemberType 'ScriptMethod' -Name 'Refresh' -Value {
+                # Mock implementation - in real SMO this updates properties from server
+            } -Force
         }
 
         It 'Should still call SetOwner even when owner is already set to desired value' {
@@ -182,6 +191,9 @@ Describe 'Set-SqlDscDatabaseOwner' -Tag 'Public' {
                 param($OwnerName)
                 throw 'Simulated SetOwner() failure'
             } -Force
+            $mockDatabaseObject | Add-Member -MemberType 'ScriptMethod' -Name 'Refresh' -Value {
+                # Mock implementation - in real SMO this updates properties from server
+            } -Force
         }
 
         It 'Should throw error when SetOwner() fails' {