From 425d62e593503b0b9df9bbcdd9173b0a6414f4bb Mon Sep 17 00:00:00 2001
From: Leonid Lapshin <originalnoe-nazvanie@yandex.ru>
Date: Tue, 1 Sep 2020 17:22:57 +0300
Subject: [PATCH] validate set for new functions

---
 images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1 | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1 b/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
index bf87912d..10e8d0cd 100644
--- a/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
+++ b/images/win/scripts/ImageHelpers/VisualStudioHelpers.ps1
@@ -69,6 +69,7 @@ function Get-VisualStudioInstallation {
     Param
     (
         [Parameter(Mandatory)]
+        [ValidateSet('VS','BuildTools')]
         [String] $VSInstallType
     )
 
@@ -92,6 +93,7 @@ function Get-VisualStudioComponents {
     Param
     (
         [Parameter(Mandatory)]
+        [ValidateSet('VS','BuildTools')]
         [String] $VSInstallType
     )
     (Get-VisualStudioInstallation -VSInstallType $VSInstallType).Packages | Where-Object type -in 'Component', 'Workload' |