✅ Add database export test (and testability)

Author: brianary

Export-DatabaseScripts.ps1

@@ -27,12 +27,12 @@ Outputs SQL scripts to files using the default options.
 
 #Requires -Version 3
 #Requires -Modules dbatools
+using namespace Microsoft.SqlServer.Management.Smo
 [CmdletBinding()][OutputType([void])] Param(
 # The database from which to export scripts.
-[Parameter(ValueFromPipeline=$true,Mandatory=$true)]
-[Microsoft.SqlServer.Management.Smo.Database] $Database,
+[Parameter(ValueFromPipeline=$true,Mandatory=$true)][Database] $Database,
 # Controls how the scripts are generated.
-[Microsoft.SqlServer.Management.Smo.ScriptingOptions] $Options = (New-DbaScriptingOption)
+[ScriptingOptions] $Options = (New-DbaScriptingOption)
 )
 Begin
 {
@@ -61,8 +61,7 @@ Begin
 	{
 		[CmdletBinding()] Param(
 		[Parameter(Position=0,Mandatory=$true)][string] $Subfolder,
-		[Parameter(Mandatory=$true,ValueFromPipeline=$true)]
-		[Microsoft.SqlServer.Management.Smo.ScriptNameObjectBase] $InputObject
+		[Parameter(Mandatory=$true,ValueFromPipeline=$true)][ScriptNameObjectBase] $InputObject
 		)
 		if(!$InputObject -or ($InputObject |Test-SystemObject)) {return}
 		$InputObject |Export-DbaScript -ScriptingOptionsObject $Options -FilePath ($InputObject |Get-ScriptName $Subfolder)
@@ -86,7 +85,7 @@ Begin
 		$Database.AsymmetricKeys |Export-DatabaseScript 'Security/Asymmetric Keys'
 		$Database.Certificates |Export-DatabaseScript 'Security/Certificates'
 		$Database.Roles |
-			Where-Object {$_ -isnot [Microsoft.SqlServer.Management.Smo.DatabaseRole] -or !$_.IsFixedRole} |
+			Where-Object {$_ -isnot [DatabaseRole] -or !$_.IsFixedRole} |
 			Export-DatabaseScript 'Security/Roles'
 		$Database.Schemas |Export-DatabaseScript 'Security/Schemas'
 		$Database.SymmetricKeys |Export-DatabaseScript 'Security/Symmetric Keys'

test/Add-NotebookCell.Tests.ps1

@@ -11,7 +11,8 @@ Describe 'Add-NotebookCell' -Tag Add-NotebookCell -Skip:$skip {
 	BeforeAll {
 		$scriptsdir,$sep = (Split-Path $PSScriptRoot),[io.path]::PathSeparator
 		if($scriptsdir -notin ($env:Path -split $sep)) {$env:Path += "$sep$scriptsdir"}
-		Add-Type -TypeDefinition @'
+		try {[void][Kernel]}
+		catch {Add-Type -TypeDefinition @'
 using System.Collections.Generic;
 public record SendEditableCode(string Language, string Content);
 public static class Kernel
@@ -23,7 +24,7 @@ public static class Kernel
     }
     public static RootMock Root = new RootMock();
 }
-'@
+'@}
 	}
 	Context 'When run within a Polyglot Notebook, appends a cell to it' -Tag AddNoteboodCell,Add,NotebookCell,Notebook {
 		It "Adding language '<Language>' code '<Code>' should happen" -TestCases @(

test/Export-DatabaseScripts.Tests.ps1

@@ -0,0 +1,61 @@
+<#
+.SYNOPSIS
+Tests exporting MS SQL database objects from the given server and database as files, into a consistent folder structure.
+#>
+
+$basename = "$(($MyInvocation.MyCommand.Name -split '\.',2)[0])."
+$skip = !(Test-Path .changes -Type Leaf) ? $false :
+	!@(Get-Content .changes |Get-Item |Select-Object -ExpandProperty Name |Where-Object {$_.StartsWith($basename)})
+if($skip) {Write-Information "No changes to $basename" -infa Continue}
+Describe 'Export-DatabaseScripts' -Tag Export-DatabaseScripts -Skip:$skip {
+	BeforeAll {
+		$scriptsdir,$sep = (Split-Path $PSScriptRoot),[io.path]::PathSeparator
+		if($scriptsdir -notin ($env:Path -split $sep)) {$env:Path += "$sep$scriptsdir"}
+		Mock Export-DbaScript {}
+		try {[void][MockObject]}
+		catch {Add-Type -TypeDefinition @'
+public class ScriptNameObjectBase {}
+public class MockObject : ScriptNameObjectBase
+{
+	public string Name { get; } = $"ObjectName{System.DateTime.Now.Microsecond}";
+	public string Schema { get; } = "dbo";
+	public bool IsSystemObject { get; }
+}
+public class Database : MockObject
+{
+	public bool ServiceBroker { get; }
+	public MockObject[] Assemblies { get; } = new MockObject[] {};
+	public MockObject[] Triggers { get; } = new MockObject[] {};
+	public MockObject[] Defaults { get; } = new MockObject[] {};
+	public MockObject[] ExtendedProperties { get; } = new MockObject[] {};
+	public MockObject[] UserDefinedFunctions { get; } = new MockObject[] {};
+	public MockObject[] Rules { get; } = new MockObject[] {};
+	public MockObject[] AsymmetricKeys { get; } = new MockObject[] {};
+	public MockObject[] Certificates { get; } = new MockObject[] {};
+	public MockObject[] Roles { get; } = new MockObject[] {};
+	public MockObject[] Schemas { get; } = new MockObject[] {};
+	public MockObject[] SymmetricKeys { get; } = new MockObject[] {};
+	public MockObject[] Users { get; } = new MockObject[] {};
+	public MockObject[] Sequences { get; } = new MockObject[] {};
+	public MockObject[] FullTextCatalogs { get; } = new MockObject[] {};
+	public MockObject[] FullTextStopLists { get; } = new MockObject[] {};
+	public MockObject[] PartitionFunctions { get; } = new MockObject[] {};
+	public MockObject[] PartitionSchemes { get; } = new MockObject[] {};
+	public MockObject[] StoredProcedures { get; } = new MockObject[] { new MockObject() };
+	public MockObject[] Synonyms { get; } = new MockObject[] {};
+	public MockObject[] Tables { get; } = new MockObject[] { new MockObject() };
+	public MockObject[] UserDefinedDataTypes { get; } = new MockObject[] {};
+	public MockObject[] XmlSchemaCollections { get; } = new MockObject[] {};
+	public MockObject[] Views { get; } = new MockObject[] { new MockObject() };
+}
+'@}
+	}
+	Context 'Exports MS SQL database objects from the given server and database as files, into a consistent folder structure' `
+		-Tag ExportDatabaseScripts,Export,DatabaseScripts,Database,SQL {
+		It "Export scripts" {
+			New-Object Database |Export-DatabaseScripts.ps1
+			Assert-MockCalled -CommandName Export-DbaScript -Times 3
+		}
+	}
+
+}