updating test recordings

Author: wiboris

src/Batch/Batch.Test/ScenarioTests/BatchAccountTests.ps1

@@ -84,7 +84,7 @@ function Test-BatchAccountEndToEnd
         {
             # Delete the account
             Remove-AzBatchAccount -Name $accountName -ResourceGroupName $resourceGroup -Force
-            $errorMessage = "The specified account does not exist."
+            $errorMessage = "was not found"
             Assert-ThrowsContains { Get-AzBatchAccount -Name $accountName -ResourceGroupName $resourceGroup } $errorMessage
         }
         finally

src/Batch/Batch.Test/ScenarioTests/ComputeNodeTests.cs

@@ -13,6 +13,8 @@
 // ----------------------------------------------------------------------------------
 
 using Microsoft.Azure.Batch;
+using Microsoft.Azure.Batch.Common;
+using Microsoft.Azure.Commands.Batch.Models;
 using Microsoft.WindowsAzure.Commands.ScenarioTest;
 using System;
 using Xunit;
@@ -114,7 +116,21 @@ public void TestGetComputeNodeRemoteLoginSettings()
                 mockContext =>
                 {
                     context = new ScenarioTestContext();
-                    ScenarioTestHelpers.CreateTestPoolVirtualMachine(this, context, poolId, targetDedicated: 2, targetLowPriority: 0);
+                    PSPoolEndpointConfiguration pSPoolEndpointConfiguration = new PSPoolEndpointConfiguration(new InboundNatPool[] {
+            new InboundNatPool(
+                name: "RDP",
+                protocol: InboundEndpointProtocol.Tcp,
+                backendPort: 3389,
+                frontendPortRangeStart: 1,
+                frontendPortRangeEnd: 1000,
+                networkSecurityGroupRules: null)
+            });
+
+                    PSNetworkConfiguration pSNetworkConfiguration = new PSNetworkConfiguration()
+                    {
+                        EndpointConfiguration = pSPoolEndpointConfiguration
+                    };
+                    ScenarioTestHelpers.CreateTestPoolVirtualMachine(this, context, poolId, targetDedicated: 2, targetLowPriority: 0, networkConfiguration: pSNetworkConfiguration);
                     ScenarioTestHelpers.WaitForSteadyPoolAllocation(this, context, poolId);
                 },
                 $"Test-GetRemoteLoginSettings '{poolId}'"

src/Batch/Batch.Test/ScenarioTests/ScenarioTestHelpers.cs

@@ -143,7 +143,8 @@ public static void CreateTestPoolVirtualMachine(
             int? targetDedicated,
             int? targetLowPriority,
             StartTask startTask = null,
-            UpgradePolicy upgradePolicy = null)
+            UpgradePolicy upgradePolicy = null,
+            PSNetworkConfiguration networkConfiguration = null)
         {
             PSStartTask psStartTask = null;
             if (startTask != null)
@@ -156,6 +157,12 @@ public static void CreateTestPoolVirtualMachine(
             {
                 psUpgradePolicy = new PSUpgradePolicy(upgradePolicy);
             }
+            PSNetworkConfiguration pSNetworkConfiguration = null;
+            if(networkConfiguration != null)
+            {
+                pSNetworkConfiguration = networkConfiguration;
+            }
+
 
             string vmSize = "STANDARD_D2S_V3";
             string publisher = "canonical";
@@ -166,7 +173,7 @@ public static void CreateTestPoolVirtualMachine(
             PSImageReference imageReference = new PSImageReference(offer: offer, publisher: publisher, sku: sku);
             PSVirtualMachineConfiguration vmConfiguration = new PSVirtualMachineConfiguration(imageReference, nodeAgent);
             vmConfiguration.NodePlacementConfiguration = new PSNodePlacementConfiguration(NodePlacementPolicyType.Zonal);
-
+            
             NewPoolParameters parameters = new NewPoolParameters(context, poolId)
             {
                 VirtualMachineSize = vmSize,
@@ -176,7 +183,8 @@ public static void CreateTestPoolVirtualMachine(
                 UpgradePolicy = psUpgradePolicy,
                 StartTask = psStartTask,
                 TaskSlotsPerNode = 1,
-                InterComputeNodeCommunicationEnabled = true
+                InterComputeNodeCommunicationEnabled = true,
+                NetworkConfiguration = pSNetworkConfiguration
             };
 
             CreatePoolIfNotExists(runner, parameters);