test/powershell/Modules/Microsoft.PowerShell.Management/Get-PSDrive.Tests.ps1 - master - githubmirror/PowerShell

test/powershell/Modules/Microsoft.PowerShell.Management/Get-PSDrive.Tests.ps1
116 строк4 KB
Jordan Borean
Fix up buffer management getting network roots (#24600)
26 ноя 2024, 09:59
26 ноя 2024, 09:5922beba4
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.
Describe "Get-PSDrive" -Tags "CI" {

    It "Should not throw" {
	Get-PSDrive | Should -Not -BeNullOrEmpty
    }

    It "Should have a name and a length property" {
	(Get-PSDrive).Name        | Should -Not -BeNullOrEmpty
	(Get-PSDrive).Root.Length | Should -Not -BeLessThan 1
    }

    It "Should be able to be called with the gdr alias" {
	{ gdr } | Should -Not -Throw

	gdr | Should -Not -BeNullOrEmpty
    }

    It "Should be the same output between Get-PSDrive and gdr" {
	$alias  = gdr
	$actual = Get-PSDrive


	$alias | Should -BeExactly $actual
    }

    It "Should return drive info"{
        (Get-PSDrive Env).Name        | Should -BeExactly Env
        (Get-PSDrive Alias).Name      | Should -BeExactly Alias

        if ($IsWindows)
        {
            (Get-PSDrive Cert).Root       | Should -Be \
            (Get-PSDrive C).Provider.Name | Should -BeExactly FileSystem
        }
        else
        {
            (Get-PSDrive /).Provider.Name | Should -BeExactly FileSystem
        }
    }

    It "Should be able to access a drive using the PSProvider switch" {
	(Get-PSDrive -PSProvider FileSystem).Name.Length | Should -BeGreaterThan 0
    }

    It "Should return true that a drive that does not exist"{
	!(Get-PSDrive fake -ErrorAction SilentlyContinue) | Should -BeTrue
	Get-PSDrive fake -ErrorAction SilentlyContinue    | Should -BeNullOrEmpty
    }
    It "Should be able to determine the amount of free space of a drive" {
        $dInfo = Get-PSDrive TESTDRIVE
        $dInfo.Free -ge 0 | Should -BeTrue
    }
    It "Should be able to determine the amount of Used space of a drive" {
        $dInfo = Get-PSDrive TESTDRIVE
        $dInfo.Used -ge 0 | Should -BeTrue
    }
}

Describe "Temp: drive" -Tag Feature {
    It "TEMP: drive exists" {
        $res = Get-PSDrive Temp
        $res.Name | Should -BeExactly "Temp"
        $res.Root | Should -BeExactly ([System.IO.Path]::GetTempPath())
    }
}

Describe "Get-PSDrive for network path" -Tags "CI","RequireAdminOnWindows" {

    It 'Check P/Invoke GetDosDevice/QueryDosDevice' -Skip:(-not $IsWindows) {
        $UsedDrives  = Get-PSDrive | Select-Object -ExpandProperty Name
        $PSDriveName = 'D'..'Z' | Where-Object -FilterScript {$_ -notin $UsedDrives} | Get-Random

        subst "$($PSDriveName):" \\localhost\c$\Windows

        $drive = Get-PSDrive $PSDriveName
        $drive.DisplayRoot | Should -BeExactly '\\localhost\c$\Windows'
        subst "$($PSDriveName):" /D
    }

    It 'Check P/Invoke for WNetGetConnection with small buffer: <SmallBuffer>' -Skip:(-not $IsWindows) -TestCases @(
        @{ SmallBuffer = $false }
        @{ SmallBuffer = $true }
    ) {
        param ($SmallBuffer)

        $UsedDrives  = Get-PSDrive | Select-Object -ExpandProperty Name
        $PSDriveName = 'D'..'Z' | Where-Object -FilterScript {$_ -notin $UsedDrives} | Get-Random

        $drive = New-PSDrive -Name $PSDriveName -PSProvider FileSystem -Root \\localhost\c$\Windows -Persist
        try {
            if ($SmallBuffer) {
                [System.Management.Automation.Internal.InternalTestHooks]::SetTestHook('WNetGetConnectionBufferSize', 4)
            }

            # The result is cached in the current instance, use a new
            # PowerShell instance to test out WNetGetConnection code.
            $ps = [PowerShell]::Create()
            $actual = $ps.AddCommand('Get-PSDrive').AddParameter('Name', $PSDriveName).Invoke()
            if ($ps.HadErrors) {
                throw $ps.Streams.Error[0]
            }

            $actual.Name | Should -BeExactly $PSDriveName
            $actual.DisplayRoot | Should -BeExactly '\\localhost\c$\Windows'
        }
        finally {
            $drive | Remove-PSDrive

            if ($SmallBuffer) {
                [System.Management.Automation.Internal.InternalTestHooks]::SetTestHook('WNetGetConnectionBufferSize', -1)
            }
        }
    }
}