tests/New-DbaDatabaseSnapshot.Tests.ps1

$CommandName = $MyInvocation.MyCommand.Name.Replace(".ps1","")
Write-Host -Object "Running $PSCommandpath" -ForegroundColor Cyan
. "$PSScriptRoot\constants.ps1"

# Targets only instance2 because it's the only one where Snapshots can happen
Describe "$CommandName Integration Tests" -Tags "IntegrationTests" {
    BeforeAll {
        if ($env:appveyor) {
            Get-Service | Where-Object { $_.DisplayName -match 'SQL Server (SQL2008R2SP2)' -or $_.DisplayName -match 'SQL Server (SQL2016)' } | Restart-Service -Force
            do {
                Start-Sleep 1
                $null = (& sqlcmd -S $script:instance1 -b -Q "select 1" -d master)
            }
            while ($lastexitcode -ne 0 -and $t++ -lt 10)

            do {
                Start-Sleep 1
                $null = (& sqlcmd -S $script:instance2 -b -Q "select 1" -d master)
            }
            while ($lastexitcode -ne 0 -and $s++ -lt 10)
        }
    }
    Context "Parameter validation" {
        It "Stops if no Database or AllDatabases" {
            { New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent } | Should Throw "You must specify"
        }
        It "Is nice by default" {
            { New-DbaDatabaseSnapshot -SqlInstance $script:instance2 *> $null } | Should Not Throw "You must specify"
        }
    }

    Context "Operations on not supported databases" {
        It "Doesn't support model, master or tempdb" {
            $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database model,master,tempdb
            $result | Should Be $null
        }

    }
    Context "Operations on databases" {
        BeforeAll {
            $server = Connect-DbaSqlServer -SqlInstance $script:instance2
            $db1 = "dbatoolsci_SnapMe"
            $db2 = "dbatoolsci_SnapMe2"
            $db3 = "dbatoolsci_SnapMe3_Offline"
            $server.Query("CREATE DATABASE $db1")
            $server.Query("CREATE DATABASE $db2")
            $server.Query("CREATE DATABASE $db3")
            $needed = Get-DbaDatabase -SqlInstance $script:instance2 -Database $db1, $db2, $db3
            $setupright = $true
            if ($needed.Count -ne 3) {
                $setupright = $false
                it "has failed setup" {
                    Set-TestInconclusive -message "Setup failed"
                }
            }
        }
        AfterAll {
            Remove-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Database $db1,$db2,$db3 -Force
            Remove-DbaDatabase -SqlInstance $script:instance2 -Database $db1,$db2,$db3
        }

        if ($setupright) {
            if (-not $env:appveyor) {
                It "Skips over offline databases nicely" {
                    $server.Query("ALTER DATABASE $db3 SET OFFLINE WITH ROLLBACK IMMEDIATE")
                    $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db3
                    $result | Should Be $null
                    $server.Query("ALTER DATABASE $db3 SET ONLINE WITH ROLLBACK IMMEDIATE")
                }
            }
            It "Refuses to accept multiple source databases with a single name target" {
                { New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db1, $db2 -Name "dbatools_Snapped" } | Should Throw
            }
            It "Halts when path is not accessible" {
                { New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Database $db1 -Path B:\Funnydbatoolspath -Silent } | Should Throw
            }
            It "Creates snaps for multiple dbs by default" {
                $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db1, $db2
                $result | Should Not Be $null
                foreach ($r in $result) {
                    $r.SnapshotOf -in @($db1, $db2) | Should Be $true
                }
            }
            It "Creates snap with the correct name" {
                $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db1 -Name "dbatools_SnapMe_right"
                $result | Should Not Be $null
                $result.SnapshotOf | Should Be $db1
                $result.Database | Should Be "dbatools_SnapMe_right"
            }
            It "Creates snap with the correct name template" {
                $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db2 -NameSuffix "dbatools_SnapMe_{0}_funny"
                $result | Should Not Be $null
                $result.SnapshotOf | Should Be $db2
                $result.Database | Should Be ("dbatools_SnapMe_{0}_funny" -f $db2)
            }
            It "Has the correct properties" {
                $null = Remove-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Database $db2 -Force
                $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db2
                $ExpectedProps = 'ComputerName,Database,DatabaseCreated,InstanceName,Notes,PrimaryFilePath,SizeMB,SnapshotDb,SnapshotOf,SqlInstance,Status'.Split(',')
                ($result.PsObject.Properties.Name | Sort-Object) | Should Be ($ExpectedProps | Sort-Object)
            }

            It "Has the correct default properties" {
                $null = Remove-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Database $db2 -Force
                $result = New-DbaDatabaseSnapshot -SqlInstance $script:instance2 -Silent -Database $db2
                $ExpectedPropsDefault = 'ComputerName,Database,DatabaseCreated,InstanceName,Notes,PrimaryFilePath,SizeMB,SnapshotOf,SqlInstance,Status'.Split(',')
                ($result.PSStandardMembers.DefaultDisplayPropertySet.ReferencedPropertyNames | Sort-Object) | Should Be ($ExpectedPropsDefault | Sort-Object)
            }
        }
    }
}