PowerShell-PowerShell/test/powershell/Modules/Microsoft.PowerShell.Management/New-PSDrive.Tests.ps1

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

Describe "Tests for New-PSDrive cmdlet." -Tag "CI","RequireAdminOnWindows" {
    Context "Validate New-PSDrive Cmdlet with -Persist switch." {
        BeforeEach {
            $UsedDrives  = Get-PSDrive | Select-Object -ExpandProperty Name
            $PSDriveName = 'D'..'Z' | Where-Object -FilterScript {$_ -notin $UsedDrives} | Get-Random
            $RemoteShare = "\\$env:COMPUTERNAME\$($env:SystemDrive.replace(':','$\'))"
        }

        AfterEach {
            Remove-PSDrive -Name $PSDriveName -Force -ErrorAction SilentlyContinue
        }

        It "Should not throw exception for persistent PSDrive creation." -Skip:(-not $IsWindows) {
            { New-PSDrive -Name $PSDriveName -PSProvider FileSystem -Root $RemoteShare -Persist -ErrorAction Stop } | Should -Not -Throw
        }

        It "Network drive initialization on pwsh startup DisplayRoot should have value of share" -Skip:(-not $IsWindows) {
            $null = New-PSDrive -Name $PSDriveName -PSProvider FileSystem -Root $RemoteShare -Persist -ErrorAction Stop
            $drive = pwsh -noprofile -outputformat XML -command "Get-PSDrive -Name $PSDriveName"
            $drive.DisplayRoot | Should -Be $RemoteShare.TrimEnd('\')
        }

        It "Should throw exception if root is not a remote share." -Skip:(-not $IsWindows) {
            { New-PSDrive -Name $PSDriveName -PSProvider FileSystem -Root "TestDrive:\" -Persist -ErrorAction Stop } | Should -Throw -ErrorId 'DriveRootNotNetworkPath'
        }

        It "Should throw exception if PSDrive is not a drive letter supported by operating system." -Skip:(-not $IsWindows) {
            $PSDriveName = 'AB'
            { New-PSDrive -Name $PSDriveName -PSProvider FileSystem -Root $RemoteShare -Persist -ErrorAction Stop } | Should -Throw -ErrorId 'DriveNameNotSupportedForPersistence'
        }
    }
}