Mock New-SqlDscDataFile and New-SqlDscFileGroup in tests to avoid type conflicts with PowerShell class types

Author: johlju

tests/Unit/Public/New-SqlDscDatabaseSnapshot.Tests.ps1

@@ -98,6 +98,36 @@ Describe 'New-SqlDscDatabaseSnapshot' -Tag 'Public' {
                 return $mockSourceDatabase
             }
 
+            # Mock New-SqlDscDataFile and New-SqlDscFileGroup to avoid type conflicts
+            # between test scope and module scope for PowerShell class types.
+            Mock -CommandName 'New-SqlDscDataFile' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $FileName)
+
+                    [DatabaseFileSpec]@{
+                        Name = $Name
+                        FileName = $FileName
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    FileName = $FileName
+                }
+            }
+
+            Mock -CommandName 'New-SqlDscFileGroup' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $Files)
+
+                    [DatabaseFileGroupSpec]@{
+                        Name = $Name
+                        Files = $Files
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    Files = $Files
+                }
+            }
+
             Mock -CommandName 'New-SqlDscDatabase' -MockWith {
                 $mockSnapshotDatabase = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Database'
                 $mockSnapshotDatabase | Add-Member -MemberType 'NoteProperty' -Name 'Name' -Value $Name -Force
@@ -242,6 +272,36 @@ Describe 'New-SqlDscDatabaseSnapshot' -Tag 'Public' {
                 return $mockDatabaseObject
             }
 
+            # Mock New-SqlDscDataFile and New-SqlDscFileGroup to avoid type conflicts
+            # between test scope and module scope for PowerShell class types.
+            Mock -CommandName 'New-SqlDscDataFile' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $FileName)
+
+                    [DatabaseFileSpec]@{
+                        Name = $Name
+                        FileName = $FileName
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    FileName = $FileName
+                }
+            }
+
+            Mock -CommandName 'New-SqlDscFileGroup' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $Files)
+
+                    [DatabaseFileGroupSpec]@{
+                        Name = $Name
+                        Files = $Files
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    Files = $Files
+                }
+            }
+
             Mock -CommandName 'New-SqlDscDatabase' -MockWith {
                 $mockSnapshotDatabase = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Database'
                 $mockSnapshotDatabase | Add-Member -MemberType 'NoteProperty' -Name 'Name' -Value $Name -Force
@@ -342,6 +402,36 @@ Describe 'New-SqlDscDatabaseSnapshot' -Tag 'Public' {
                 return $mockDevSourceDatabase
             }
 
+            # Mock New-SqlDscDataFile and New-SqlDscFileGroup to avoid type conflicts
+            # between test scope and module scope for PowerShell class types.
+            Mock -CommandName 'New-SqlDscDataFile' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $FileName)
+
+                    [DatabaseFileSpec]@{
+                        Name = $Name
+                        FileName = $FileName
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    FileName = $FileName
+                }
+            }
+
+            Mock -CommandName 'New-SqlDscFileGroup' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $Files)
+
+                    [DatabaseFileGroupSpec]@{
+                        Name = $Name
+                        Files = $Files
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    Files = $Files
+                }
+            }
+
             Mock -CommandName 'New-SqlDscDatabase' -MockWith {
                 $mockSnapshotDatabase = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Database'
                 $mockSnapshotDatabase | Add-Member -MemberType 'NoteProperty' -Name 'Name' -Value $Name -Force
@@ -393,6 +483,36 @@ Describe 'New-SqlDscDatabaseSnapshot' -Tag 'Public' {
                 return $mockEvalSourceDatabase
             }
 
+            # Mock New-SqlDscDataFile and New-SqlDscFileGroup to avoid type conflicts
+            # between test scope and module scope for PowerShell class types.
+            Mock -CommandName 'New-SqlDscDataFile' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $FileName)
+
+                    [DatabaseFileSpec]@{
+                        Name = $Name
+                        FileName = $FileName
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    FileName = $FileName
+                }
+            }
+
+            Mock -CommandName 'New-SqlDscFileGroup' -MockWith {
+                InModuleScope -ScriptBlock {
+                    param ($Name, $Files)
+
+                    [DatabaseFileGroupSpec]@{
+                        Name = $Name
+                        Files = $Files
+                    }
+                } -Parameters @{
+                    Name = $Name
+                    Files = $Files
+                }
+            }
+
             Mock -CommandName 'New-SqlDscDatabase' -MockWith {
                 $mockSnapshotDatabase = New-Object -TypeName 'Microsoft.SqlServer.Management.Smo.Database'
                 $mockSnapshotDatabase | Add-Member -MemberType 'NoteProperty' -Name 'Name' -Value $Name -Force