Question

我最近发现我不需要使用Import-Module来使用我的高级PowerShell函数，我只能在ps1文件中匿名定义一个函数。

不幸的是，我的Pester单元测试被打破了。我似乎无法在下面的列表中模拟对New-Object的调用。通常情况下，我会点源以下代码，并在我的范围内定义Get-StockQuote函数。现在点源ps1文件没有帮助，因为无论如何我通过文件名调用该函数。

如何使用Pester使用New-Object的模拟实现来测试下面的代码？

注意：对于问题而言，此代码显然是微不足道的，我正在使用的代码测试确实需要New-Object的模拟实现。

# Source listing of the file: Get-StockQuote.ps1
<#
.Synopsis
Looks up a stock quote
.Description
Uses the yahoo api to retrieve a recent quote for a given stock.
.Parameter Symbol
The stock trading symbol
.Example
Get-StockQuote.ps1 -Symbol AMZN
Prints the following line to the output
440.84
#>
[CmdletBinding()]
Param(
    [parameter(Mandatory=$false)]
    [string]$Symbol
)
BEGIN {
    Set-StrictMode -Version 1
}
PROCESS {
    (New-Object System.Net.WebClient).DownloadString("http://finance.yahoo.com/d/quotes.csv?s=$Symbol&f=l1")
}
END {
}

Answer 1

所以我找到了一种方法，通过使用与BEGIN块中的文件名相同的名称定义命名函数并从PROCESS块调用它。

[CmdletBinding()]
Param(
    [parameter(Mandatory=$false)]
    [string]$Symbol
)

 BEGIN {
    Set-StrictMode -Version 1
    Function Get-StockQuote {
        [CmdletBinding()]
        Param(
            [parameter(Mandatory=$false)]
            [string]$Symbol
        )
        BEGIN{}
        PROCESS{
            (New-Object System.Net.WebClient).DownloadString("http://finance.yahoo.com/d/quotes.csv?s=$Symbol&f=l1")
        }
        END{}
    }
 }
 PROCESS {
    Get-StockQuote @PSBoundParameters
 }
 END {
 }

这样，在点源我的ps1文件后，我将在范围内有函数定义，Pester将开始正常工作。

$here = Split-Path -Parent $MyInvocation.MyCommand.Path
$sut = (Split-Path -Leaf $MyInvocation.MyCommand.Path).Replace(".Tests.", ".")

. "$here\$sut"

Describe "Get a stock quote" {
    Mock New-Object {
        $retval = [pscustomobject]@{}
        Add-Member -InputObject $retval -MemberType ScriptMethod DownloadString {
            param( [string] $url )
            if ($url -imatch 'AMZN') {
                return 500.01
            }
            return 100.00
        }
        return $retval
    } -ParameterFilter {$TypeName -and ($TypeName -ilike 'System.Net.WebClient') }
    Context "when called for AMZN" {
        $result = Get-StockQuote -Symbol AMZN
        It "Should RETURN 500.01" {
            $result | should be 500.01
        }
    }
    Context "when called for anything else" {
        $result = Get-StockQuote -Symbol MSFT
        It "Should RETURN 100.00" {
            $result | should be 100.00
        }
    }
}

Answer 2

PowerShell脚本（.ps1文件）在自己的范围内运行，称为脚本范围。所以我认为Pester很难模仿它使用的cmdlet。

在解决方法中，您必须声明一个看到cmdlet的模拟版本的函数。

如何使用Pester来模拟匿名高级函数中的函数

2 个答案: