Add a test for IValidateSetValuesGenerator in a module (#5830)

iSazonov · web-flow · commit b6f18e710128 · 2018-01-10T13:02:33.000+04:00
Add a test that has been skipped in PR #5702
diff --git a/test/powershell/Language/Classes/Scripting.Classes.Attributes.Tests.ps1 b/test/powershell/Language/Classes/Scripting.Classes.Attributes.Tests.ps1
@@ -410,6 +410,39 @@ Describe 'ValidateSet support a dynamically generated set' -Tag "CI" {
                 Get-TestValidateSetPS6 -Param1 "AnyTestString" -ErrorAction Stop
             } | ShouldBeErrorId "TypeNotFound"
         }
+
+        It 'IValidateSetValuesGenerator works in PowerShell module' {
+            $moduleFile = Join-Path $TestDrive -ChildPath "Test-Module-$((New-Guid).Guid).psm1"
+            $module = @'
+                class ValidateSetTest : System.Management.Automation.IValidateSetValuesGenerator
+                {
+                    [string[]] GetValidValues()
+                    {
+                        return 'Hello', 'World'
+                    }
+                }
+
+                function Test-ValidateSet
+                {
+                    [CmdletBinding()]
+                    param (
+                        [Parameter(Mandatory = $true)]
+                        [ValidateSet([ValidateSetTest])]
+                        [string[]]
+                        $Item
+                    )
+                    $Item
+                }
+'@
+            Set-Content -Path $moduleFile -Value $module -Force
+
+            try {
+                Import-Module -Name $moduleFile -Force
+                Test-ValidateSet 'Hello' | Should Be 'Hello'
+            } finally {
+                Remove-Module -Name $moduleFile -Force
+            }
+        }
     }
 
     Context 'CachedValidValuesGeneratorBase class tests' {
