From 9b6a86bb1a4d939f373b1aabcfafaacf66a088d2 Mon Sep 17 00:00:00 2001
From: Ivan Liu <liubing@microsoft.com>
Date: Tue, 11 Aug 2020 16:26:08 +0800
Subject: [PATCH] add test case for AzPowerShellSetup

---
 .../DataFactoryV2.Test.csproj                   |  2 +-
 .../ScenarioTests/IntegrationRuntimeTests.ps1   | 17 +++++++++++------
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/DataFactory/DataFactoryV2.Test/DataFactoryV2.Test.csproj b/src/DataFactory/DataFactoryV2.Test/DataFactoryV2.Test.csproj
index 90052e1b39bd..60b26df6c846 100644
--- a/src/DataFactory/DataFactoryV2.Test/DataFactoryV2.Test.csproj
+++ b/src/DataFactory/DataFactoryV2.Test/DataFactoryV2.Test.csproj
@@ -11,7 +11,7 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Microsoft.Azure.Management.DataFactory" Version="4.9.0" />
+    <PackageReference Include="Microsoft.Azure.Management.DataFactory" Version="4.10.0" />
     <PackageReference Include="Microsoft.DataTransfer.Gateway.Encryption" Version="3.11.6898.1" />
   </ItemGroup>
 
diff --git a/src/DataFactory/DataFactoryV2.Test/ScenarioTests/IntegrationRuntimeTests.ps1 b/src/DataFactory/DataFactoryV2.Test/ScenarioTests/IntegrationRuntimeTests.ps1
index 79bfc729ec64..83ac6fda9214 100644
--- a/src/DataFactory/DataFactoryV2.Test/ScenarioTests/IntegrationRuntimeTests.ps1
+++ b/src/DataFactory/DataFactoryV2.Test/ScenarioTests/IntegrationRuntimeTests.ps1
@@ -174,12 +174,17 @@ function Test-SsisAzure-IntegrationRuntime
 		$licenseKey = New-Object Microsoft.Azure.Management.DataFactory.Models.SecureString('fakelicensekey')
 		$setup4 = New-Object Microsoft.Azure.Management.DataFactory.Models.ComponentSetup($componentName2, $licenseKey)
 
-		$setups = New-Object System.Collections.ArrayList
-		$setups.Add($setup1)
-		$setups.Add($setup2)
-		# Disable these two setup as it cannot be faked and the other two have already covered the function test
-		# $setups.Add($setup3)
-		# $setups.Add($setup4)
+        # Create setup for Azure PowerShell
+        $version = '4.5.0'
+        $setup5 = New-Object Microsoft.Azure.Management.DataFactory.Models.AzPowerShellSetup($version)
+
+        $setups = New-Object System.Collections.ArrayList
+        $setups.Add($setup1)
+        $setups.Add($setup2)
+        # Disable these two setup as it cannot be faked and the other two have already covered the function test
+        # $setups.Add($setup3)
+        # $setups.Add($setup4)
+        $setups.Add($setup5)
 
         # Replace following variables with network resource ids
         $vnetId = $Env:VnetId